diff --git a/src/crewai/agent.py b/src/crewai/agent.py index dc0a544c7..6c8b7d20e 100644 --- a/src/crewai/agent.py +++ b/src/crewai/agent.py @@ -3,16 +3,12 @@ import subprocess import time from typing import ( Any, - Callable, - Dict, - List, Literal, Optional, - Sequence, - 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 23dbb1403..cb84fba8c 100644 --- a/src/crewai/agents/crew_agent_executor.py +++ b/src/crewai/agents/crew_agent_executor.py @@ -1,4 +1,6 @@ -from typing import Any, Callable, Dict, List, Optional, Union +from typing import Any, Optional, Union +from typing import Dict, List +from collections.abc import Callable from crewai.agents.agent_builder.base_agent import BaseAgent from crewai.agents.agent_builder.base_agent_executor_mixin import CrewAgentExecutorMixin diff --git a/src/crewai/llm.py b/src/crewai/llm.py index 153aa5ff9..baf3ee61a 100644 --- a/src/crewai/llm.py +++ b/src/crewai/llm.py @@ -8,16 +8,13 @@ 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 @@ -35,10 +32,7 @@ from crewai.events.types.tool_usage_events import ( ToolUsageFinishedEvent, ToolUsageErrorEvent, ) -from crewai.utilities.exceptions import ( - LLMContextLengthExceededException, - LLMQuotaLimitExceededException, -) +from crewai.utilities.exceptions import LLMQuotaLimitExceededException with warnings.catch_warnings(): warnings.simplefilter("ignore", UserWarning)