From a0b898df86efde75e85d355b19daf47e2cb0fea7 Mon Sep 17 00:00:00 2001
From: Alexander_Kabui
Date: Mon, 8 Jan 2024 10:29:56 +0300
Subject: disable process wait

---
 gn3/llms/client.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/gn3/llms/client.py b/gn3/llms/client.py
index 18c0217..77ae32e 100644
--- a/gn3/llms/client.py
+++ b/gn3/llms/client.py
@@ -57,7 +57,7 @@ class GeneNetworkQAClient(Session):
 
     BASE_URL = 'https://genenetwork.fahamuai.com/api/tasks'
 
-    def __init__(self, account, api_key, version="v3", timeout=120, total_retries=5, backoff_factor=30):
+    def __init__(self, account, api_key, version="v3", timeout=5, total_retries=5, backoff_factor=30):
         super().__init__()
         self.headers.update(
             {"Authorization": "Bearer " + api_key})
@@ -132,8 +132,9 @@ class GeneNetworkQAClient(Session):
         return res, 1
 
     def custom_request(self, method, url, *args, **kwargs):
-        max_retries = 5
-        retry_delay = 10
+
+        max_retries = 1
+        retry_delay = 1
 
         response = super().request(method, url, *args, **kwargs)
 
-- 
cgit v1.2.3