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",
]
# 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

View File

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