openai api fix

Former-commit-id: ba7a468
kyegomez · Nov 11, 2023 · bb90b27 · bb90b27
1 parent 712bd0d
commit bb90b27
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. "