From e8617b30379be726e5877f1ad1eed44e11a78775 Mon Sep 17 00:00:00 2001
From: Aymeric <aymeric.roucher@gmail.com>
Date: Mon, 6 Jan 2025 19:33:50 +0100
Subject: [PATCH] Finalize multiagent tests

---
 tests/test_agents.py | 28 ++++++++++++++--------------
 1 file changed, 14 insertions(+), 14 deletions(-)

diff --git a/tests/test_agents.py b/tests/test_agents.py
index 4a77519..8804042 100644
--- a/tests/test_agents.py
+++ b/tests/test_agents.py
@@ -369,21 +369,21 @@ class AgentTests(unittest.TestCase):
             ):
                 if len(messages) < 3:
                     return """
-        Thought: Let's call our search agent.
-        Code:
-        ```py
-        result = search_agent("Who is the current US president?")
-        ```<end_code>
-        """
+Thought: Let's call our search agent.
+Code:
+```py
+result = search_agent("Who is the current US president?")
+```<end_code>
+"""
                 else:
                     assert "Report on the current US president" in str(messages)
                     return """
-        Thought: Let's return the report.
-        Code:
-        ```py
-        final_answer("Final report.")
-        ```<end_code>
-        """
+Thought: Let's return the report.
+Code:
+```py
+final_answer("Final report.")
+```<end_code>
+"""
 
             def get_tool_call(
                 self, messages, available_tools, stop_sequences=None, grammar=None
@@ -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()