refactor: 迁移旧的处理模块

2026-07-22 04:16:07 +00:00 · 2023-03-27 13:09:40 +00:00
parent 564c5d937d
commit b799de7995
8 changed files with 249 additions and 0 deletions
@@ -0,0 +1,31 @@
+from ..mgr import AbstractCommand, Context
+
+import pkg.openai.session
+import pkg.utils.context
+
+class ResetCommand(AbstractCommand):
+    name = 'reset'
+    description = '重置当前会话'
+    usage = 'reset'
+    aliases = ['重置', '重置会话']
+    privilege = 0
+
+    @classmethod
+    def process(cls, ctx: Context) -> tuple[bool, list, str]:
+        params = ctx.params
+        session_name = ctx.session_name
+        
+        reply = ""
+
+        if len(params) == 0:
+            pkg.openai.session.get_session(session_name).reset(explicit=True)
+            reply = ["[bot]会话已重置"]
+        else:
+            try:
+                import pkg.openai.dprompt as dprompt
+                pkg.openai.session.get_session(session_name).reset(explicit=True, use_prompt=params[0])
+                reply = ["[bot]会话已重置，使用场景预设:{}".format(dprompt.mode_inst().get_full_name(params[0]))]
+            except Exception as e:
+                reply = ["[bot]会话重置失败：{}".format(e)]
+        
+        return True, reply, ''