comfyanonymous/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-02-19 14:40:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add a public API function for registering node replacements, refactor code accordingly

Browse Source
This commit is contained in:
Jedrzej Kosinski
2026-02-14 19:28:02 -08:00
parent 1ab9d4a9db
commit 07d1ee2ca9
3 changed files with 17 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
app/node_replace_manager.py
View File

@@ -7,7 +7,7 @@ if TYPE_CHECKING:
from comfy_api.latest._node_replace import NodeReplace
from comfy_execution.graph_utils import is_link
from nodes import NODE_CLASS_MAPPINGS
import nodes
class NodeStruct(TypedDict):
inputs: dict[str, str | int | float | bool | tuple[str, int]]
@@ -48,7 +48,7 @@ class NodeReplaceManager:
for node_number, node_struct in prompt.items():
class_type = node_struct["class_type"]
# need replacement if not in NODE_CLASS_MAPPINGS and has replacement
if class_type not in NODE_CLASS_MAPPINGS.keys() and self.has_replacement(class_type):
if class_type not in nodes.NODE_CLASS_MAPPINGS.keys() and self.has_replacement(class_type):
need_replacement.add(node_number)
# keep track of connections
for input_id, input_value in node_struct["inputs"].items():
@@ -65,7 +65,7 @@ class NodeReplaceManager:
replacement = replacements[0]
new_node_id = replacement.new_node_id
# if replacement is not a valid node, skip trying to replace it as will only cause confusion
if new_node_id not in NODE_CLASS_MAPPINGS.keys():
if new_node_id not in nodes.NODE_CLASS_MAPPINGS.keys():
continue
# first, replace node id (class_type)
new_node_struct = copy_node_struct(node_struct, empty_inputs=True)