OliveSensorAPI/scripts/qa_generation/util/prompt_loader.py

15 lines
412 B
Python
Raw Normal View History

2024-03-07 17:56:07 +08:00
from config.config import system_prompt_file_path
2024-03-16 13:12:15 +08:00
from config.config import wash_prompt_file_path
2024-03-07 17:56:07 +08:00
def load_system_prompt() -> str:
with open(system_prompt_file_path, 'r', encoding='utf-8') as f:
system_prompt = f.read()
return system_prompt
2024-03-16 13:12:15 +08:00
def load_wash_prompt() -> str:
with open(wash_prompt_file_path, 'r', encoding='utf-8') as f:
wash_prompt = f.read()
return wash_prompt