mirror of
https://github.com/langbot-app/LangBot.git
synced 2025-11-25 11:29:39 +08:00
52 lines
1.2 KiB
Python
52 lines
1.2 KiB
Python
# 多情景预设值管理
|
|
|
|
__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)
|