feat: 使用模板储存默认的json格式的情景预设

This commit is contained in:
Rock Chin
2023-03-10 23:26:36 +08:00
parent 4269c7927e
commit 54cc75506f
4 changed files with 19 additions and 15 deletions
+2
View File
@@ -12,3 +12,5 @@ logs/
sensitive.json sensitive.json
temp/ temp/
current_tag current_tag
scenario/
!scenario/default-template.json
+4
View File
@@ -332,6 +332,10 @@ if __name__ == '__main__':
if not os.path.exists("sensitive.json"): if not os.path.exists("sensitive.json"):
shutil.copy("sensitive-template.json", "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目录 # 检查temp目录
if not os.path.exists("temp/"): if not os.path.exists("temp/"):
os.mkdir("temp/") os.mkdir("temp/")
+1 -3
View File
@@ -88,10 +88,8 @@ def get_prompt(name: str = None) -> list:
json_content = json.load(f) json_content = json.load(f)
logging.debug('succeed to load json: {}'.format(json_file)) logging.debug('succeed to load json: {}'.format(json_file))
return json_content['prompt'] return json_content['prompt']
except FileNotFoundError: except FileNotFoundError:
raise KeyError("未找到JSON情景预设: " + name)
raise KeyError("未找到Json情景预设: " + name)
# 默认预设方式 # 默认预设方式
elif preset_mode == 'default': elif preset_mode == 'default':