diff --git a/src/crewai/tools/agent_tools/base_agent_tools.py b/src/crewai/tools/agent_tools/base_agent_tools.py index d0317d088..31289de34 100644 --- a/src/crewai/tools/agent_tools/base_agent_tools.py +++ b/src/crewai/tools/agent_tools/base_agent_tools.py @@ -1,4 +1,3 @@ -import logging from typing import Optional, Union from pydantic import Field @@ -7,7 +6,7 @@ from crewai.agents.agent_builder.base_agent import BaseAgent from crewai.task import Task from crewai.tools.base_tool import BaseTool from crewai.utilities import I18N - +import logging logger = logging.getLogger(__name__) diff --git a/tests/test_manager_llm_delegation.py b/tests/test_manager_llm_delegation.py index 8ad2889ee..c868e1785 100644 --- a/tests/test_manager_llm_delegation.py +++ b/tests/test_manager_llm_delegation.py @@ -1,7 +1,8 @@ +from unittest.mock import MagicMock + import pytest from crewai import Agent, Task from crewai.tools.agent_tools.base_agent_tools import BaseAgentTool -from unittest.mock import MagicMock class TestAgentTool(BaseAgentTool): """Concrete implementation of BaseAgentTool for testing."""