refactor: 重构部分代码 #61

2023-04-01 01:02:16 +08:00
parent 1a137a8639
commit e6f6bee7ee
7 changed files with 403 additions and 305 deletions
@@ -1,15 +1,27 @@
 import abc
+import json
+
+import requests
+from model.provider.provider import Provider

 class Command:
-    def __init__(self):
-        pass
+    def __init__(self, provider: Provider):
+        self.provider = Provider

    @abc.abstractmethod
    def check_command(self, message):
-        pass
+        if message.startswith("reset") or message.startswith("重置"):
+            return True, self.reset()
+        elif message.startswith("help") or message.startswith("帮助"):
+            return True, self.help()
+        return False, None

    def reset(self):
-        return False
+        res = self.provider.forget()
+        if res:
+            return res, "重置成功"
+        else:
+            return res, "重置失敗"
    
    def set(self):
        return False
@@ -21,7 +33,16 @@ class Command:
        return False
    
    def help(self):
-        return False
+        ol_version = 'Unknown'
+        try:
+            global version
+            res = requests.get("https://soulter.top/channelbot/update.json")
+            res_obj = json.loads(res.text)
+            ol_version = res_obj['version']
+        except BaseException:
+            pass
+        return True, f"[Github项目名: QQChannelChatGPT，有问题请前往提交issue，欢迎Star此项目~]\n\n当前版本:{version}\n最新版本:{str(ol_version)}\n请及时更新！\n\n指令面板：\n/status 查看机器人key状态\n/count 查看机器人统计信息\n/reset 重置会话\n/his 查看历史记录\n/token 查看会话token数\n/help 查看帮助\n/set 人格指令菜单\n/key 动态添加key"
+
    
    def status(self):
        return False
