Spaces:
Sleeping
Sleeping
Update OpenRouter_Agent.py
Browse files- OpenRouter_Agent.py +4 -4
OpenRouter_Agent.py
CHANGED
|
@@ -54,7 +54,7 @@ class MultiAgentSystem:
|
|
| 54 |
)
|
| 55 |
|
| 56 |
self.web_agent = CodeAgent(
|
| 57 |
-
|
| 58 |
tools=[WebSearchTool(), VisitWebpageTool(), WikipediaSearchTool()],
|
| 59 |
name="web_agent",
|
| 60 |
description=(
|
|
@@ -75,7 +75,7 @@ class MultiAgentSystem:
|
|
| 75 |
)
|
| 76 |
|
| 77 |
self.info_agent = CodeAgent(
|
| 78 |
-
|
| 79 |
tools=[PythonInterpreterTool(), image_reasoning_tool],
|
| 80 |
name="info_agent",
|
| 81 |
description=(
|
|
@@ -94,7 +94,7 @@ class MultiAgentSystem:
|
|
| 94 |
)
|
| 95 |
|
| 96 |
self.manager_agent = CodeAgent(
|
| 97 |
-
|
| 98 |
tools=[FinalAnswerTool()],
|
| 99 |
managed_agents=[self.web_agent, self.info_agent],
|
| 100 |
name="manager_agent",
|
|
@@ -117,7 +117,7 @@ class MultiAgentSystem:
|
|
| 117 |
)
|
| 118 |
|
| 119 |
def check_reasoning(self, final_answer, agent_memory):
|
| 120 |
-
|
| 121 |
verification_prompt = (
|
| 122 |
f"Here is a user-given task and the agent steps: {agent_memory.get_succinct_steps()}. "
|
| 123 |
f"The proposed final answer is: {final_answer}. "
|
|
|
|
| 54 |
)
|
| 55 |
|
| 56 |
self.web_agent = CodeAgent(
|
| 57 |
+
model =self.qwen_model,
|
| 58 |
tools=[WebSearchTool(), VisitWebpageTool(), WikipediaSearchTool()],
|
| 59 |
name="web_agent",
|
| 60 |
description=(
|
|
|
|
| 75 |
)
|
| 76 |
|
| 77 |
self.info_agent = CodeAgent(
|
| 78 |
+
model =self.qwen_model,
|
| 79 |
tools=[PythonInterpreterTool(), image_reasoning_tool],
|
| 80 |
name="info_agent",
|
| 81 |
description=(
|
|
|
|
| 94 |
)
|
| 95 |
|
| 96 |
self.manager_agent = CodeAgent(
|
| 97 |
+
model =self.deepseek_model,
|
| 98 |
tools=[FinalAnswerTool()],
|
| 99 |
managed_agents=[self.web_agent, self.info_agent],
|
| 100 |
name="manager_agent",
|
|
|
|
| 117 |
)
|
| 118 |
|
| 119 |
def check_reasoning(self, final_answer, agent_memory):
|
| 120 |
+
model = self.gemini_model
|
| 121 |
verification_prompt = (
|
| 122 |
f"Here is a user-given task and the agent steps: {agent_memory.get_succinct_steps()}. "
|
| 123 |
f"The proposed final answer is: {final_answer}. "
|