fix: full-scenario 命名和目录名错误问题 (#731)

This commit is contained in:
RockChinQ
2024-03-18 21:05:54 +08:00
parent 32bd194bfc
commit c6347bea45
+3 -3
View File
@@ -8,15 +8,15 @@ from .. import entities
from ....provider import entities as llm_entities from ....provider import entities as llm_entities
@loader.loader_class("full_scenario") @loader.loader_class("full-scenario")
class ScenarioPromptLoader(loader.PromptLoader): class ScenarioPromptLoader(loader.PromptLoader):
"""加载scenario目录下的json""" """加载scenario目录下的json"""
async def load(self): async def load(self):
"""加载Prompt """加载Prompt
""" """
for file in os.listdir("data/scenarios"): for file in os.listdir("data/scenario"):
with open("data/scenarios/{}".format(file), "r", encoding="utf-8") as f: with open("data/scenario/{}".format(file), "r", encoding="utf-8") as f:
file_str = f.read() file_str = f.read()
file_name = file.split(".")[0] file_name = file.split(".")[0]
file_json = json.loads(file_str) file_json = json.loads(file_str)