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")