Merge pull request #371 from whiteducksoftware/fix/join-spec-graph

Fix JoinSpec graph visualization

Author: AndreRatzenberger

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "flock-core"
-version = "0.5.311"
+version = "0.5.312"
 description = "Flock: A declrative framework for building and orchestrating AI agents."
 readme = "README.md"
 requires-python = ">=3.12"

src/flock/api/graph_builder.py

@@ -480,7 +480,7 @@ def _build_pending_join_edges(
         start_counter: int,
     ) -> list[GraphEdge]:
         """Build pending edges for JoinSpec correlation groups."""
-        from flock.dashboard.routes.helpers import _get_correlation_groups
+        from flock.components.server.control.helpers import _get_correlation_groups
 
         edges: list[GraphEdge] = []
         correlation_groups = _get_correlation_groups(
@@ -555,7 +555,7 @@ def _build_pending_batch_edges(
         start_counter: int,
     ) -> list[GraphEdge]:
         """Build pending edges for BatchSpec accumulation."""
-        from flock.dashboard.routes.helpers import _get_batch_state
+        from flock.components.server.control.helpers import _get_batch_state
 
         edges: list[GraphEdge] = []
         batch_state = _get_batch_state(
@@ -835,7 +835,7 @@ def _build_logic_config_for_subscription(self, agent, subscription, idx):
             }
 
             # Phase 1.2.1: Get waiting state from CorrelationEngine
-            from flock.dashboard.routes.helpers import _get_correlation_groups
+            from flock.components.server.control.helpers import _get_correlation_groups
 
             correlation_groups = _get_correlation_groups(
                 self._orchestrator._correlation_engine, agent.name, idx
@@ -868,7 +868,7 @@ def _build_logic_config_for_subscription(self, agent, subscription, idx):
                 )
 
             # Phase 1.2.1: Get waiting state from BatchEngine
-            from flock.dashboard.routes.helpers import _get_batch_state
+            from flock.components.server.control.helpers import _get_batch_state
 
             batch_state = _get_batch_state(
                 self._orchestrator._batch_engine, agent.name, idx, batch_spec