fix bug

SAGIRI-kawaii · Mar 7, 2023 · 3616afe · 3616afe
1 parent e56f194
commit 3616afe
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 8 deletions.
diff --git a/modules/self_contained/av_bt.py b/modules/self_contained/av_bt.py
diff --git a/modules/self_contained/chat_gpt/__init__.py b/modules/self_contained/chat_gpt/__init__.py
@@ -73,7 +73,7 @@ async def chat_gpt(
             "当前内置预设：\n" + "\n".join([f"{i} ({v['name']})：{v['description']}" for i, v in preset_dict.items()])
         )
     if new_thread.matched:
-        _ = await manager.new(group, member, preset.result.display if preset.matched else "")
+        _ = await manager.new(group, member, (preset.result.display.strip() if preset.matched else ""))
     response = await manager.send_message(group, member, content.result.display.strip())
     if text.matched:
         await app.send_group_message(group, MessageChain(response), quote=source)

diff --git a/modules/self_contained/chat_gpt/gpt3_5.py b/modules/self_contained/chat_gpt/gpt3_5.py
@@ -35,10 +35,5 @@ async def send_message(self, message: str) -> str:
         self.message["messages"].append(result)
         return result["content"]
 
-    def reset(self, reserved_preset: bool = True, preset: str = ""):
-        if reserved_preset:
-            self.message["messages"] = self.message["messages"][:1]
-        else:
-            self.message["messages"] = [
-                {"role": "system", "content": preset}
-            ]
+    def reset(self, preset: str = ""):
+        self.message["messages"] = [{"role": "system", "content": preset}] if preset else [self.message["messages"][0]]