wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-04-20 23:42:36 +08:00
Code Issues Actions 12 Packages Projects Releases Wiki Activity

server: negotiate metadata aware transport for progress text

Browse Source
This commit is contained in:
Rattus 2026-04-13 20:44:19 +10:00
parent acd718598e
commit 9ff4b7a7fb
2 changed files with 12 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
comfy_api/feature_flags.py
View File

@ -12,6 +12,7 @@ from comfy.cli_args import args
# Default server capabilities # Default server capabilities
SERVER_FEATURE_FLAGS: dict[str, Any] = { SERVER_FEATURE_FLAGS: dict[str, Any] = {
"supports_preview_metadata": True, "supports_preview_metadata": True,
"supports_progress_text_metadata": True,
"max_upload_size": args.max_upload_size * 1024 * 1024, # Convert MB to bytes "max_upload_size": args.max_upload_size * 1024 * 1024, # Convert MB to bytes
"extension": {"manager": {"supports_v4": True}}, "extension": {"manager": {"supports_v4": True}},
"node_replacements": True, "node_replacements": True,

14
server.py
View File

@ -45,6 +45,7 @@ from app.subgraph_manager import SubgraphManager
from app.node_replace_manager import NodeReplaceManager from app.node_replace_manager import NodeReplaceManager
from typing import Optional, Union from typing import Optional, Union
from api_server.routes.internal.internal_routes import InternalRoutes from api_server.routes.internal.internal_routes import InternalRoutes
from comfy_execution.utils import get_executing_context
from protocol import BinaryEventTypes from protocol import BinaryEventTypes
# Import cache control middleware # Import cache control middleware
@ -1282,7 +1283,14 @@ class PromptServer():
text = text.encode("utf-8") text = text.encode("utf-8")
node_id_bytes = str(node_id).encode("utf-8") node_id_bytes = str(node_id).encode("utf-8")
# Pack the node_id length as a 4-byte unsigned integer, followed by the node_id bytes target_sid = sid if sid is not None else self.client_id
message = struct.pack(">I", len(node_id_bytes)) + node_id_bytes + text message = b""
self.send_sync(BinaryEventTypes.TEXT, message, sid) if feature_flags.supports_feature(self.sockets_metadata, target_sid, "supports_progress_text_metadata"):
current = get_executing_context()
prompt_id_bytes = str(current.prompt_id).encode("utf-8") if current is not None else b""
message = struct.pack(">I", len(prompt_id_bytes)) + prompt_id_bytes
message += struct.pack(">I", len(node_id_bytes)) + node_id_bytes + text
self.send_sync(BinaryEventTypes.TEXT, message, target_sid)