diff --git a/swarms/models/openai_models.py b/swarms/models/openai_models.py index b4de06df..704b0cfe 100644 --- 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.error.APIError, - openai.error.APIConnectionError, - openai.error.RateLimitError, - openai.error.ServiceUnavailableError, + openai.Timeout, + openai.APIError, + openai.APIConnectionError, + openai.RateLimitError, ] return create_base_retry_decorator( error_types=errors, max_retries=llm.max_retries, run_manager=run_manager