From 8517a1462ac7151696de5be917e80c844daba9a8 Mon Sep 17 00:00:00 2001
From: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
Date: Tue, 31 Dec 2024 05:36:32 +0000
Subject: [PATCH] fix: resolve import sorting issues in agent.py and
 test_agent.py

Co-Authored-By: Joe Moura <joao@crewai.com>
---
 src/crewai/agent.py | 3 +--
 tests/test_agent.py | 4 +++-
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/src/crewai/agent.py b/src/crewai/agent.py
index 6617f258b..1a0c099a2 100644
--- a/src/crewai/agent.py
+++ b/src/crewai/agent.py
@@ -157,8 +157,7 @@ class Agent(BaseAgent):
             "AWS_REGION_NAME",
         ]
         
-        # Initialize LLM parameters with proper typing
-        from typing import Any, Dict, List, Union, Optional
+        # Initialize LLM parameters
         llm_params: Dict[str, Any] = {}
 
         # Handle different cases for self.llm
diff --git a/tests/test_agent.py b/tests/test_agent.py
index eaad771af..adf246744 100644
--- a/tests/test_agent.py
+++ b/tests/test_agent.py
@@ -1,10 +1,12 @@
 import os
-import pytest
 from unittest import mock
 
+import pytest
+
 from crewai.agent import Agent
 from crewai.llm import LLM
 
+
 def test_agent_with_custom_llm():
     """Test creating an agent with a custom LLM."""
     custom_llm = LLM(model="gpt-4")