Merge pull request #1577 from xyshell/patch-1

Update chat_gpt_bot.py retry APIConnectionError
zhayujie · Dec 8, 2023 · b9dfdce · b9dfdce
2 parents 9d4afea + 55df191
commit b9dfdce
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/bot/chatgpt/chat_gpt_bot.py b/bot/chatgpt/chat_gpt_bot.py
@@ -148,8 +148,9 @@ def reply_text(self, session: ChatGPTSession, api_key=None, args=None, retry_cou
                     time.sleep(10)
             elif isinstance(e, openai.error.APIConnectionError):
                 logger.warn("[CHATGPT] APIConnectionError: {}".format(e))
-                need_retry = False
                 result["content"] = "我连接不到你的网络"
+                if need_retry:
+                    time.sleep(5)
             else:
                 logger.exception("[CHATGPT] Exception: {}".format(e))
                 need_retry = False