fix linters

This commit is contained in:
João Moura
2024-12-27 16:31:20 -03:00
parent 9a55b54977
commit b5f2161e34
6 changed files with 17 additions and 18 deletions

View File

@@ -1,7 +1,7 @@
from typing import Optional
from pydantic import BaseModel, Field
from crewai.tools.base_tool import BaseTool
from crewai.utilities import I18N
from pydantic import BaseModel, Field
from typing import Optional
i18n = I18N()

View File

@@ -1,6 +1,6 @@
from typing import Optional
from pydantic import BaseModel, Field
from crewai.tools.agent_tools.base_agent_tools import BaseAgentTool
from pydantic import BaseModel, Field
from typing import Optional
class DelegateWorkToolSchema(BaseModel):

View File

@@ -1,26 +1,25 @@
import ast
import datetime
import time
from difflib import SequenceMatcher
import time
from textwrap import dedent
from typing import Any, List, Union
from crewai.tools.structured_tool import CrewStructuredTool
import crewai.utilities.events as events
from crewai.agents.tools_handler import ToolsHandler
from crewai.task import Task
from crewai.telemetry import Telemetry
from crewai.tools import BaseTool
from crewai.tools.structured_tool import CrewStructuredTool
from crewai.tools.tool_calling import InstructorToolCalling, ToolCalling
from crewai.tools.tool_usage_events import ToolUsageError, ToolUsageFinished
from crewai.utilities import I18N, Converter, ConverterError, Printer
from crewai.utilities import Converter, ConverterError, I18N, Printer
import crewai.utilities.events as events
try:
import agentops # type: ignore
except ImportError:
agentops = None
OPENAI_BIGGER_MODELS = ["gpt-4", "gpt-4o", "o1-preview", "o1-mini"]
OPENAI_BIGGER_MODELS = ["gpt-4", "gpt-4o", "o1-preview", "o1-mini", "o1", "o3", "o3-mini"]
class ToolUsageErrorException(Exception):

View File

@@ -417,9 +417,11 @@ 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."""
query: str = Field(..., description="Query to process")
@@ -470,9 +472,9 @@ def test_crew_with_delegating_agents_should_not_override_task_tools():
@pytest.mark.vcr(filter_headers=["authorization"])
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 typing import Type
class TestToolInput(BaseModel):
"""Input schema for TestTool."""
@@ -584,9 +586,9 @@ def test_task_tools_override_agent_tools_with_allow_delegation():
"""
Test that task tools override agent tools while preserving delegation tools when allow_delegation=True
"""
from typing import Type
from crewai.tools import BaseTool
from pydantic import BaseModel, Field
from typing import Type
class TestToolInput(BaseModel):
query: str = Field(..., description="Query to process")
@@ -2883,10 +2885,10 @@ def test_task_tools_preserve_code_execution_tools():
"""
Test that task tools don't override code execution tools when allow_code_execution=True
"""
from pydantic import BaseModel, Field
from typing import Type
from crewai.tools import BaseTool
from crewai_tools import CodeInterpreterTool
from pydantic import BaseModel, Field
class TestToolInput(BaseModel):
"""Input schema for TestTool."""

View File

@@ -1,10 +1,10 @@
"""Test Knowledge creation and querying functionality."""
import pytest
from pathlib import Path
from typing import List, Union
from unittest.mock import patch
import pytest
from crewai.knowledge.source.crew_docling_source import CrewDoclingSource
from crewai.knowledge.source.csv_knowledge_source import CSVKnowledgeSource
from crewai.knowledge.source.excel_knowledge_source import ExcelKnowledgeSource

View File

@@ -1,11 +1,9 @@
"""Tests for task guardrails functionality."""
from unittest.mock import Mock
import pytest
from crewai.task import Task
from crewai.tasks.task_output import TaskOutput
from unittest.mock import Mock
def test_task_without_guardrail():
"""Test that tasks work normally without guardrails (backward compatibility)."""