mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-09 08:08:32 +00:00
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:
@@ -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
|
||||||
|
|||||||
@@ -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")
|
||||||
|
|||||||
Reference in New Issue
Block a user