| import os | |
| import json | |
| SCENARIOS = { | |
| 'sword': 'prompts/system_sword.txt', | |
| 'potion': 'prompts/system_potion.txt', | |
| 'key': 'prompts/system_key.txt' | |
| } | |
| def load_system_prompt(item_id): | |
| base_path = 'prompts/system_base.txt' | |
| item_path = SCENARIOS.get(item_id) | |
| base_prompt = "" | |
| if os.path.exists(base_path): | |
| with open(base_path, 'r') as f: | |
| base_prompt = f.read() | |
| item_prompt = "" | |
| if item_path and os.path.exists(item_path): | |
| with open(item_path, 'r') as f: | |
| item_prompt = f.read() | |
| return f"{base_prompt}\n\n{item_prompt}" | |
| def log_trace(entry): | |
| os.makedirs('data', exist_ok=True) | |
| with open('data/trace.jsonl', 'a') as f: | |
| f.write(json.dumps(entry) + '\n') |