Spaces:
Sleeping
Sleeping
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)
|