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

Imho retries should be performed for ServiceUnavailableError (which
tends to happen to me quite often).
This commit is contained in:
Oliver Klingefjord 2023-02-13 22:30:06 +01:00 committed by GitHub
parent fc2502cd81
commit 20889205e8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -163,6 +163,7 @@ class BaseOpenAI(BaseLLM, BaseModel):
| retry_if_exception_type(openai.error.APIError) | retry_if_exception_type(openai.error.APIError)
| retry_if_exception_type(openai.error.APIConnectionError) | retry_if_exception_type(openai.error.APIConnectionError)
| retry_if_exception_type(openai.error.RateLimitError) | retry_if_exception_type(openai.error.RateLimitError)
| retry_if_exception_type(openai.error.ServiceUnavailableError)
), ),
before_sleep=before_sleep_log(logger, logging.WARNING), before_sleep=before_sleep_log(logger, logging.WARNING),
) )