aljaz/dev-gpt
1
0
Fork 0
mirror of https://github.com/aljazceru/dev-gpt.git synced 2026-01-02 13:24:22 +01:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #110 from jina-ai/fix_catch_api_error_from_openai

Browse Source 
⚾ fix: catch api error from openai
This commit is contained in:
Florian Hönicke
2023-05-23 16:34:11 +02:00
committed by GitHub
parent 37cb4f6b03 8eba10b3d9
commit 28527c9b6a
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