Fix branch join convergence and remote log visibility gating

This commit is contained in:
Robert Wojciechowski
2026-03-02 06:57:41 +00:00
parent cfed768528
commit dd3fbfadbe
7 changed files with 82 additions and 7 deletions
+4 -1
View File
@@ -603,7 +603,10 @@ def _override_join_nodes(
assigned_branch = -1
inputs = node.setdefault("inputs", {})
inputs["multi_job_id"] = job_id_map.get(node_id, node_id)
# Group all joins under the same DistributedBranch into one queue so
# worker-branch senders converge into the master's active join node.
job_source_id = upstream_branch_id or node_id
inputs["multi_job_id"] = job_id_map.get(job_source_id, job_source_id)
inputs["is_worker"] = not is_master
inputs["enabled_worker_ids"] = enabled_json
inputs["assigned_branch"] = assigned_branch
+1
View File
@@ -327,6 +327,7 @@ async def orchestrate_distributed_execution(
prompt_index.nodes_for_class("DistributedCollector")
+ prompt_index.nodes_for_class("DistributedListCollector")
+ prompt_index.nodes_for_class("DistributedJoin")
+ prompt_index.nodes_for_class("DistributedBranch")
+ prompt_index.nodes_for_class("UltimateSDUpscaleDistributed")
)
for node_id in queue_node_ids: