diff --git a/src/crewai/flow/flow.py b/src/crewai/flow/flow.py index 5af4a4075..2babbe57c 100644 --- a/src/crewai/flow/flow.py +++ b/src/crewai/flow/flow.py @@ -22,6 +22,10 @@ from pydantic import BaseModel, Field, ValidationError from crewai.flow.flow_visualizer import plot_flow from crewai.flow.persistence.base import FlowPersistence from crewai.flow.utils import get_possible_return_constants +from crewai.traces.unified_trace_controller import ( + init_flow_main_trace, + trace_flow_step, +) from crewai.utilities.events.crewai_event_bus import crewai_event_bus from crewai.utilities.events.flow_events import ( FlowCreatedEvent, @@ -32,14 +36,6 @@ from crewai.utilities.events.flow_events import ( MethodExecutionFinishedEvent, MethodExecutionStartedEvent, ) -from crewai.flow.flow_visualizer import plot_flow -from crewai.flow.persistence.base import FlowPersistence -from crewai.flow.utils import get_possible_return_constants -from crewai.telemetry import Telemetry -from crewai.traces.unified_trace_controller import ( - init_flow_main_trace, - trace_flow_step, -) from crewai.utilities.printer import Printer logger = logging.getLogger(__name__) diff --git a/src/crewai/llm.py b/src/crewai/llm.py index 29c9f93a8..2d8a08228 100644 --- a/src/crewai/llm.py +++ b/src/crewai/llm.py @@ -31,8 +31,8 @@ with warnings.catch_warnings(): from litellm.utils import supports_response_schema -from crewai.utilities.events import crewai_event_bus from crewai.traces.unified_trace_controller import trace_llm_call +from crewai.utilities.events import crewai_event_bus from crewai.utilities.exceptions.context_window_exceeding_exception import ( LLMContextLengthExceededException, )