From 8953af61337eecb4f01dec3b6fcb9ed37e221a97 Mon Sep 17 00:00:00 2001 From: Brandon Hancock Date: Fri, 14 Feb 2025 15:24:29 -0500 Subject: [PATCH] Fix ruff issues --- tests/crew_test.py | 18 ++++++++++++------ 1 file changed, 12 insertions(+), 6 deletions(-) diff --git a/tests/crew_test.py b/tests/crew_test.py index 8b74c75b5..b9a8b560d 100644 --- a/tests/crew_test.py +++ b/tests/crew_test.py @@ -6,9 +6,9 @@ from concurrent.futures import Future from unittest import mock from unittest.mock import MagicMock, patch -import instructor import pydantic_core import pytest + from crewai.agent import Agent from crewai.agents.cache import CacheHandler from crewai.crew import Crew @@ -538,11 +538,13 @@ def test_crew_with_delegating_agents(): @pytest.mark.vcr(filter_headers=["authorization"]) def test_crew_with_delegating_agents_should_not_override_task_tools(): + from typing import Type - from crewai.tools import BaseTool from pydantic import BaseModel, Field + from crewai.tools import BaseTool + class TestToolInput(BaseModel): """Input schema for TestTool.""" @@ -601,9 +603,10 @@ def test_crew_with_delegating_agents_should_not_override_task_tools(): def test_crew_with_delegating_agents_should_not_override_agent_tools(): from typing import Type - from crewai.tools import BaseTool from pydantic import BaseModel, Field + from crewai.tools import BaseTool + class TestToolInput(BaseModel): """Input schema for TestTool.""" @@ -664,9 +667,10 @@ def test_crew_with_delegating_agents_should_not_override_agent_tools(): def test_task_tools_override_agent_tools(): from typing import Type - from crewai.tools import BaseTool from pydantic import BaseModel, Field + from crewai.tools import BaseTool + class TestToolInput(BaseModel): """Input schema for TestTool.""" @@ -721,9 +725,10 @@ def test_task_tools_override_agent_tools_with_allow_delegation(): """ from typing import Type - from crewai.tools import BaseTool from pydantic import BaseModel, Field + from crewai.tools import BaseTool + class TestToolInput(BaseModel): query: str = Field(..., description="Query to process") @@ -3424,10 +3429,11 @@ def test_task_tools_preserve_code_execution_tools(): """ from typing import Type - from crewai.tools import BaseTool from crewai_tools import CodeInterpreterTool from pydantic import BaseModel, Field + from crewai.tools import BaseTool + class TestToolInput(BaseModel): """Input schema for TestTool."""