openai api fix

kyegomez · Nov 11, 2023 · ba7a468 · ba7a468
1 parent 285a3bd
commit ba7a468
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/swarms/models/openai_chat.py b/swarms/models/openai_chat.py
@@ -102,7 +102,7 @@ def _create_retry_decorator(
 
     errors = [
         openai.Timeout,
-        openai.error.APIError,
+        openai.APIError,
         openai.error.APIConnectionError,
         openai.error.RateLimitError,
         openai.error.ServiceUnavailableError,

diff --git a/swarms/models/openai_models.py b/swarms/models/openai_models.py
@@ -788,10 +788,10 @@ def validate_environment(cls, values: Dict) -> Dict:
             if openai_organization:
                 raise Exception("The 'openai.organization' option isn't read in the client API. You will need to pass it when you instantiate the client, e.g. 'OpenAI(organization=openai_organization)'")
             if openai_proxy:
-                raise Exception("The 'openai.proxy' option isn't read in the client API. You will need to pass it when you instantiate the client, e.g. 'OpenAI(proxy={
+                openai.proxy = {
                     "http": openai_proxy,
                     "https": openai_proxy,
-                })'")  # type: ignore[assignment]  # noqa: E501
+                }  # type: ignore[assignment]  # noqa: E501
         except ImportError:
             raise ImportError(
                 "Could not import openai python package. "