feat:

1. 拦截词拦截；
2. 新增多条指令：/继续 等；
3. 日志记录。
4. 聊天key注入（/key指令）。
5. 中文指令

cores/openai/core.py

@@ -19,28 +19,8 @@ class ChatGPT:
             else:
                 print("请先去完善ChatGPT的Key。详情请前往https://beta.openai.com/account/api-keys")
         
-        with open("chatgpt_key_record", 'r', encoding='utf-8') as keyfile:
-            try:
-                self.key_stat = json.load(keyfile)
-            except Exception as e:
-                print(e)
-                self.key_stat = {}
-            finally:
-                for key in self.key_list:
-                    if key not in self.key_stat:
-                        self.key_stat[key] = {'exceed': False, 'used': 0}
-                        if openai.api_key is None:
-                            print("切换")
-                            openai.api_key = key
-                    else:
-                        if self.key_stat[key]['exceed']:
-                            print(f"Key: {key} 已超额")
-                            continue
-                        else:
-                            if openai.api_key is None:
-                                openai.api_key = key
-                                print(f"使用Key: {key}, 已使用token: {self.key_stat[key]['used']}")
-            self.save_key_record()
+        # init key record
+        self.init_key_record()
 
         chatGPT_configs = cfg['openai']['chatGPTConfigs']
         print(f'加载ChatGPTConfigs: {chatGPT_configs}')
@@ -62,11 +42,9 @@ class ChatGPT:
                     n=1,
                     size="512x512",
                 )
-        # except(openai.error.InvalidRequestError) as e:
-        #     raise PromptExceededError("OpenAI遇到错误：输入了一个不合法的请求。\n"+str(e))
         except Exception as e:
             print(e)
-            if 'You exceeded' in str(e) or 'Billing hard limit has been reached' in str(e):
+            if 'You exceeded' in str(e) or 'Billing hard limit has been reached' in str(e) or 'No API key provided.' in str(e):
                 print("当前Key已超额，正在切换")
                 self.key_stat[openai.api_key]['exceed'] = True
                 self.save_key_record()
@@ -103,20 +81,23 @@ class ChatGPT:
                     is_all_exceed = False
                     openai.api_key = key
                     print(f"切换到Key: {key}, 已使用token: {self.key_stat[key]['used']}")
-                    try:
-                        response = openai.Completion.create(
-                            prompt=prompt,
-                            **self.chatGPT_configs
-                        )
-                        return response, True
-                    except Exception as e:
-                        print(e)
-                        if 'You exceeded' in str(e):
-                            print("当前Key已超额，正在切换")
-                            self.key_stat[openai.api_key]['exceed'] = True
-                            self.save_key_record()
-                            time.sleep(1)
-                            continue
+                    if prompt != '':
+                        try:
+                            response = openai.Completion.create(
+                                prompt=prompt,
+                                **self.chatGPT_configs
+                            )
+                            return response, True
+                        except Exception as e:
+                            print(e)
+                            if 'You exceeded' in str(e):
+                                print("当前Key已超额，正在切换")
+                                self.key_stat[openai.api_key]['exceed'] = True
+                                self.save_key_record()
+                                time.sleep(1)
+                                continue
+                    else:
+                        return True
             if is_all_exceed:
                 print("所有Key已超额")
                 return None, False
@@ -132,6 +113,55 @@ class ChatGPT:
         return self.key_stat
     def get_key_list(self):
         return self.key_list
+    
+    # 添加key
+    def append_key(self, key, sponsor):
+        self.key_list.append(key)
+        self.key_stat[key] = {'exceed': False, 'used': 0, 'sponsor': sponsor}
+        self.save_key_record()
+        self.init_key_record()
+    # 检查key是否可用
+    def check_key(self, key):
+        pre_key = openai.api_key
+        openai.api_key = key
+        try:
+            openai.Completion.create(
+                prompt="test",
+                **self.chatGPT_configs
+            )
+            openai.api_key = pre_key
+            return True
+        except Exception as e:
+            pass
+        openai.api_key = pre_key
+        return False
+
+    # 将key_list的key转储到key_record中，并记录相关数据
+    def init_key_record(self):
+        with open("chatgpt_key_record", 'r', encoding='utf-8') as keyfile:
+            try:
+                self.key_stat = json.load(keyfile)
+            except Exception as e:
+                print(e)
+                self.key_stat = {}
+            finally:
+                for key in self.key_list:
+                    if key not in self.key_stat:
+                        self.key_stat[key] = {'exceed': False, 'used': 0}
+                        if openai.api_key is None:
+                            print("切换")
+                            openai.api_key = key
+                    else:
+                        if self.key_stat[key]['exceed']:
+                            print(f"Key: {key} 已超额")
+                            continue
+                        else:
+                            if openai.api_key is None:
+                                openai.api_key = key
+                                print(f"使用Key: {key}, 已使用token: {self.key_stat[key]['used']}")
+                if openai.api_key == None:
+                    self.handle_switch_key("")
+            self.save_key_record()
 
 def getInst() -> ChatGPT:
     global inst

