Increased timeout and max retries

VRSEN · Aug 18, 2024 · 43b1267 · 43b1267
1 parent c7dc1f5
commit 43b1267
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/agency_swarm/util/oai.py b/agency_swarm/util/oai.py
@@ -20,8 +20,8 @@ def get_openai_client():
             if api_key is None:
                 raise ValueError("OpenAI API key is not set. Please set it using set_openai_key.")
             client = openai.OpenAI(api_key=api_key,
-                                   timeout=httpx.Timeout(60.0, read=30, connect=5.0),
-                                   max_retries=5,
+                                   timeout=httpx.Timeout(60.0, read=40, connect=5.0),
+                                   max_retries=10,
                                    default_headers={"OpenAI-Beta": "assistants=v2"})
     return client