fix: resolve import sorting issues in agent.py and test_agent.py

Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
Devin AI
2024-12-31 05:36:32 +00:00
parent 452aa9f173
commit 8517a1462a
2 changed files with 4 additions and 3 deletions

View File

@@ -157,8 +157,7 @@ class Agent(BaseAgent):
"AWS_REGION_NAME", "AWS_REGION_NAME",
] ]
# Initialize LLM parameters with proper typing # Initialize LLM parameters
from typing import Any, Dict, List, Union, Optional
llm_params: Dict[str, Any] = {} llm_params: Dict[str, Any] = {}
# Handle different cases for self.llm # Handle different cases for self.llm

View File

@@ -1,10 +1,12 @@
import os import os
import pytest
from unittest import mock from unittest import mock
import pytest
from crewai.agent import Agent from crewai.agent import Agent
from crewai.llm import LLM from crewai.llm import LLM
def test_agent_with_custom_llm(): def test_agent_with_custom_llm():
"""Test creating an agent with a custom LLM.""" """Test creating an agent with a custom LLM."""
custom_llm = LLM(model="gpt-4") custom_llm = LLM(model="gpt-4")