mahadewa commited on
Commit
54adbbb
·
verified ·
1 Parent(s): 77c3307

Update router.py

Browse files
Files changed (1) hide show
  1. router.py +2 -3
router.py CHANGED
@@ -39,7 +39,6 @@ class RouterAgent(BaseAgent):
39
 
40
  Args:
41
  question: The question to process
42
- task_id: The task ID for debugging purposes
43
 
44
  Returns:
45
  The answer from the appropriate specialized agent
@@ -54,12 +53,12 @@ class RouterAgent(BaseAgent):
54
 
55
  # Delegate to the selected specialized agent
56
  if agent_type in self.agents:
57
- return self.agents[agent_type](question, task_id, **kwargs)
58
 
59
  # Fallback to research agent if available
60
  if "research_question" in self.agents:
61
  print(f"Agent '{agent_type}' not found. Falling back to research_question agent.")
62
- return self.agents["research_question"](question, task_id, **kwargs)
63
 
64
  # Ultimate fallback
65
  return f"No appropriate agent found for question type: {agent_type}"
 
39
 
40
  Args:
41
  question: The question to process
 
42
 
43
  Returns:
44
  The answer from the appropriate specialized agent
 
53
 
54
  # Delegate to the selected specialized agent
55
  if agent_type in self.agents:
56
+ return self.agents[agent_type](question, **kwargs)
57
 
58
  # Fallback to research agent if available
59
  if "research_question" in self.agents:
60
  print(f"Agent '{agent_type}' not found. Falling back to research_question agent.")
61
+ return self.agents["research_question"](question, **kwargs)
62
 
63
  # Ultimate fallback
64
  return f"No appropriate agent found for question type: {agent_type}"