diff options
| -rw-r--r-- | gn3/llms/client.py | 6 |
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) |