From 20889205e83b615e18739a9dc340c30242d02fcf Mon Sep 17 00:00:00 2001 From: Oliver Klingefjord Date: Mon, 13 Feb 2023 22:30:06 +0100 Subject: [PATCH] Added retry for openai.error.ServiceUnavailableError (#1022) Imho retries should be performed for ServiceUnavailableError (which tends to happen to me quite often). --- langchain/llms/openai.py | 1 + 1 file changed, 1 insertion(+) diff --git a/langchain/llms/openai.py b/langchain/llms/openai.py index 8716c0e3041..6bb73122fe9 100644 --- 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), )