@@ -0,0 +1,169 @@
+from model.command.command import Command
+from model.provider.provider_openai_official import ProviderOpenAIOfficial
+from cores.qqbot.personality import personalities
+
+class CommandOpenAIOfficial(Command):
+    def __init__(self, provider: ProviderOpenAIOfficial):
+        self.provider = provider
+        
+    def check_command(self, message: str, session_id: str, user_name: str):
+        if message.startswith("reset") or message.startswith("重置"):
+            return True, self.reset(session_id)
+        elif message.startswith("his") or message.startswith("历史"):
+            return True, self.his(message)
+        elif message.startswith("token"):
+            return True, self.token(session_id)
+        elif message.startswith("gpt"):
+            return True, self.gpt()
+        elif message.startswith("status") or message.startswith("状态"):
+            return True, self.status()
+        elif message.startswith("count") or message.startswith("统计"):
+            return True, self.count()
+        elif message.startswith("help") or message.startswith("帮助"):
+            return True, self.help()
+        elif message.startswith("key") or message.startswith("动态添加key"):
+            return True, self.key(message, user_name)
+        elif message.startswith("unset"):
+            return True, self.set(session_id)
+        elif message.startswith("set"):
+            return True, self.set(message, session_id)
+        elif message.startswith("画"):
+            return True, self.draw(message)
+        
+        return False, None
+    
+    def reset(self, session_id: str):
+        self.provider.forget(session_id)
+        return True, "重置成功"
+    
+    def his(self, message: str, session_id: str, name: str):
+        #分页，每页5条
+        msg = ''
+        size_per_page = 3
+        page = 1
+        if message[4:]:
+            page = int(message[4:])
+        # 检查是否有过历史记录
+        if session_id not in self.provider.session_dict:
+            msg = f"{name} 的历史记录为空"
+        l = self.provider.session_dict[session_id]
+        max_page = len(l)//size_per_page + 1 if len(l)%size_per_page != 0 else len(l)//size_per_page
+        p = self.provider.get_prompts_by_cache_list(self.provider.session_dict[session_id], divide=True, paging=True, size=size_per_page, page=page)
+        return True, f"历史记录如下：\n{p}\n第{page}页 | 共{max_page}页\n*输入/his 2跳转到第2页"
+    
+    def token(self, session_id: str):
+        return True, f"会话的token数: {self.provider.get_user_usage_tokens(self.provider.session_dict[session_id])}\n系统最大缓存token数: {self.provider.max_tokens}"
+
+    def gpt(self):
+        return True, f"OpenAI GPT配置:\n {self.provider.chatGPT_configs}"
+    
+    def status(self):
+        chatgpt_cfg_str = ""
+        key_stat = self.provider.get_key_stat()
+        index = 1
+        max = 9000000
+        gg_count = 0
+        total = 0
+        tag = ''
+        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}: {key_stat[key]['used']}/{max} {sponsor}赞助{tag}\n"
+            index += 1
+        return True, f"⭐使用情况({str(gg_count)}个已用):\n{chatgpt_cfg_str}⏰全频道已用{total}tokens"
+
+    def count(self):
+        guild_count, guild_msg_count, guild_direct_msg_count, session_count = self.provider.get_stat()
+        return True, f"当前会话数: {len(self.provider.session_dict)}\n共有频道数: {guild_count} \n共有消息数: {guild_msg_count}\n私信数: {guild_direct_msg_count}\n历史会话数: {session_count}"
+
+    def key(self, message: str, user_name: str):
+        l = message.split(" ")
+        if len(l) == 1:
+            msg = "感谢您赞助key，key为官方API使用，请以以下格式赞助:\n/key xxxxx"
+            return True, msg
+        key = l[1]
+        if self.provider.check_key(key):
+            self.provider.append_key(key, user_name)
+            return True, f"*★,°*:.☆(￣▽￣)/$:*.°★* 。\n该Key被验证为有效。感谢{user_name}赞助~"
+        else:
+            return True, "该Key被验证为无效。也许是输入错误了，或者重试。"
+
+    def unset(self, session_id: str):
+        self.provider.now_personality = {}
+        self.provider.forget(session_id)
+        return True, "已清除人格并重置历史记录。"
+
+    def set(self, message: str, session_id: str):
+        l = message.split(" ")
+        if len(l) == 1:
+            return True, f"【由Github项目QQChannelChatGPT支持】\n\n【人格文本由PlexPt开源项目awesome-chatgpt-prompts-zh提供】\n\n这个是人格设置指令。\n设置人格: \n/set 人格名。例如/set 编剧\n人格列表: /set list\n人格详细信息: /set view 人格名\n自定义人格: /set 人格文本\n清除人格: /unset\n【当前人格】: {np}"
+        elif l[1] == "list":
+            msg = "人格列表：\n"
+            for key in personalities.keys():
+                msg += f"  |-{key}\n"
+            msg += '\n\n*输入/set view 人格名查看人格详细信息'
+            msg += '\n*不定时更新人格库，请及时更新本项目。'
+            return True, msg
+        elif l[1] == "view":
+            if len(l) == 2:
+                return True, "请输入/set view 人格名"
+            ps = l[2].strip()
+            if ps in personalities:
+                msg = f"人格{ps}的详细信息：\n"
+                msg += f"{personalities[ps]}\n"
+            else:
+                msg = f"人格{ps}不存在"
+            return True, msg
+        else:
+            ps = l[1].strip()
+            if ps in personalities:
+                self.provider.now_personality = {
+                    'name': ps,
+                    'prompt': personalities[ps]
+                }
+                self.provider.session_dict[session_id] = []
+                new_record = {
+                    "user": {
+                        "role": "system",
+                        "content": personalities[ps],
+                    },
+                    'usage_tokens': 0,
+                    'single-tokens': 0
+                }
+                self.provider.session_dict[session_id].append(new_record)
+                return True, f"人格{ps}已设置."
+            else:
+                self.provider.now_personality = {
+                    'name': '自定义人格',
+                    'prompt': ps
+                }
+                new_record = {
+                    "user": {
+                        "role": "system",
+                        "content": ps,
+                    },
+                    'usage_tokens': 0,
+                    'single-tokens': 0
+                }
+                self.provider.session_dict[session_id] = []
+                self.provider.session_dict[session_id].append(new_record)
+                return True, f"自定义人格已设置。 \n人格信息: {ps}"
+            
+    def draw(self, message):
+        try:
+            img_url = self.provider.image_chat(message)
+            return True, img_url
+        except Exception as e:
+            if 'exceeded' in str(e):
+                return f"OpenAI API错误。原因：\n{str(e)} \n超额了。可自己搭建一个机器人(Github仓库：QQChannelChatGPT)"
+            return False, f"图片生成失败: {e}"
+        
+    
+
+    
+    
@@ -0,0 +1,16 @@
+from model.command.command import Command
+from model.provider.provider_rev_chatgpt import ProviderRevChatGPT
+
+class CommandRevChatGPT(Command):
+    def __init__(self, provider: ProviderRevChatGPT):
+        super().__init__(provider)
+        
+    def check_command(self, message: str):
+        hit, res = super().check_command(message)
+        if hit:
+            return res
+        
+        return False, None
+
+    
+        
@@ -0,0 +1,14 @@
+from model.command.command import Command
+from model.provider.provider_rev_edgegpt import ProviderRevEdgeGPT
+
+class CommandRevEdgeGPT(Command):
+    def __init__(self, provider: ProviderRevEdgeGPT):
+        super().__init__(provider)
+        
+    def check_command(self, message: str):
+        hit, res = super().check_command(message)
+        if hit:
+            return res
+        
+        return False, None
+