diff --git a/src/crewai/agent.py b/src/crewai/agent.py index 6c8b7d20e..1f7586071 100644 --- a/src/crewai/agent.py +++ b/src/crewai/agent.py @@ -3,11 +3,14 @@ import subprocess import time from typing import ( Any, + Dict, + List, Literal, Optional, + Tuple, + Type, Union, ) -from typing import Dict, List, Tuple, Type from collections.abc import Callable, Sequence from pydantic import Field, InstanceOf, PrivateAttr, model_validator diff --git a/src/crewai/agents/crew_agent_executor.py b/src/crewai/agents/crew_agent_executor.py index cb84fba8c..9d874d65c 100644 --- a/src/crewai/agents/crew_agent_executor.py +++ b/src/crewai/agents/crew_agent_executor.py @@ -1,5 +1,4 @@ -from typing import Any, Optional, Union -from typing import Dict, List +from typing import Any, Dict, List, Optional, Union from collections.abc import Callable from crewai.agents.agent_builder.base_agent import BaseAgent diff --git a/src/crewai/llm.py b/src/crewai/llm.py index baf3ee61a..ef2a32420 100644 --- a/src/crewai/llm.py +++ b/src/crewai/llm.py @@ -8,13 +8,16 @@ from collections import defaultdict from contextlib import contextmanager from typing import ( Any, + DefaultDict, + Dict, + List, Literal, Optional, + Type, TypedDict, Union, cast, ) -from typing import DefaultDict, Dict, List, Type from datetime import datetime from dotenv import load_dotenv from litellm.types.utils import ChatCompletionDeltaToolCall