Spaces:
Running
Running
Update agent.py
Browse files
agent.py
CHANGED
@@ -351,6 +351,14 @@ class MagAgent:
|
|
351 |
SpeechToTextTool(),
|
352 |
]
|
353 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
354 |
try:
|
355 |
with open("prompts.yaml") as f:
|
356 |
self.prompt_templates = yaml.safe_load(f)
|
|
|
351 |
SpeechToTextTool(),
|
352 |
]
|
353 |
|
354 |
+
# Load prompts with required templates
|
355 |
+
default_prompts = {
|
356 |
+
"system_prompt": "Default system instructions...",
|
357 |
+
"managed_agent": "Default subtask template...",
|
358 |
+
"planning": "Default planning template...",
|
359 |
+
"final_answer": "Default answer format: {answer}"
|
360 |
+
}
|
361 |
+
|
362 |
try:
|
363 |
with open("prompts.yaml") as f:
|
364 |
self.prompt_templates = yaml.safe_load(f)
|