fix(graph_engine): block response nodes during streaming (#26364)

2025-09-28 17:46:39 +08:00 · 2025-09-28 17:46:39 +08:00 · d00a72a435
parent 36580221aa
commit d00a72a435
1 changed files with 2 additions and 1 deletions
--- a/api/core/workflow/graph_engine/response_coordinator/coordinator.py
+++ b/api/core/workflow/graph_engine/response_coordinator/coordinator.py
@ -212,10 +212,11 @@ class ResponseStreamCoordinator:
                edge = self._graph.edges[edge_id]
                source_node = self._graph.nodes[edge.tail]

-                # Check if node is a branch/container (original behavior)
+                # Check if node is a branch, container, or response node
                if source_node.execution_type in {
                    NodeExecutionType.BRANCH,
                    NodeExecutionType.CONTAINER,
+                    NodeExecutionType.RESPONSE,
                } or source_node.blocks_variable_output(variable_selectors):
                    blocking_edges.append(edge_id)