Spaces:
Sleeping
Sleeping
Update agent.py
Browse files
agent.py
CHANGED
|
@@ -342,6 +342,15 @@ class MagAgent:
|
|
| 342 |
max_tokens=8192
|
| 343 |
)
|
| 344 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 345 |
self.tools = [
|
| 346 |
UniversalLoader(),
|
| 347 |
CrossVerifiedSearch(), # Replaces individual search tools
|
|
|
|
| 342 |
max_tokens=8192
|
| 343 |
)
|
| 344 |
|
| 345 |
+
# Add explicit managed_agent template validation
|
| 346 |
+
required_managed_agent_vars = {'task', 'question_analysis', 'subtasks', 'validation_rules'}
|
| 347 |
+
if missing_vars := required_managed_agent_vars - set(self.prompt_templates["managed_agent"]["variables"]):
|
| 348 |
+
raise ValueError(f"Missing required variables in managed_agent template: {missing_vars}")
|
| 349 |
+
|
| 350 |
+
# Verify template contains actual task placeholder
|
| 351 |
+
if "{task}" not in self.prompt_templates["managed_agent"]["template"]:
|
| 352 |
+
raise ConfigurationError("Managed agent template missing {task} placeholder")
|
| 353 |
+
|
| 354 |
self.tools = [
|
| 355 |
UniversalLoader(),
|
| 356 |
CrossVerifiedSearch(), # Replaces individual search tools
|