fixing linter

src/crewai/memory/storage/mem0_storage.py

@@ -1,9 +1,8 @@
 import os
 from typing import Any, Dict, List
 
-from mem0 import MemoryClient
-
 from crewai.memory.storage.interface import Storage
+from mem0 import MemoryClient
 
 
 class Mem0Storage(Storage):

src/crewai/memory/storage/rag_storage.py

@@ -7,7 +7,6 @@ import uuid
 from typing import Any, Dict, List, Optional
 
 from chromadb.api import ClientAPI
-
 from crewai.memory.storage.base_rag_storage import BaseRAGStorage
 from crewai.utilities import EmbeddingConfigurator
 from crewai.utilities.constants import MAX_FILE_NAME_LENGTH

src/crewai/tools/agent_tools/add_image_tool.py

@@ -1,5 +1,5 @@
-from pydantic import BaseModel, Field
 from crewai.tools.base_tool import BaseTool
+from pydantic import BaseModel, Field
 
 
 class AddImageToolSchema(BaseModel):

src/crewai/tools/agent_tools/ask_question_tool.py

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

src/crewai/tools/agent_tools/base_agent_tools.py

@@ -1,11 +1,10 @@
 from typing import Optional, Union
 
-from pydantic import Field
-
 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
+from pydantic import Field
 
 
 class BaseAgentTool(BaseTool):

src/crewai/tools/agent_tools/delegate_work_tool.py

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

src/crewai/tools/tool_usage.py

@@ -5,12 +5,12 @@ from difflib import SequenceMatcher
 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

src/crewai/utilities/evaluators/crew_evaluator_handler.py

@@ -1,14 +1,13 @@
 from collections import defaultdict
 
-from pydantic import BaseModel, Field
-from rich.box import HEAVY_EDGE
-from rich.console import Console
-from rich.table import Table
-
 from crewai.agent import Agent
 from crewai.task import Task
 from crewai.tasks.task_output import TaskOutput
 from crewai.telemetry import Telemetry
+from pydantic import BaseModel, Field
+from rich.box import HEAVY_EDGE
+from rich.console import Console
+from rich.table import Table
 
 
 class TaskEvaluationPydanticOutput(BaseModel):

src/crewai/utilities/evaluators/task_evaluator.py

@@ -1,9 +1,8 @@
 from typing import List
 
-from pydantic import BaseModel, Field
-
 from crewai.utilities import Converter
 from crewai.utilities.pydantic_schema_parser import PydanticSchemaParser
+from pydantic import BaseModel, Field
 
 agentops = None
 try:

tests/agent_test.py

@@ -5,7 +5,6 @@ from unittest import mock
 from unittest.mock import patch
 
 import pytest
-
 from crewai import Agent, Crew, Task
 from crewai.agents.cache import CacheHandler
 from crewai.agents.crew_agent_executor import CrewAgentExecutor
@@ -116,9 +115,8 @@ def test_custom_llm_temperature_preservation():
 
 @pytest.mark.vcr(filter_headers=["authorization"])
 def test_agent_execute_task():
-    from langchain_openai import ChatOpenAI
-
     from crewai import Task
