diff --git a/src/crewai/crew.py b/src/crewai/crew.py index 5c2af882e..cf144148c 100644 --- a/src/crewai/crew.py +++ b/src/crewai/crew.py @@ -39,7 +39,6 @@ from crewai.tasks.conditional_task import ConditionalTask from crewai.tasks.task_output import TaskOutput from crewai.tools.agent_tools.agent_tools import AgentTools from crewai.tools.base_tool import BaseTool, Tool -from crewai.utilities.exceptions.llm_error import LLMError from crewai.types.usage_metrics import UsageMetrics from crewai.utilities import I18N, FileHandler, Logger, RPMController from crewai.utilities.constants import TRAINING_DATA_FILE @@ -58,6 +57,7 @@ from crewai.utilities.events.crew_events import ( ) from crewai.utilities.events.crewai_event_bus import crewai_event_bus from crewai.utilities.events.event_listener import EventListener +from crewai.utilities.exceptions.llm_error import LLMError from crewai.utilities.formatter import ( aggregate_raw_outputs_from_task_outputs, aggregate_raw_outputs_from_tasks, diff --git a/tests/crew_test.py b/tests/crew_test.py index 327bfe0d5..e98ef17d7 100644 --- a/tests/crew_test.py +++ b/tests/crew_test.py @@ -1503,6 +1503,7 @@ async def test_async_kickoff_for_each_async_empty_input(): async def test_kickoff_async_error_handling(): """Tests error handling in kickoff_async when kickoff raises an error.""" from unittest.mock import patch + from crewai.utilities.exceptions.llm_error import LLMError inputs = {"topic": "dog"} @@ -1537,7 +1538,10 @@ async def test_kickoff_async_error_handling(): async def test_kickoff_async_context_length_error_handling(): """Tests error handling in kickoff_async when kickoff raises a context length error.""" from unittest.mock import patch - from crewai.utilities.exceptions.context_window_exceeding_exception import LLMContextLengthExceededException + + from crewai.utilities.exceptions.context_window_exceeding_exception import ( + LLMContextLengthExceededException, + ) from crewai.utilities.exceptions.llm_error import LLMError inputs = {"topic": "dog"}