From 0e8195ae6111b61e7a7b0d0decde3a8595ca399c Mon Sep 17 00:00:00 2001 From: lichengzhe <38408577@qq.com> Date: Mon, 15 May 2023 13:55:14 +0800 Subject: [PATCH] Bad Gateway exception retry --- bot/chatgpt/chat_gpt_bot.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/bot/chatgpt/chat_gpt_bot.py b/bot/chatgpt/chat_gpt_bot.py index b46751a..d48bf38 100644 --- a/bot/chatgpt/chat_gpt_bot.py +++ b/bot/chatgpt/chat_gpt_bot.py @@ -134,12 +134,17 @@ class ChatGPTBot(Bot, OpenAIImage): result["content"] = "我没有收到你的消息" if need_retry: time.sleep(5) + elif isinstance(e, openai.error.APIError:): + logger.warn("[CHATGPT] Bad Gateway: {}".format(e)) + result["content"] = "请再问我一次" + if need_retry: + time.sleep(10) elif isinstance(e, openai.error.APIConnectionError): logger.warn("[CHATGPT] APIConnectionError: {}".format(e)) need_retry = False result["content"] = "我连接不到你的网络" else: - logger.warn("[CHATGPT] Exception: {}".format(e)) + logger.exception("[CHATGPT] Exception: {}".format(e)) need_retry = False self.sessions.clear_session(session.session_id)