Update router.py
Browse files
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,
|
| 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,
|
| 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}"
|