mirror of
https://github.com/RockChinQ/QChatGPT.git
synced 2024-11-16 11:42:44 +08:00
feat: 使用模板储存默认的json格式的情景预设
This commit is contained in:
parent
4269c7927e
commit
54cc75506f
4
.gitignore
vendored
4
.gitignore
vendored
|
@ -11,4 +11,6 @@ prompts/
|
|||
logs/
|
||||
sensitive.json
|
||||
temp/
|
||||
current_tag
|
||||
current_tag
|
||||
scenario/
|
||||
!scenario/default-template.json
|
4
main.py
4
main.py
|
@ -332,6 +332,10 @@ if __name__ == '__main__':
|
|||
if not os.path.exists("sensitive.json"):
|
||||
shutil.copy("sensitive-template.json", "sensitive.json")
|
||||
|
||||
# 检查是否有scenario/default.json
|
||||
if not os.path.exists("scenario/default.json"):
|
||||
shutil.copy("scenario/default-template.json", "scenario/default.json")
|
||||
|
||||
# 检查temp目录
|
||||
if not os.path.exists("temp/"):
|
||||
os.mkdir("temp/")
|
||||
|
|
|
@ -88,10 +88,8 @@ def get_prompt(name: str = None) -> list:
|
|||
json_content = json.load(f)
|
||||
logging.debug('succeed to load json: {}'.format(json_file))
|
||||
return json_content['prompt']
|
||||
|
||||
except FileNotFoundError:
|
||||
|
||||
raise KeyError("未找到Json情景预设: " + name)
|
||||
raise KeyError("未找到JSON情景预设: " + name)
|
||||
|
||||
# 默认预设方式
|
||||
elif preset_mode == 'default':
|
||||
|
|
|
@ -1,12 +1,12 @@
|
|||
{
|
||||
"prompt": [
|
||||
{
|
||||
"role": "system",
|
||||
"content": "You are a helpful assistant. 如果我需要帮助,你要说“输入!help获得帮助”"
|
||||
},
|
||||
{
|
||||
"role": "assistant",
|
||||
"content": "好的,我是一个能干的AI助手。 如果你需要帮助,我会说“输入!help获得帮助”"
|
||||
}
|
||||
]
|
||||
{
|
||||
"prompt": [
|
||||
{
|
||||
"role": "system",
|
||||
"content": "You are a helpful assistant. 如果我需要帮助,你要说“输入!help获得帮助”"
|
||||
},
|
||||
{
|
||||
"role": "assistant",
|
||||
"content": "好的,我是一个能干的AI助手。 如果你需要帮助,我会说“输入!help获得帮助”"
|
||||
}
|
||||
]
|
||||
}
|
Loading…
Reference in New Issue
Block a user