From 8eba10b3d9ca0fee764f74127a69598d2660be37 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Florian=20Ho=CC=88nicke?= Date: Tue, 23 May 2023 16:29:16 +0200 Subject: [PATCH] =?UTF-8?q?=E2=9A=BE=20fix:=20catch=20api=20error=20from?= =?UTF-8?q?=20openai?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- dev_gpt/apis/gpt.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/dev_gpt/apis/gpt.py b/dev_gpt/apis/gpt.py index 335eab0..ec6ec99 100644 --- a/dev_gpt/apis/gpt.py +++ b/dev_gpt/apis/gpt.py @@ -10,7 +10,7 @@ from langchain.callbacks import CallbackManager from langchain.callbacks.streaming_stdout import StreamingStdOutCallbackHandler from langchain.chat_models import ChatOpenAI from langchain.schema import HumanMessage, SystemMessage, BaseMessage, AIMessage -from openai.error import RateLimitError +from openai.error import RateLimitError, APIError from requests.exceptions import ConnectionError, ChunkedEncodingError from urllib3.exceptions import InvalidChunkLength @@ -149,7 +149,7 @@ class _GPTConversation: response = self._chat(self.messages) self.conversation_logger.log(self.messages, response) break - except (ConnectionError, InvalidChunkLength, ChunkedEncodingError) as e: + except (ConnectionError, InvalidChunkLength, ChunkedEncodingError, APIError) as e: print('There was a connection error. Retrying...') if i == 9: raise e