+    from langchain_openai import ChatOpenAI
 
     agent = Agent(
         role="Math Tutor",
@@ -862,7 +860,6 @@ def test_agent_function_calling_llm():
     from unittest.mock import patch
 
     import instructor
-
     from crewai.tools.tool_usage import ToolUsage
 
     with (

tests/agents/agent_builder/base_agent_test.py

@@ -1,10 +1,9 @@
 import hashlib
 from typing import Any, List, Optional
 
-from pydantic import BaseModel
-
 from crewai.agents.agent_builder.base_agent import BaseAgent
 from crewai.tools.base_tool import BaseTool
+from pydantic import BaseModel
 
 
 class TestAgent(BaseAgent):

tests/agents/test_crew_agent_parser.py

@@ -1,5 +1,4 @@
 import pytest
-
 from crewai.agents.crew_agent_executor import (
     AgentAction,
     AgentFinish,

tests/cli/authentication/test_auth_main.py

@@ -2,7 +2,6 @@ import unittest
 from unittest.mock import MagicMock, patch
 
 import requests
-
 from crewai.cli.authentication.main import AuthenticationCommand
 
 

tests/cli/authentication/test_utils.py

@@ -3,9 +3,8 @@ import unittest
 from datetime import datetime, timedelta
 from unittest.mock import MagicMock, patch
 
-from cryptography.fernet import Fernet
-
 from crewai.cli.authentication.utils import TokenManager, validate_token
+from cryptography.fernet import Fernet
 
 
 class TestValidateToken(unittest.TestCase):

tests/cli/cli_test.py

@@ -3,7 +3,6 @@ from unittest import mock
 
 import pytest
 from click.testing import CliRunner
-
 from crewai.cli.cli import (
     deploy_create,
     deploy_list,

tests/cli/deploy/test_deploy_main.py

@@ -5,10 +5,9 @@ from unittest.mock import MagicMock, Mock, patch
 
 import pytest
 import requests
-from requests.exceptions import JSONDecodeError
-
 from crewai.cli.deploy.main import DeployCommand
 from crewai.cli.utils import parse_toml
+from requests.exceptions import JSONDecodeError
 
 
 class TestDeployCommand(unittest.TestCase):

tests/cli/test_crew_test.py

@@ -2,7 +2,6 @@ import subprocess
 from unittest import mock
 
 import pytest
-
 from crewai.cli import evaluate_crew
 
 

tests/cli/test_git.py

@@ -1,5 +1,4 @@
 import pytest
-
 from crewai.cli.git import Repository
 
 

tests/cli/test_utils.py

@@ -3,7 +3,6 @@ import shutil
 import tempfile
 
 import pytest
-
 from crewai.cli import utils
 
 

tests/cli/tools/test_main.py

@@ -7,9 +7,8 @@ from io import StringIO
 from unittest import mock
 from unittest.mock import MagicMock, patch
 
-from pytest import raises
-
 from crewai.cli.tools.main import ToolCommand
+from pytest import raises
 
 
 @contextmanager

tests/crew_test.py

@@ -9,7 +9,6 @@ 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
@@ -417,6 +416,7 @@ 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
 
@@ -471,6 +471,7 @@ 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
 
@@ -527,6 +528,7 @@ def test_crew_with_delegating_agents_should_not_override_agent_tools():
 @pytest.mark.vcr(filter_headers=["authorization"])
 def test_task_tools_override_agent_tools():
     from typing import Type
+
     from crewai.tools import BaseTool
     from pydantic import BaseModel, Field
 
@@ -585,6 +587,7 @@ 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
 
@@ -2884,6 +2887,7 @@ def test_task_tools_preserve_code_execution_tools():
     Test that task tools don't override code execution tools when allow_code_execution=True
     """
     from typing import Type
+
     from crewai.tools import BaseTool
     from crewai_tools import CodeInterpreterTool
     from pydantic import BaseModel, Field

tests/flow_test.py

@@ -3,7 +3,6 @@
 import asyncio
 
 import pytest
-
 from crewai.flow.flow import Flow, and_, listen, or_, router, start
 
 

tests/llm_test.py

@@ -1,5 +1,4 @@
 import pytest
-
 from crewai.agents.agent_builder.utilities.base_token_process import TokenProcess
 from crewai.llm import LLM
 from crewai.utilities.token_counter_callback import TokenCalcHandler

tests/memory/long_term_memory_test.py

@@ -1,5 +1,4 @@
 import pytest
-
 from crewai.memory.long_term.long_term_memory import LongTermMemory
 from crewai.memory.long_term.long_term_memory_item import LongTermMemoryItem
 

tests/memory/short_term_memory_test.py

@@ -1,7 +1,6 @@
 from unittest.mock import patch
 
 import pytest
-
 from crewai.agent import Agent
 from crewai.crew import Crew
 from crewai.memory.short_term.short_term_memory import ShortTermMemory

tests/project_test.py

@@ -1,5 +1,4 @@
 import pytest
-
 from crewai.agent import Agent
 from crewai.crew import Crew
 from crewai.project import CrewBase, after_kickoff, agent, before_kickoff, crew, task

tests/task_test.py

@@ -6,13 +6,12 @@ import os
 from unittest.mock import MagicMock, patch
 
 import pytest
-from pydantic import BaseModel
-from pydantic_core import ValidationError
-
 from crewai import Agent, Crew, Process, Task
 from crewai.tasks.conditional_task import ConditionalTask
 from crewai.tasks.task_output import TaskOutput
 from crewai.utilities.converter import Converter
+from pydantic import BaseModel
+from pydantic_core import ValidationError
 
 
 def test_task_tool_reflect_agent_tools():

tests/tools/agent_tools/agent_tools_test.py

@@ -1,7 +1,6 @@
 """Test Agent creation and execution basic functionality."""
 
 import pytest
-
 from crewai.agent import Agent
 from crewai.tools.agent_tools.agent_tools import AgentTools
 

tests/tools/test_structured_tool.py

@@ -1,9 +1,8 @@
 from typing import Optional
 
 import pytest
-from pydantic import BaseModel, Field
-
 from crewai.tools.structured_tool import CrewStructuredTool
+from pydantic import BaseModel, Field
 
 
 # Test fixtures

tests/tools/test_tool_usage.py

@@ -3,11 +3,10 @@ import random
 from unittest.mock import MagicMock
 
 import pytest
-from pydantic import BaseModel, Field
-
 from crewai import Agent, Task
 from crewai.tools import BaseTool
 from crewai.tools.tool_usage import ToolUsage
+from pydantic import BaseModel, Field
 
 
 class RandomNumberToolInput(BaseModel):

tests/utilities/evaluators/test_crew_evaluator_handler.py

@@ -1,7 +1,6 @@
 from unittest import mock
 
 import pytest
-
 from crewai.agent import Agent
 from crewai.crew import Crew
 from crewai.task import Task

tests/utilities/test_converter.py

@@ -3,8 +3,6 @@ from typing import Dict, List, Optional
 from unittest.mock import MagicMock, Mock, patch
 
 import pytest
-from pydantic import BaseModel
-
 from crewai.llm import LLM
 from crewai.utilities.converter import (
     Converter,
@@ -18,6 +16,7 @@ from crewai.utilities.converter import (
     validate_model,
 )
 from crewai.utilities.pydantic_schema_parser import PydanticSchemaParser
+from pydantic import BaseModel
 
 
 # Sample Pydantic models for testing

tests/utilities/test_file_handler.py

@@ -2,7 +2,6 @@ import os
 import unittest
 
 import pytest
-
 from crewai.utilities.file_handler import PickleHandler
 
 

tests/utilities/test_i18n.py

@@ -1,5 +1,4 @@
 import pytest
-
 from crewai.utilities.i18n import I18N
 
 

tests/utilities/test_planning_handler.py

@@ -1,7 +1,6 @@
 from unittest.mock import patch
 
 import pytest
-
 from crewai.agent import Agent
 from crewai.task import Task
 from crewai.tasks.task_output import TaskOutput