http request header fixes

1 files changed, 3 insertions, 3 deletions

diff --git a/gn3/llms/client.py b/gn3/llms/client.py
index 10764b1..18c0217 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=5, total_retries=5, backoff_factor=30):
+    def __init__(self, account, api_key, version="v3", timeout=120, total_retries=5, backoff_factor=30):
         super().__init__()
         self.headers.update(
             {"Authorization": "Bearer " + api_key})
@@ -132,8 +132,8 @@ class GeneNetworkQAClient(Session):
         return res, 1
 
     def custom_request(self, method, url, *args, **kwargs):
-        max_retries = 20
-        retry_delay = 20
+        max_retries = 5
+        retry_delay = 10
 
         response = super().request(method, url, *args, **kwargs)