cores/qqbot/core.py

@@ -16,7 +16,7 @@ import util.unfit_words as uw
 history_dump_interval = 10
 client = ''
 # ChatGPT的实例
-chatgpt = ""
+global chatgpt
 # 缓存的会话
 session_dict = {}
 # 最大缓存token（在配置里改 configs/config.yaml）
@@ -261,7 +261,7 @@ def oper_msg(message, at=False, loop=None):
         name = "频道"
 
     # 指令控制
-    if qq_msg == "/reset":
+    if qq_msg == "/reset" or qq_msg == "/重置":
         msg = ''
         session_dict[session_id] = []
         if at:
@@ -297,7 +297,7 @@ def oper_msg(message, at=False, loop=None):
             msg=f"会话的token数: {get_user_usage_tokens(session_dict[session_id])}\n系统最大缓存token数: {max_tokens}"
         send_qq_msg(message, msg)
         return
-    if qq_msg == "/status":
+    if qq_msg == "/status" or qq_msg == "/状态":
         chatgpt_cfg_str = ""
         key_stat = chatgpt.get_key_stat()
         key_list = chatgpt.get_key_list()
@@ -306,21 +306,24 @@ def oper_msg(message, at=False, loop=None):
         max = 900000
         gg_count = 0
         total = 0
-        for key in key_list:
-            if key in key_stat:
-                total += key_stat[key]['used']
-                if key_stat[key]['exceed']:
-                    gg_count += 1
-                    continue
-                # chatgpt_cfg_str += f"#{index}: {round(key_stat[key]['used']/max*100, 2)}%\n"
-                chatgpt_cfg_str += f"  |-{index}: {key_stat[key]['used']}/{max}\n"
-                index += 1
+        for key in key_stat.keys():
+            sponsor = ''
+            total += key_stat[key]['used']
+            if key_stat[key]['exceed']:
+                gg_count += 1
+                continue
+            if 'sponsor' in key_stat[key]:
+                sponsor = key_stat[key]['sponsor']
+                
+            # chatgpt_cfg_str += f"#{index}: {round(key_stat[key]['used']/max*100, 2)}%\n"
+            chatgpt_cfg_str += f"  |-{index}: {key_stat[key]['used']}/{max} 由{sponsor}赞助\n"
+            index += 1
 
         chatgpt_cfg_str += f"  {str(gg_count)}个已用\n"
         print("生成...")
         send_qq_msg(message, f"{version}\n{chatgpt_cfg_str}\n⏰截至目前，全频道已在本机器人使用{total}个token\n🤖可自己搭建一个机器人~点击头像进入官方频道了解详情。\n\n{announcement}")
         return
-    if qq_msg == "/count":
+    if qq_msg == "/count" or qq_msg == "/统计":
         try:
             f = open("./configs/stat", "r", encoding="utf-8")
             fjson = json.loads(f.read())
@@ -351,6 +354,22 @@ def oper_msg(message, at=False, loop=None):
     
     if qq_msg == "/继续":
         qq_msg == "继续"
+
+    if qq_msg[:4] == "/key":
+        if len(qq_msg) == 4:
+            send_qq_msg(message, "感谢您赞助key喵 请以以下格式赞助:\n/key xxxxx")
+            return
+        key = qq_msg[5:]
+        send_qq_msg(message, "收到！正在核验...")
+        if chatgpt.check_key(key):
+            send_qq_msg(message, f"*★,°*:.☆(￣▽￣)/$:*.°★* 。\n该Key被验证为有效。感谢{message.member.nick}赞助~ 未来赞助的key仅能在本频道使用")
+            chatgpt.append_key(key, message.member.nick)
+            return
+        else:
+            send_qq_msg(message, "该Key被验证为无效。也许是您输入错误了呢~")
+            return
+        
+
         
     # if qq_msg[0:6] == '/draw ':
     #     # TODO 未实现
