zlei6/ComfyUI
1
0
Fork 0
mirror of https://github.com/Comfy-Org/ComfyUI.git synced 2026-08-24 18:10:27 +08:00
Code Issues Packages Projects Releases Wiki Activity

Include workflow_id in all execution WebSocket messages (CORE-198) (#13684)

Browse Source
This commit is contained in:
Christian Byrne
2026-05-14 15:11:34 -07:00
committed by GitHub
parent 7a063e83a7
commit 4f6018982d
7 changed files with 398 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
main.py
View File

@@ -29,6 +29,7 @@ import logging
import sys
from comfy_execution.progress import get_progress_state
from comfy_execution.utils import get_executing_context
from comfy_execution.jobs import extract_workflow_id
from comfy_api import feature_flags
from app.database.db import init_db, dependencies_available
@@ -317,6 +318,12 @@ def prompt_worker(q, server_instance):
for k in sensitive:
extra_data[k] = sensitive[k]
# Capture the workflow id for this prompt before execution: the
# executor clears server.last_workflow_id in its finally block, so
# reading it after e.execute() returns would emit workflow_id=None
# on the terminal "executing" reset below.
workflow_id = extract_workflow_id(extra_data)
asset_seeder.pause()
e.execute(item[2], prompt_id, extra_data, item[4])
@@ -330,7 +337,7 @@ def prompt_worker(q, server_instance):
completed=e.success,
messages=e.status_messages), process_item=remove_sensitive)
if server_instance.client_id is not None:
server_instance.send_sync("executing", {"node": None, "prompt_id": prompt_id}, server_instance.client_id)
server_instance.send_sync("executing", {"node": None, "prompt_id": prompt_id, "workflow_id": workflow_id}, server_instance.client_id)
current_time = time.perf_counter()
execution_time = current_time - execution_start_time
@@ -393,7 +400,7 @@ def hijack_progress(server_instance):
prompt_id = server_instance.last_prompt_id
if node_id is None:
node_id = server_instance.last_node_id
progress = {"value": value, "max": total, "prompt_id": prompt_id, "node": node_id}
progress = {"value": value, "max": total, "prompt_id": prompt_id, "workflow_id": getattr(server_instance, 'last_workflow_id', None), "node": node_id}
get_progress_state().update_progress(node_id, value, total, preview_image)
server_instance.send_sync("progress", progress, server_instance.client_id)