Sfoglia il codice sorgente

[feat] catch connection failed exception

sunxin.181 3 anni fa
parent
commit
a2db765ade
1 ha cambiato i file con 9 aggiunte e 0 eliminazioni
  1. 9 0
      bot/chatgpt/chat_gpt_bot.py

+ 9 - 0
bot/chatgpt/chat_gpt_bot.py

@@ -63,6 +63,15 @@ class ChatGPTBot(Bot):
                 return self.reply_text(query, user_id, retry_count+1)
             else:
                 return "提问太快啦,请休息一下再问我吧"
+        except openai.error.APIConnectionError as e:
+            # api connection exception
+            logger.warn(e)
+            logger.warn("[OPEN_AI] APIConnection failed")
+            return "我连接不到你的网络"
+        except openai.error.Timeout as e:
+            logger.warn(e)
+            logger.warn("[OPEN_AI] Timeout")
+            return "我没有收到你的消息"
         except Exception as e:
             # unknown exception
             logger.exception(e)