mirror of
https://github.com/RockChinQ/QChatGPT.git
synced 2024-11-16 19:57:04 +08:00
75 lines
1.8 KiB
Python
75 lines
1.8 KiB
Python
# 多情景预设值管理
|
|
|
|
__current__ = "default"
|
|
|
|
__prompts_from_files__ = {}
|
|
|
|
|
|
def read_prompt_from_file() -> str:
|
|
"""从文件读取预设值"""
|
|
# 读取prompts/目录下的所有文件，以文件名为键，文件内容为值
|
|
# 保存在__prompts_from_files__中
|
|
global __prompts_from_files__
|
|
import os
|
|
|
|
__prompts_from_files__ = {}
|
|
for file in os.listdir("prompts"):
|
|
with open(os.path.join("prompts", file), encoding="utf-8") as f:
|
|
__prompts_from_files__[file] = f.read()
|
|
|
|
|
|
def get_prompt_dict() -> dict:
|
|
"""获取预设值字典"""
|
|
import config
|
|
default_prompt = config.default_prompt
|
|
if type(default_prompt) == str:
|
|
default_prompt = {"default": default_prompt}
|
|
elif type(default_prompt) == dict:
|
|
pass
|
|
else:
|
|
raise TypeError("default_prompt must be str or dict")
|
|
|
|
# 将文件中的预设值合并到default_prompt中
|
|
for key in __prompts_from_files__:
|
|
default_prompt[key] = __prompts_from_files__[key]
|
|
|
|
return default_prompt
|
|
|
|
|
|
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)
|