sort imports with isort rules by ruff linter (#1730)

* sort imports

* update

---------

Co-authored-by: Brandon Hancock (bhancock_ai) <109994880+bhancockio@users.noreply.github.com>
Co-authored-by: Eduardo Chiarotti <dudumelgaco@hotmail.com>

src/crewai/agents/parser.py

@@ -1,5 +1,6 @@
 import re
 from typing import Any, Union
+
 from json_repair import repair_json
 
 from crewai.utilities import I18N

src/crewai/cli/authentication/main.py

@@ -5,9 +5,10 @@ from typing import Any, Dict
 import requests
 from rich.console import Console
 
+from crewai.cli.tools.main import ToolCommand
+
 from .constants import AUTH0_AUDIENCE, AUTH0_CLIENT_ID, AUTH0_DOMAIN
 from .utils import TokenManager, validate_token
-from crewai.cli.tools.main import ToolCommand
 
 console = Console()
 
@@ -79,7 +80,9 @@ class AuthenticationCommand:
                         style="yellow",
                     )
 
-                console.print("\n[bold green]Welcome to CrewAI Enterprise![/bold green]\n")
+                console.print(
+                    "\n[bold green]Welcome to CrewAI Enterprise![/bold green]\n"
+                )
                 return
 
             if token_data["error"] not in ("authorization_pending", "slow_down"):

src/crewai/cli/authentication/token.py

@@ -1,10 +1,9 @@
 from .utils import TokenManager
 
+
 def get_auth_token() -> str:
     """Get the authentication token."""
     access_token = TokenManager().get_token()
     if not access_token:
         raise Exception()
     return access_token
-
-

src/crewai/cli/command.py

@@ -1,8 +1,9 @@
 import requests
 from requests.exceptions import JSONDecodeError
 from rich.console import Console
-from crewai.cli.plus_api import PlusAPI
+
 from crewai.cli.authentication.token import get_auth_token
+from crewai.cli.plus_api import PlusAPI
 from crewai.telemetry.telemetry import Telemetry
 
 console = Console()

src/crewai/cli/config.py

@@ -1,13 +1,19 @@
 import json
 from pathlib import Path
-from pydantic import BaseModel, Field
 from typing import Optional
 
+from pydantic import BaseModel, Field
+
 DEFAULT_CONFIG_PATH = Path.home() / ".config" / "crewai" / "settings.json"
 
+
 class Settings(BaseModel):
-    tool_repository_username: Optional[str] = Field(None, description="Username for interacting with the Tool Repository")
-    tool_repository_password: Optional[str] = Field(None, description="Password for interacting with the Tool Repository")
+    tool_repository_username: Optional[str] = Field(
+        None, description="Username for interacting with the Tool Repository"
+    )
+    tool_repository_password: Optional[str] = Field(
+        None, description="Password for interacting with the Tool Repository"
+    )
     config_path: Path = Field(default=DEFAULT_CONFIG_PATH, exclude=True)
 
     def __init__(self, config_path: Path = DEFAULT_CONFIG_PATH, **data):

src/crewai/cli/plus_api.py

@@ -1,9 +1,11 @@
-from typing import Optional
-import requests
 from os import getenv
-from crewai.cli.version import get_crewai_version
+from typing import Optional
 from urllib.parse import urljoin
 
