feat: 支持设置默认人格

model/command/openai_official.py

@@ -189,7 +189,7 @@ class CommandOpenAIOfficial(Command):
     def unset(self, session_id: str):
         if self.provider is None:
             return False, "未启动OpenAI ChatGPT语言模型.", "unset"
-        self.provider.now_personality = {}
+        self.provider.curr_personality = {}
         self.provider.forget(session_id)
         return True, "已清除人格并重置历史记录。", "unset"
 
@@ -200,7 +200,7 @@ class CommandOpenAIOfficial(Command):
         if len(l) == 1:
             return True, f"【人格文本由PlexPt开源项目awesome-chatgpt-pr \
         ompts-zh提供】\n设置人格: \n/set 人格名。例如/set 编剧\n人格列表: /set list\n人格详细信息: \
-        /set view 人格名\n自定义人格: /set 人格文本\n重置会话(清除人格): /reset\n重置会话(保留人格): /reset p\n【当前人格】: {str(self.provider.now_personality)}", "set"
+        /set view 人格名\n自定义人格: /set 人格文本\n重置会话(清除人格): /reset\n重置会话(保留人格): /reset p\n【当前人格】: {str(self.provider.curr_personality)}", "set"
         elif l[1] == "list":
             msg = "人格列表：\n"
             for key in personalities.keys():
@@ -221,7 +221,7 @@ class CommandOpenAIOfficial(Command):
         else:
             ps = l[1].strip()
             if ps in personalities:
-                self.provider.now_personality = {
+                self.provider.curr_personality = {
                     'name': ps,
                     'prompt': personalities[ps]
                 }
@@ -243,7 +243,7 @@ class CommandOpenAIOfficial(Command):
                 self.personality_str = message
                 return True, f"人格{ps}已设置。", "set"
             else:
-                self.provider.now_personality = {
+                self.provider.curr_personality = {
                     'name': '自定义人格',
                     'prompt': ps
                 }

model/provider/openai_official.py

@@ -83,7 +83,7 @@ class ProviderOpenAIOfficial(Provider):
         threading.Thread(target=self.dump_history, daemon=True).start()
 
         # 人格
-        self.now_personality = {}
+        self.curr_personality = {}
 
 
     # 转储历史记录
@@ -109,11 +109,30 @@ class ProviderOpenAIOfficial(Provider):
             # 每隔10分钟转储一次
             time.sleep(10*self.history_dump_interval)
 
+    def personality_set(self, default_personality: dict, session_id: str):
+        self.curr_personality = default_personality
+        new_record = {
+            "user": {
+                "role": "user",
+                "content": default_personality['prompt'],
+            },
+            "AI": {
+                "role": "assistant",
+                "content": "好的，接下来我会扮演这个角色。"
+            },
+            'type': "personality",
+            'usage_tokens': 0,
+            'single-tokens': 0
+        }
+        self.session_dict[session_id].append(new_record)
+
+
     def text_chat(self, prompt, 
                   session_id = None, 
                   image_url = None, 
                   function_call=None,
-                  extra_conf: dict = None):
+                  extra_conf: dict = None,
+                  default_personality: dict = None):
         if session_id is None:
             session_id = "unknown"
             if "unknown" in self.session_dict:
@@ -136,6 +155,12 @@ class ProviderOpenAIOfficial(Provider):
                 f.flush()
                 f.close()
         
+        if len(self.session_dict[session_id]) == 0:
+            # 设置默认人格
+            if default_personality is not None:
+                self.personality_set(default_personality, session_id)
+
+
         # 使用 tictoken 截断消息
         _encoded_prompt = self.enc.encode(prompt)
         if self.openai_model_configs['max_tokens'] < len(_encoded_prompt):

model/provider/provider.py

@@ -5,7 +5,7 @@ class Provider:
         pass
     
     @abc.abstractmethod
-    def text_chat(self, prompt, session_id, image_url: None, function_call: None):
+    def text_chat(self, prompt, session_id, image_url: None, function_call: None, extra_conf: dict = None, default_personality: dict = None) -> str:
         pass
 
     @abc.abstractmethod

model/provider/rev_chatgpt.py

@@ -101,7 +101,12 @@ class ProviderRevChatGPT(Provider):
         # print("[RevChatGPT] "+str(resp))
         return resp
 
-    def text_chat(self, prompt, session_id = None, image_url = None, function_call=None) -> str:
+    def text_chat(self, prompt,
+                   session_id = None,
+                   image_url = None,
+                   function_call=None,
+                   extra_conf: dict = None,
+                   default_personality: dict = None) -> str:
 
         # 选择一个人少的账号。
         selected_revstat = None