SergeyO7 commited on
Commit
95c00fb
·
verified ·
1 Parent(s): ea87167

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +9 -0
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