diff --git a/tests/test_target_agents_delegation.py b/tests/test_target_agents_delegation.py index acb64bc48..c0e03d4a5 100644 --- a/tests/test_target_agents_delegation.py +++ b/tests/test_target_agents_delegation.py @@ -1,10 +1,6 @@ """Test target_agents delegation functionality.""" -import pytest from crewai.agent import Agent -from crewai.crew import Crew -from crewai.task import Task -from crewai.tools.agent_tools.agent_tools import AgentTools def test_target_agents_filters_delegation_tools(): """Test that target_agents properly filters available agents for delegation.""" diff --git a/tests/tools/agent_tools/agent_tools_test.py b/tests/tools/agent_tools/agent_tools_test.py index 2e8f3315f..3391bc951 100644 --- a/tests/tools/agent_tools/agent_tools_test.py +++ b/tests/tools/agent_tools/agent_tools_test.py @@ -134,7 +134,8 @@ def test_ask_question_to_wrong_agent(): def test_target_agents_delegation_filtering(): """Test that target_agents properly filters delegation targets.""" - researcher = Agent( + # that an agent with target_agents=["writer"] would only delegate to writer + Agent( role="researcher", goal="make the best research and analysis on content about AI and AI agents", backstory="You're an expert researcher, specialized in technology",