Merge pull request #1760 from xiexin12138/feature-优化智谱-AI-的命令操作

add feature 优化智谱 AI 的命令操作，使其支持重置会话
zhayujie · Mar 25, 2024 · 90581c8 · 90581c8
2 parents cac7a62 + a25d7ea
commit 90581c8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/plugins/godcmd/godcmd.py b/plugins/godcmd/godcmd.py
@@ -313,7 +313,7 @@ def on_handle_context(self, e_context: EventContext):
                     except Exception as e:
                         ok, result = False, "你没有设置私有GPT模型"
                 elif cmd == "reset":
-                    if bottype in [const.OPEN_AI, const.CHATGPT, const.CHATGPTONAZURE, const.LINKAI, const.BAIDU, const.XUNFEI, const.QWEN, const.GEMINI]:
+                    if bottype in [const.OPEN_AI, const.CHATGPT, const.CHATGPTONAZURE, const.LINKAI, const.BAIDU, const.XUNFEI, const.QWEN, const.GEMINI, const.ZHIPU_AI]:
                         bot.sessions.clear_session(session_id)
                         if Bridge().chat_bots.get(bottype):
                             Bridge().chat_bots.get(bottype).sessions.clear_session(session_id)
@@ -339,7 +339,7 @@ def on_handle_context(self, e_context: EventContext):
                             ok, result = True, "配置已重载"
                         elif cmd == "resetall":
                             if bottype in [const.OPEN_AI, const.CHATGPT, const.CHATGPTONAZURE, const.LINKAI,
-                                           const.BAIDU, const.XUNFEI, const.QWEN, const.GEMINI]:
+                                           const.BAIDU, const.XUNFEI, const.QWEN, const.GEMINI, const.ZHIPU_AI]:
                                 channel.cancel_all_session()
                                 bot.sessions.clear_all_session()
                                 ok, result = True, "重置所有会话成功"