From e58b0a8d70464aac71141be887968865c0a374b4 Mon Sep 17 00:00:00 2001 From: Lorenze Jay Date: Wed, 17 Jul 2024 08:20:05 -0700 Subject: [PATCH] fixed test --- tests/crew_test.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/tests/crew_test.py b/tests/crew_test.py index 3c60d9b1f..923b35417 100644 --- a/tests/crew_test.py +++ b/tests/crew_test.py @@ -370,17 +370,20 @@ def test_manager_agent_delegating_to_all_agents(): crew.kickoff() - # Check if the manager agent has the correct tools assert crew.manager_agent is not None assert crew.manager_agent.tools is not None - assert len(crew.manager_agent.tools) == 1 + assert len(crew.manager_agent.tools) == 2 + print( + "crew.manager_agent.tools[0].description", + crew.manager_agent.tools[0].description, + ) assert ( - "Delegate a specific task to one of the following coworkers: Researcher, Writer\n" + "Delegate a specific task to one of the following coworkers: Researcher, Senior Writer\n" in crew.manager_agent.tools[0].description ) assert ( - "Ask a specific question to one of the following coworkers: Researcher, Writer\n" + "Ask a specific question to one of the following coworkers: Researcher, Senior Writer\n" in crew.manager_agent.tools[1].description )