Added retry for openai.error.ServiceUnavailableError (#1022)

Imho retries should be performed for ServiceUnavailableError (which tends to happen to me quite often).
1 year ago · 20889205e8
parent fc2502cd81
commit 20889205e8
1 changed files with 1 additions and 0 deletions
--- a/langchain/llms/openai.py
+++ b/langchain/llms/openai.py
@ -163,6 +163,7 @@ class BaseOpenAI(BaseLLM, BaseModel):
                | retry_if_exception_type(openai.error.APIError)
                | retry_if_exception_type(openai.error.APIConnectionError)
                | retry_if_exception_type(openai.error.RateLimitError)
+                | retry_if_exception_type(openai.error.ServiceUnavailableError)
            ),
            before_sleep=before_sleep_log(logger, logging.WARNING),
        )