KeyArgo
/
smolagents
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Finalize multiagent tests

This commit is contained in:
Aymeric 2025-01-06 19:33:50 +01:00
parent ff8e20d93f
commit e8617b3037
1 changed files with 14 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/test_agents.py
View File

@ -398,7 +398,7 @@ class AgentTests(unittest.TestCase):
assert "Report on the current US president" in str(messages)
return (
"final_answer",
{"prompt": "Final report."},
"Final report.",
"call_0",
)
manager_model = FakeModelMultiagentsManagerAgent()
@ -409,7 +409,7 @@ class AgentTests(unittest.TestCase):
):
return (
"final_answer",
{"prompt": "Report on the current US president"},
{"report": "Report on the current US president"},
"call_0",
)
managed_model = FakeModelMultiagentsManagedAgent()