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