diff --git a/src/crewai/flow/flow.py b/src/crewai/flow/flow.py index cac5962a7..8db501ea4 100644 --- a/src/crewai/flow/flow.py +++ b/src/crewai/flow/flow.py @@ -17,13 +17,13 @@ from typing import ( from blinker import Signal from pydantic import BaseModel, ValidationError +from crewai.flow.core_flow_utils import get_possible_return_constants from crewai.flow.flow_events import ( FlowFinishedEvent, FlowStartedEvent, MethodExecutionFinishedEvent, MethodExecutionStartedEvent, ) -from crewai.flow.core_flow_utils import get_possible_return_constants from crewai.telemetry import Telemetry T = TypeVar("T", bound=Union[BaseModel, Dict[str, Any]]) diff --git a/src/crewai/flow/flow_visualizer.py b/src/crewai/flow/flow_visualizer.py index c1c8d61c8..04ba8a815 100644 --- a/src/crewai/flow/flow_visualizer.py +++ b/src/crewai/flow/flow_visualizer.py @@ -5,12 +5,11 @@ from pathlib import Path from pyvis.network import Network -from crewai.flow.path_utils import safe_path_join, validate_file_path - from crewai.flow.config import COLORS, NODE_STYLES +from crewai.flow.flow_visual_utils import calculate_node_levels from crewai.flow.html_template_handler import HTMLTemplateHandler from crewai.flow.legend_generator import generate_legend_items_html, get_legend_items -from crewai.flow.flow_visual_utils import calculate_node_levels +from crewai.flow.path_utils import safe_path_join, validate_file_path from crewai.flow.visualization_utils import ( add_edges, add_nodes_to_network, diff --git a/src/crewai/flow/path_utils.py b/src/crewai/flow/path_utils.py index e770477a5..53aecd6aa 100644 --- a/src/crewai/flow/path_utils.py +++ b/src/crewai/flow/path_utils.py @@ -8,7 +8,7 @@ interface for path operations. import os from pathlib import Path -from typing import Union, List, Optional +from typing import List, Optional, Union def safe_path_join(base_dir: Union[str, Path], filename: str) -> str: