feat: 支持多场景预设切换

pkg/openai/dprompt.py

@@ -0,0 +1,51 @@
+# 多情景预设值管理
+
+__current__ = "default"
+
+def get_prompt_dict() -> dict:
+    """获取预设值字典"""
+    import config
+    default_prompt = config.default_prompt
+    if type(default_prompt) == str:
+        return {"default": default_prompt}
+    elif type(default_prompt) == dict:
+        return default_prompt
+    else:
+        raise TypeError("default_prompt must be str or dict")
+
+
+def set_current(name):
+    global __current__
+    for key in get_prompt_dict():
+        if key.lower().startswith(name.lower()):
+            __current__ = key
+            return
+    raise KeyError("未找到情景预设: " + name)
+
+
+def get_current():
+    global __current__
+    return __current__
+
+
+def set_to_default():
+    global __current__
+    default_dict = get_prompt_dict()
+
+    if "default" in default_dict:
+        __current__ = "default"
+    else:
+        __current__ = list(default_dict.keys())[0]
+
+
+def get_prompt(name: str = None) -> str:
+    """获取预设值"""
+    if name is None:
+        name = get_current()
+
+    default_dict = get_prompt_dict()
+
+    for key in default_dict:
+        if key.lower().startswith(name.lower()):
+            return default_dict[key]
+    raise KeyError("未找到情景预设: " + name)

pkg/openai/session.py

@@ -89,13 +89,21 @@ class Session:
             logging.debug('{},lock release successfully,{}'.format(self.name, self.response_lock))
 
     # 从配置文件获取会话预设信息
-    def get_default_prompt(self):
+    def get_default_prompt(self, use_default: str=None):
         config = pkg.utils.context.get_config()
+
+        import pkg.openai.dprompt as dprompt
+
+        if use_default is None:
+            current_default_prompt = dprompt.get_prompt(dprompt.get_current())
+        else:
+            current_default_prompt = dprompt.get_prompt(use_default)
+
         user_name = config.user_name if hasattr(config, 'user_name') and config.user_name != '' else 'You'
         bot_name = config.bot_name if hasattr(config, 'bot_name') and config.bot_name != '' else 'Bot'
 
-        return (user_name + ":{}\n".format(config.default_prompt) + bot_name + ":好的\n") \
-            if hasattr(config, 'default_prompt') and config.default_prompt != '' else ''
+        return (user_name + ":{}\n".format(current_default_prompt) + bot_name + ":好的\n") \
+            if current_default_prompt != '' else ''
 
     def __init__(self, name: str):
         self.name = name
@@ -242,8 +250,8 @@ class Session:
                                     self.prompt)
 
     # 重置session
-    def reset(self, explicit: bool = False, expired: bool = False, schedule_new: bool = True):
-        if self.prompt != self.get_default_prompt():
+    def reset(self, explicit: bool = False, expired: bool = False, schedule_new: bool = True, use_prompt: str = None):
+        if not self.prompt.endswith(':好的\n'):
             self.persistence()
             if explicit:
                 # 触发插件事件
@@ -259,7 +267,7 @@ class Session:
 
             if expired:
                 pkg.utils.context.get_database_manager().set_session_expired(self.name, self.create_timestamp)
-        self.prompt = self.get_default_prompt()
+        self.prompt = self.get_default_prompt(use_prompt)
         self.create_timestamp = int(time.time())
         self.last_interact_timestamp = int(time.time())
         self.just_switched_to_exist_session = False