aljaz/dev-gpt
1
0
Fork 0
mirror of https://github.com/aljazceru/dev-gpt.git synced 2025-12-19 22:54:21 +01:00
Code Issues Packages Projects Releases Wiki Activity

⚾ fix: catch api error from openai

Browse Source
This commit is contained in:
Florian Hönicke
2023-05-23 16:29:16 +02:00
parent ba46eb3ac1
commit 8eba10b3d9
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
dev_gpt/apis/gpt.py
View File

@@ -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