+import requests
+
+from crewai.cli.version import get_crewai_version
+
 
 class PlusAPI:
     """

src/crewai/cli/reset_memories_command.py

@@ -1,11 +1,12 @@
 import subprocess
+
 import click
 
+from crewai.knowledge.storage.knowledge_storage import KnowledgeStorage
 from crewai.memory.entity.entity_memory import EntityMemory
 from crewai.memory.long_term.long_term_memory import LongTermMemory
 from crewai.memory.short_term.short_term_memory import ShortTermMemory
 from crewai.utilities.task_output_storage_handler import TaskOutputStorageHandler
-from crewai.knowledge.storage.knowledge_storage import KnowledgeStorage
 
 
 def reset_memories_command(

src/crewai/cli/version.py

@@ -1,6 +1,6 @@
 import importlib.metadata
 
+
 def get_crewai_version() -> str:
     """Get the version number of CrewAI running the CLI"""
     return importlib.metadata.version("crewai")
-

src/crewai/knowledge/storage/base_knowledge_storage.py

@@ -1,5 +1,5 @@
 from abc import ABC, abstractmethod
-from typing import Dict, Any, List, Optional
+from typing import Any, Dict, List, Optional
 
 
 class BaseKnowledgeStorage(ABC):

src/crewai/knowledge/storage/knowledge_storage.py

@@ -14,9 +14,9 @@ from chromadb.config import Settings
 
 from crewai.knowledge.storage.base_knowledge_storage import BaseKnowledgeStorage
 from crewai.utilities import EmbeddingConfigurator
+from crewai.utilities.constants import KNOWLEDGE_DIRECTORY
 from crewai.utilities.logger import Logger
 from crewai.utilities.paths import db_storage_path
-from crewai.utilities.constants import KNOWLEDGE_DIRECTORY
 
 
 @contextlib.contextmanager

src/crewai/memory/contextual/contextual_memory.py

@@ -1,4 +1,4 @@
-from typing import Optional, Dict, Any
+from typing import Any, Dict, Optional
 
 from crewai.memory import EntityMemory, LongTermMemory, ShortTermMemory, UserMemory
 

src/crewai/memory/memory.py

@@ -1,4 +1,4 @@
-from typing import Any, Dict, Optional, List
+from typing import Any, Dict, List, Optional
 
 from crewai.memory.storage.rag_storage import RAGStorage
 

src/crewai/memory/short_term/short_term_memory.py

@@ -1,4 +1,5 @@
 from typing import Any, Dict, Optional
+
 from crewai.memory.memory import Memory
 from crewai.memory.short_term.short_term_memory_item import ShortTermMemoryItem
 from crewai.memory.storage.rag_storage import RAGStorage
@@ -32,7 +33,10 @@ class ShortTermMemory(Memory):
                 storage
                 if storage
                 else RAGStorage(
-                    type="short_term", embedder_config=embedder_config, crew=crew, path=path
+                    type="short_term",
+                    embedder_config=embedder_config,
+                    crew=crew,
+                    path=path,
                 )
             )
         super().__init__(storage)

src/crewai/memory/storage/mem0_storage.py

@@ -2,6 +2,7 @@ import os
 from typing import Any, Dict, List
 
 from mem0 import MemoryClient
+
 from crewai.memory.storage.interface import Storage
 
 

src/crewai/project/utils.py

@@ -1,5 +1,6 @@
 from functools import wraps
 
+
 def memoize(func):
     cache = {}
 

src/crewai/task.py

@@ -1,11 +1,11 @@
 import datetime
 import json
-from pathlib import Path
 import threading
 import uuid
 from concurrent.futures import Future
 from copy import copy
 from hashlib import md5
+from pathlib import Path
 from typing import Any, Dict, List, Optional, Set, Tuple, Type, Union
 
 from opentelemetry.trace import Span

src/crewai/telemetry/telemetry.py

@@ -21,7 +21,9 @@ with suppress_warnings():
 
 
 from opentelemetry import trace  # noqa: E402
-from opentelemetry.exporter.otlp.proto.http.trace_exporter import OTLPSpanExporter # noqa: E402
+from opentelemetry.exporter.otlp.proto.http.trace_exporter import (
+    OTLPSpanExporter,  # noqa: E402
+)
 from opentelemetry.sdk.resources import SERVICE_NAME, Resource  # noqa: E402
 from opentelemetry.sdk.trace import TracerProvider  # noqa: E402
 from opentelemetry.sdk.trace.export import BatchSpanProcessor  # noqa: E402

src/crewai/tools/agent_tools/agent_tools.py

@@ -1,9 +1,9 @@
-from crewai.tools.base_tool import BaseTool
 from crewai.agents.agent_builder.base_agent import BaseAgent
+from crewai.tools.base_tool import BaseTool
 from crewai.utilities import I18N
 
-from .delegate_work_tool import DelegateWorkTool
 from .ask_question_tool import AskQuestionTool
+from .delegate_work_tool import DelegateWorkTool
 
 
 class AgentTools:

src/crewai/tools/agent_tools/ask_question_tool.py

@@ -1,7 +1,9 @@
-from crewai.tools.agent_tools.base_agent_tools import BaseAgentTool
 from typing import Optional
+
 from pydantic import BaseModel, Field
 
+from crewai.tools.agent_tools.base_agent_tools import BaseAgentTool
+
 
 class AskQuestionToolSchema(BaseModel):
     question: str = Field(..., description="The question to ask")

src/crewai/tools/agent_tools/base_agent_tools.py

@@ -1,9 +1,10 @@
 from typing import Optional, Union
+
 from pydantic import Field
 
-from crewai.tools.base_tool import BaseTool
 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
 
 

src/crewai/tools/agent_tools/delegate_work_tool.py

@@ -1,8 +1,9 @@
-from crewai.tools.agent_tools.base_agent_tools import BaseAgentTool
 from typing import Optional
 
 from pydantic import BaseModel, Field
 
+from crewai.tools.agent_tools.base_agent_tools import BaseAgentTool
+
 
 class DelegateWorkToolSchema(BaseModel):
     task: str = Field(..., description="The task to delegate")

src/crewai/tools/tool_usage_events.py

@@ -1,6 +1,7 @@
-from typing import Any, Dict
-from pydantic import BaseModel
 from datetime import datetime
+from typing import Any, Dict
+
+from pydantic import BaseModel
 
 
 class ToolUsageEvent(BaseModel):

src/crewai/utilities/crew_json_encoder.py

@@ -1,8 +1,9 @@
-from datetime import datetime, date
 import json
-from uuid import UUID
-from pydantic import BaseModel
+from datetime import date, datetime
 from decimal import Decimal
+from uuid import UUID
+
+from pydantic import BaseModel
 
 
 class CrewJSONEncoder(json.JSONEncoder):

src/crewai/utilities/crew_pydantic_output_parser.py

@@ -1,10 +1,11 @@
 import json
-import regex
 from typing import Any, Type
 
-from crewai.agents.parser import OutputParserException
+import regex
 from pydantic import BaseModel, ValidationError
 
+from crewai.agents.parser import OutputParserException
+
 
 class CrewPydanticOutputParser:
     """Parses the text into pydantic models"""

src/crewai/utilities/embedding_configurator.py

@@ -1,6 +1,7 @@
 import os
 from typing import Any, Dict, cast
-from chromadb import EmbeddingFunction, Documents, Embeddings
+
+from chromadb import Documents, EmbeddingFunction, Embeddings
 from chromadb.api.types import validate_embedding_function
 
 

src/crewai/utilities/evaluators/crew_evaluator_handler.py

@@ -1,13 +1,14 @@
 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/events.py

@@ -1,7 +1,7 @@
-from typing import Any, Callable, Generic, List, Dict, Type, TypeVar
 from functools import wraps
-from pydantic import BaseModel
+from typing import Any, Callable, Dict, Generic, List, Type, TypeVar
 
+from pydantic import BaseModel
 
 T = TypeVar("T")
 EVT = TypeVar("EVT", bound=BaseModel)

src/crewai/utilities/planning_handler.py

@@ -1,4 +1,5 @@
 from typing import Any, List, Optional
+
 from pydantic import BaseModel, Field
 
 from crewai.agent import Agent

src/crewai/utilities/prompts.py

@@ -1,5 +1,7 @@
-from pydantic import BaseModel, Field
 from typing import Any, Optional
+
+from pydantic import BaseModel, Field
+
 from crewai.utilities import I18N
 
 

src/crewai/utilities/pydantic_schema_parser.py

@@ -1,4 +1,4 @@
-from typing import Type, get_args, get_origin, Union
+from typing import Type, Union, get_args, get_origin
 
 from pydantic import BaseModel
 

src/crewai/utilities/task_output_storage_handler.py

@@ -1,6 +1,8 @@
-from pydantic import BaseModel, Field
 from datetime import datetime
-from typing import Dict, Any, Optional, List
+from typing import Any, Dict, List, Optional
+
+from pydantic import BaseModel, Field
+
 from crewai.memory.storage.kickoff_task_outputs_storage import (
     KickoffTaskOutputsSQLiteStorage,
 )

src/crewai/utilities/token_counter_callback.py

@@ -1,5 +1,6 @@
 from litellm.integrations.custom_logger import CustomLogger
 from litellm.types.utils import Usage
+
 from crewai.agents.agent_builder.utilities.base_token_process import TokenProcess
 
 
@@ -11,7 +12,7 @@ class TokenCalcHandler(CustomLogger):
         if self.token_cost_process is None:
             return
 
-        usage : Usage = response_obj["usage"]
+        usage: Usage = response_obj["usage"]
         self.token_cost_process.sum_successful_requests(1)
         self.token_cost_process.sum_prompt_tokens(usage.prompt_tokens)
         self.token_cost_process.sum_completion_tokens(usage.completion_tokens)