andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-11 00:58:30 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Fix ruff issues

Browse Source
This commit is contained in:
Brandon Hancock
2025-02-14 15:24:29 -05:00
parent 2b438baad4
commit 8953af6133
1 changed files with 12 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
tests/crew_test.py
View File

@@ -6,9 +6,9 @@ from concurrent.futures import Future
from unittest import mock from unittest import mock
from unittest.mock import MagicMock, patch from unittest.mock import MagicMock, patch
import instructor
import pydantic_core import pydantic_core
import pytest import pytest
from crewai.agent import Agent from crewai.agent import Agent
from crewai.agents.cache import CacheHandler from crewai.agents.cache import CacheHandler
from crewai.crew import Crew from crewai.crew import Crew
@@ -538,11 +538,13 @@ def test_crew_with_delegating_agents():
@pytest.mark.vcr(filter_headers=["authorization"]) @pytest.mark.vcr(filter_headers=["authorization"])
def test_crew_with_delegating_agents_should_not_override_task_tools(): def test_crew_with_delegating_agents_should_not_override_task_tools():
from typing import Type from typing import Type
from crewai.tools import BaseTool
from pydantic import BaseModel, Field from pydantic import BaseModel, Field
from crewai.tools import BaseTool
class TestToolInput(BaseModel): class TestToolInput(BaseModel):
"""Input schema for TestTool.""" """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(): def test_crew_with_delegating_agents_should_not_override_agent_tools():
from typing import Type from typing import Type
from crewai.tools import BaseTool
from pydantic import BaseModel, Field from pydantic import BaseModel, Field
from crewai.tools import BaseTool
class TestToolInput(BaseModel): class TestToolInput(BaseModel):
"""Input schema for TestTool.""" """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(): def test_task_tools_override_agent_tools():
from typing import Type from typing import Type
from crewai.tools import BaseTool
from pydantic import BaseModel, Field from pydantic import BaseModel, Field
from crewai.tools import BaseTool
class TestToolInput(BaseModel): class TestToolInput(BaseModel):
"""Input schema for TestTool.""" """Input schema for TestTool."""
@@ -721,9 +725,10 @@ def test_task_tools_override_agent_tools_with_allow_delegation():
""" """
from typing import Type from typing import Type
from crewai.tools import BaseTool
from pydantic import BaseModel, Field from pydantic import BaseModel, Field
from crewai.tools import BaseTool
class TestToolInput(BaseModel): class TestToolInput(BaseModel):
query: str = Field(..., description="Query to process") query: str = Field(..., description="Query to process")
@@ -3424,10 +3429,11 @@ def test_task_tools_preserve_code_execution_tools():
""" """
from typing import Type from typing import Type
from crewai.tools import BaseTool
from crewai_tools import CodeInterpreterTool from crewai_tools import CodeInterpreterTool
from pydantic import BaseModel, Field from pydantic import BaseModel, Field
from crewai.tools import BaseTool
class TestToolInput(BaseModel): class TestToolInput(BaseModel):
"""Input schema for TestTool.""" """Input schema for TestTool."""