From 507f35a6027d263159d37816e033021b578d27db Mon Sep 17 00:00:00 2001
From: Alexander_Kabui
Date: Fri, 5 Jan 2024 22:49:56 +0300
Subject: http request header fixes

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

(limited to 'gn3/llms/client.py')

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)
 
-- 
cgit v1.2.3