perf: 完善OpenAI接口异常处理 #1

config-template.py

@@ -58,6 +58,9 @@ session_expire_time = 60 * 20
 # 日志级别
 logging_level = logging.INFO
 
+# 管理员QQ号，用于接收报错等通知
+admin_qq = 0
+
 # 定制帮助消息
 help_message = """此机器人通过调用OpenAI的GPT-3大型语言模型生成回复，不具有情感。
 你可以用自然语言与其交流，回复的消息中[GPT]开头的为模型生成的语言，[bot]开头的为程序提示。

pkg/qqbot/manager.py

@@ -1,3 +1,7 @@
+import asyncio
+import threading
+
+import openai.error
 from mirai import At, GroupMessage, MessageEvent, Mirai, Plain, StrangerMessage, WebSocketAdapter, FriendMessage, Image
 
 import config
@@ -57,76 +61,99 @@ class QQBotManager:
         session_name = "{}_{}".format(launcher_type, launcher_id)
 
         if text_message.startswith('!') or text_message.startswith("！"):  # 指令
+            try:
+                logging.info("[{}]发起指令:{}".format(session_name, text_message[:min(20, len(text_message))] + (
+                    "..." if len(text_message) > 20 else "")))
 
-            logging.info("[{}]发起指令:{}".format(session_name, text_message[:min(20, len(text_message))]+("..." if len(text_message) > 20 else "")))
+                cmd = text_message[1:].strip().split(' ')[0]
 
-            cmd = text_message[1:].strip().split(' ')[0]
-
-            params = text_message[1:].strip().split(' ')[1:]
-            if cmd == 'help':
-                reply = "[bot]" + help_text
-            elif cmd == 'reset':
-                pkg.openai.session.get_session(session_name).reset(explicit=True)
-                reply = "[bot]会话已重置"
-            elif cmd == 'last':
-                result = pkg.openai.session.get_session(session_name).last_session()
-                if result is None:
-                    reply = "[bot]没有前一次的对话"
-                else:
-                    datetime_str = datetime.datetime.fromtimestamp(result.create_timestamp).strftime('%Y-%m-%d %H:%M:%S')
-                    reply = "[bot]已切换到前一次的对话：\n创建时间:{}\n".format(datetime_str) + result.prompt[:min(100, len(result.prompt))] + \
-                            ("..." if len(result.prompt) > 100 else "#END#")
-            elif cmd == 'next':
-                result = pkg.openai.session.get_session(session_name).next_session()
-                if result is None:
-                    reply = "[bot]没有后一次的对话"
-                else:
-                    datetime_str = datetime.datetime.fromtimestamp(result.create_timestamp).strftime('%Y-%m-%d %H:%M:%S')
-                    reply = "[bot]已切换到后一次的对话：\n创建时间:{}\n".format(datetime_str) + result.prompt[:min(100, len(result.prompt))] + \
-                            ("..." if len(result.prompt) > 100 else "#END#")
-            elif cmd == 'prompt':
-                reply = "[bot]当前对话所有内容：\n" + pkg.openai.session.get_session(session_name).prompt
-            elif cmd == 'list':
-                pkg.openai.session.get_session(session_name).persistence()
-                page = 0
-
-                if len(params) > 0:
-                    try:
-                        page = int(params[0])
-                    except ValueError:
-                        pass
-
-                results = pkg.openai.session.get_session(session_name).list_history(page=page)
-                if len(results) == 0:
-                    reply = "[bot]第{}页没有历史会话".format(page)
-                else:
-                    reply = "[bot]历史会话 第{}页：\n".format(page)
-                    current = -1
-                    for i in range(len(results)):
-                        # 时间(使用create_timestamp转换) 序号 部分内容
-                        datetime_obj = datetime.datetime.fromtimestamp(results[i]['create_timestamp'])
-                        reply += "#{} 创建:{} {}\n".format(i + page * 10, datetime_obj.strftime("%Y-%m-%d %H:%M:%S"),
-                                                          results[i]['prompt'][:min(20, len(results[i]['prompt']))])
-                        if results[i]['create_timestamp'] == pkg.openai.session.get_session(session_name).create_timestamp:
-                            current = i + page * 10
-
-                    reply += "\n以上信息倒序排列"
-                    if current != -1:
-                        reply += ",当前会话是 #{}\n".format(current)
+                params = text_message[1:].strip().split(' ')[1:]
+                if cmd == 'help':
+                    reply = "[bot]" + help_text
+                elif cmd == 'reset':
+                    pkg.openai.session.get_session(session_name).reset(explicit=True)
+                    reply = "[bot]会话已重置"
+                elif cmd == 'last':
+                    result = pkg.openai.session.get_session(session_name).last_session()
+                    if result is None:
+                        reply = "[bot]没有前一次的对话"
                     else:
-                        reply += ",当前处于全新会话或不在此页"
+                        datetime_str = datetime.datetime.fromtimestamp(result.create_timestamp).strftime(
+                            '%Y-%m-%d %H:%M:%S')
+                        reply = "[bot]已切换到前一次的对话：\n创建时间:{}\n".format(datetime_str) + result.prompt[
+                                                                                                   :min(100,
+                                                                                                        len(result.prompt))] + \
+                                ("..." if len(result.prompt) > 100 else "#END#")
+                elif cmd == 'next':
+                    result = pkg.openai.session.get_session(session_name).next_session()
+                    if result is None:
+                        reply = "[bot]没有后一次的对话"
+                    else:
+                        datetime_str = datetime.datetime.fromtimestamp(result.create_timestamp).strftime(
+                            '%Y-%m-%d %H:%M:%S')
+                        reply = "[bot]已切换到后一次的对话：\n创建时间:{}\n".format(datetime_str) + result.prompt[
+                                                                                                   :min(100,
+                                                                                                        len(result.prompt))] + \
+                                ("..." if len(result.prompt) > 100 else "#END#")
+                elif cmd == 'prompt':
+                    reply = "[bot]当前对话所有内容：\n" + pkg.openai.session.get_session(session_name).prompt
+                elif cmd == 'list':
+                    pkg.openai.session.get_session(session_name).persistence()
+                    page = 0
 
+                    if len(params) > 0:
+                        try:
+                            page = int(params[0])
+                        except ValueError:
+                            pass
+
+                    results = pkg.openai.session.get_session(session_name).list_history(page=page)
+                    if len(results) == 0:
+                        reply = "[bot]第{}页没有历史会话".format(page)
+                    else:
+                        reply = "[bot]历史会话 第{}页：\n".format(page)
+                        current = -1
+                        for i in range(len(results)):
+                            # 时间(使用create_timestamp转换) 序号 部分内容
+                            datetime_obj = datetime.datetime.fromtimestamp(results[i]['create_timestamp'])
+                            reply += "#{} 创建:{} {}\n".format(i + page * 10,
+                                                               datetime_obj.strftime("%Y-%m-%d %H:%M:%S"),
+                                                               results[i]['prompt'][
+                                                               :min(20, len(results[i]['prompt']))])
+                            if results[i]['create_timestamp'] == pkg.openai.session.get_session(
+                                    session_name).create_timestamp:
+                                current = i + page * 10
+
+                        reply += "\n以上信息倒序排列"
+                        if current != -1:
+                            reply += ",当前会话是 #{}\n".format(current)
+                        else:
+                            reply += ",当前处于全新会话或不在此页"
+            except Exception as e:
+                self.notify_admin("{}指令执行失败:{}".format(session_name, e))
+                logging.exception(e)
+                reply = "[bot]err:{}".format(e)
         else:  # 消息
-            logging.info("[{}]发送消息:{}".format(session_name, text_message[:min(20, len(text_message))]+("..." if len(text_message) > 20 else "")))
+            logging.info("[{}]发送消息:{}".format(session_name, text_message[:min(20, len(text_message))] + (
+                "..." if len(text_message) > 20 else "")))
 
             session = pkg.openai.session.get_session(session_name)
             try:
                 reply = "[GPT]" + session.append(text_message)
+            except openai.error.APIConnectionError as e:
+                self.notify_admin("{}会话调用API失败:{}".format(session_name, e))
+                # logging.exception(e)
+                reply = "[bot]err:调用API失败，请联系作者，或等待修复"
+            except openai.error.RateLimitError as e:
+                # logging.exception(e)
+                self.notify_admin("API调用额度超限,请向OpenAI账户充值或在config.py中更换api_key")
+                reply = "[bot]err:API调用额度超额，请联系作者，或等待修复"
             except Exception as e:
                 logging.exception(e)
                 reply = "[bot]err:{}".format(e)
 
-        logging.info("回复[{}]消息:{}".format(session_name, reply[:min(100, len(reply))]+("..." if len(reply) > 100 else "")))
+        logging.info(
+            "回复[{}]消息:{}".format(session_name, reply[:min(100, len(reply))] + ("..." if len(reply) > 100 else "")))
 
         return reply
 
@@ -197,6 +224,11 @@ class QQBotManager:
         if reply != '':
             return await self.bot.send(event, reply)
 
+    def notify_admin(self, message: str):
+        if config.admin_qq is not None and config.admin_qq != 0:
+            send_task = self.bot.send_friend_message(config.admin_qq, "[bot]{}".format(message))
+            threading.Thread(target=asyncio.run, args=(send_task,)).start()
+
 
 def get_inst() -> QQBotManager:
     global inst