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>

tests/agents/agent_builder/base_agent_test.py

@@ -1,9 +1,10 @@
 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,10 +1,11 @@
 import pytest
-from crewai.agents.parser import CrewAgentParser
+
 from crewai.agents.crew_agent_executor import (
     AgentAction,
     AgentFinish,
     OutputParserException,
 )
+from crewai.agents.parser import CrewAgentParser
 
 
 @pytest.fixture

tests/cli/authentication/test_auth_main.py

@@ -2,6 +2,7 @@ import unittest
 from unittest.mock import MagicMock, patch
 
 import requests
+
 from crewai.cli.authentication.main import AuthenticationCommand
 
 
@@ -47,7 +48,9 @@ class TestAuthenticationCommand(unittest.TestCase):
     @patch("crewai.cli.authentication.main.requests.post")
     @patch("crewai.cli.authentication.main.validate_token")
     @patch("crewai.cli.authentication.main.console.print")
-    def test_poll_for_token_success(self, mock_print, mock_validate_token, mock_post, mock_tool):
+    def test_poll_for_token_success(
+        self, mock_print, mock_validate_token, mock_post, mock_tool
+    ):
         mock_response = MagicMock()
         mock_response.status_code = 200
         mock_response.json.return_value = {
@@ -62,7 +65,9 @@ class TestAuthenticationCommand(unittest.TestCase):
         self.auth_command._poll_for_token({"device_code": "123456"})
 
         mock_validate_token.assert_called_once_with("TOKEN")
-        mock_print.assert_called_once_with("\n[bold green]Welcome to CrewAI Enterprise![/bold green]\n")
+        mock_print.assert_called_once_with(
+            "\n[bold green]Welcome to CrewAI Enterprise![/bold green]\n"
+        )
 
     @patch("crewai.cli.authentication.main.requests.post")
     @patch("crewai.cli.authentication.main.console.print")

tests/cli/authentication/test_utils.py

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

tests/cli/config_test.py

@@ -1,10 +1,12 @@
-import unittest
 import json
-import tempfile
 import shutil
+import tempfile
+import unittest
 from pathlib import Path
+
 from crewai.cli.config import Settings
 
+
 class TestSettings(unittest.TestCase):
     def setUp(self):
         self.test_dir = Path(tempfile.mkdtemp())
@@ -20,8 +22,7 @@ class TestSettings(unittest.TestCase):
 
     def test_initialization_with_data(self):
         settings = Settings(
-            config_path=self.config_path,
-            tool_repository_username="user1"
+            config_path=self.config_path, tool_repository_username="user1"
         )
         self.assertEqual(settings.tool_repository_username, "user1")
         self.assertIsNone(settings.tool_repository_password)
@@ -37,22 +38,23 @@ class TestSettings(unittest.TestCase):
     def test_merge_file_and_input_data(self):
         self.config_path.parent.mkdir(parents=True, exist_ok=True)
         with self.config_path.open("w") as f:
-            json.dump({
-                "tool_repository_username": "file_user",
-                "tool_repository_password": "file_pass"
-            }, f)
+            json.dump(
+                {
+                    "tool_repository_username": "file_user",
+                    "tool_repository_password": "file_pass",
+                },
+                f,
+            )
 
         settings = Settings(
-            config_path=self.config_path,
-            tool_repository_username="new_user"
+            config_path=self.config_path, tool_repository_username="new_user"
         )
         self.assertEqual(settings.tool_repository_username, "new_user")
         self.assertEqual(settings.tool_repository_password, "file_pass")
 
     def test_dump_new_settings(self):
         settings = Settings(
-            config_path=self.config_path,
-            tool_repository_username="user1"
+            config_path=self.config_path, tool_repository_username="user1"
         )
         settings.dump()
 
@@ -67,8 +69,7 @@ class TestSettings(unittest.TestCase):
             json.dump({"existing_setting": "value"}, f)
 
         settings = Settings(
-            config_path=self.config_path,
-            tool_repository_username="user1"
+            config_path=self.config_path, tool_repository_username="user1"
         )
         settings.dump()
 
@@ -79,10 +80,7 @@ class TestSettings(unittest.TestCase):
         self.assertEqual(saved_data["tool_repository_username"], "user1")
 
     def test_none_values(self):
-        settings = Settings(
-            config_path=self.config_path,
-            tool_repository_username=None
-        )
+        settings = Settings(config_path=self.config_path, tool_repository_username=None)
         settings.dump()
 
         with self.config_path.open("r") as f:

tests/cli/test_git.py

@@ -1,6 +1,7 @@
-from crewai.cli.git import Repository
 import pytest
 
+from crewai.cli.git import Repository
+
 
 @pytest.fixture()
 def repository(fp):

tests/cli/test_plus_api.py

@@ -1,6 +1,7 @@
 import os
 import unittest
 from unittest.mock import MagicMock, patch
+
 from crewai.cli.plus_api import PlusAPI
 
 

tests/cli/test_utils.py

@@ -1,7 +1,9 @@
-import pytest
+import os
 import shutil
 import tempfile
-import os
+
+import pytest
+
 from crewai.cli import utils
 
 

tests/flow_test.py

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

tests/knowledge/knowledge_test.py

@@ -3,6 +3,8 @@
 from pathlib import Path
 from unittest.mock import patch
 
+import pytest
+
 from crewai.knowledge.source.csv_knowledge_source import CSVKnowledgeSource
 from crewai.knowledge.source.excel_knowledge_source import ExcelKnowledgeSource
 from crewai.knowledge.source.json_knowledge_source import JSONKnowledgeSource
@@ -10,8 +12,6 @@ from crewai.knowledge.source.pdf_knowledge_source import PDFKnowledgeSource
 from crewai.knowledge.source.string_knowledge_source import StringKnowledgeSource
 from crewai.knowledge.source.text_file_knowledge_source import TextFileKnowledgeSource
 
-import pytest
-
 
 @pytest.fixture(autouse=True)
 def mock_vector_db():

tests/memory/short_term_memory_test.py

@@ -1,5 +1,7 @@
-import pytest
 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/task_test.py

@@ -6,12 +6,13 @@ 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/test_tool_usage.py

@@ -6,8 +6,8 @@ import pytest
 from pydantic import BaseModel, Field
 
 from crewai import Agent, Task
-from crewai.tools.tool_usage import ToolUsage
 from crewai.tools import BaseTool
+from crewai.tools.tool_usage import ToolUsage
 
 
 class RandomNumberToolInput(BaseModel):

tests/utilities/evaluators/test_crew_evaluator_handler.py

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