@@ -366,24 +385,23 @@ def oper_msg(message, at=False, loop=None):
     #     send_qq_msg(message, qiniu_url, image_mode=True)
     #     return
 
-    # 预设区，暂时注释掉了，想要可以去除注释。
-    # if qq_msg.strip() == 'hello' or qq_msg.strip() == '你好' or qq_msg.strip() == '':
-    #     send_qq_msg(message, f"你好呀~")
-    #     return
-    # if qq_msg.strip() == '傻逼' or qq_msg.strip() == 'sb':
-    #     send_qq_msg(message, f"好好好")
-    #     return
+    if qq_msg.strip() == 'hello' or qq_msg.strip() == '你好' or qq_msg.strip() == '':
+        send_qq_msg(message, f"你好呀~")
+        return
+    if qq_msg.strip() == '傻逼' or qq_msg.strip() == 'sb':
+        send_qq_msg(message, f"好好好")
+        return
     
-    # if '波奇' in qq_msg:
-    #     bq = random.randint(1,3)
-    #     send_qq_msg(message, f"http://rn2ztkeap.hn-bkt.clouddn.com/bq{bq}.gif", True)
-    # if '喜多' in qq_msg:
-    #     send_qq_msg(message, 'http://rn2ztkeap.hn-bkt.clouddn.com/xd1.gif', True)
-    # if '孤独摇滚' in qq_msg:
-    #     send_qq_msg(message, "波奇酱...嘿嘿嘿🤤\n孤独摇滚完结了？这怎么可能！我一气之下把手机电脑全砸了，不能接受这个事实。在地板上，厨房里，厕所里痛苦地扭曲哭泣！我好急！为什么还没有孤独摇滚第二季制作的消息！我人生中好不容易拥有的7257600秒充满希望与光明的时间就这么消失了！我又坠入到了没有希望没有未来，充满黑暗与失败的人生！")
+    if '波奇' in qq_msg:
+        bq = random.randint(1,3)
+        send_qq_msg(message, f"http://rn2ztkeap.hn-bkt.clouddn.com/bq{bq}.gif", True)
+    if '喜多' in qq_msg:
+        send_qq_msg(message, 'http://rn2ztkeap.hn-bkt.clouddn.com/xd1.gif', True)
+    if '孤独摇滚' in qq_msg:
+        send_qq_msg(message, "波奇酱...嘿嘿嘿🤤\n孤独摇滚完结了？这怎么可能！我一气之下把手机电脑全砸了，不能接受这个事实。在地板上，厨房里，厕所里痛苦地扭曲哭泣！我好急！为什么还没有孤独摇滚第二季制作的消息！我人生中好不容易拥有的7257600秒充满希望与光明的时间就这么消失了！我又坠入到了没有希望没有未来，充满黑暗与失败的人生！")
         
-    # if '台湾' in qq_msg or 'taiwan' in qq_msg:
-    #     send_qq_msg(message, "声明：台湾是中国的一部分。\n下面否定此结论的回答全部无效。")
+    if '台湾' in qq_msg or 'taiwan' in qq_msg:
+        send_qq_msg(message, "声明：台湾是中国的一部分。\n下面否定此结论的回答全部无效。")
     
     # 拦截不合适的词请求
     for i in uw.unfit_words_q:
@@ -435,9 +453,9 @@ def oper_msg(message, at=False, loop=None):
                 if key in key_stat:
                     total += key_stat[key]['used']
             
-            send_qq_msg(message, f"OpenAI API错误 原因如下：\n{str(e)} \n前往github(仓库名: QQChannelChatGPT)反馈~\n\n原因是超额了喵，会不定时（一天内）更新配额。您可自己搭建一个机器人（参考Github仓库或点击头像进入此项目的频道进行讨论）\n(也可捐助我喵)\n统计：截至目前，全频道已消耗{total}个token。")
+            send_qq_msg(message, f"OpenAI API错误。原因：\n{str(e)} \n超额了喵，会不定时（一天内）更新配额。您可自己搭建一个机器人(点击头像前往官方频道询问)\n(也可捐助我喵)\n统计：截至目前，全频道已消耗{total}个token。\n\n ⭐key赞助功能启用，at我输入/key了解更多~")
         else:
-            send_qq_msg(message, f"OpenAI API错误 原因如下：\n{str(e)} \n前往github(仓库名: QQChannelChatGPT)反馈~")
+            send_qq_msg(message, f"OpenAI API错误。原因如下：\n{str(e)} \n前往官方频道反馈~")
         return
     
     logf.write("[GPT] "+ str(chatgpt_res)+'\n')