fix: method error not found

Former-commit-id: b92394ad3965807008506ab524529feeda818ec2
1 year ago · 4ac0a1daba
parent 11f06035b4
commit 4ac0a1daba
1 changed files with 4 additions and 5 deletions
--- a/swarms/models/openai_models.py
+++ b/swarms/models/openai_models.py
@ -103,11 +103,10 @@ def _create_retry_decorator(
    import openai
    errors = [
-        openai.error.Timeout,
+        openai.Timeout,
-        openai.error.APIError,
+        openai.APIError,
-        openai.error.APIConnectionError,
+        openai.APIConnectionError,
-        openai.error.RateLimitError,
+        openai.RateLimitError,
        openai.error.ServiceUnavailableError,
    ]
    return create_base_retry_decorator(
        error_types=errors, max_retries=llm.max_retries, run_manager=run_manager