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

Add public register_node_replacement function to node_replace, add NodeReplaceManager + GET /api/node_replacements

Browse Source
This commit is contained in:
Jedrzej Kosinski
2026-01-21 17:52:58 -08:00
parent 5faf2e3cfd
commit 191834c633
3 changed files with 37 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
comfy_api/latest/_node_replace.py
View File

@@ -1,6 +1,13 @@
from __future__ import annotations
from typing import Any
import app.node_replace_manager
def register_node_replacement(node_replace: NodeReplace):
"""
Register node replacement.
"""
app.node_replace_manager.register_node_replacement(node_replace)
class NodeReplace:
@@ -12,8 +19,8 @@ class NodeReplace:
def __init__(self,
new_node_id: str,
old_node_id: str,
input_mapping: list[InputMap],
output_mapping: list[OutputMap],
input_mapping: list[InputMap] | None=None,
output_mapping: list[OutputMap] | None=None,
):
self.new_node_id = new_node_id
self.old_node_id = old_node_id
@@ -27,8 +34,8 @@ class NodeReplace:
return {
"new_node_id": self.new_node_id,
"old_node_id": self.old_node_id,
"input_mapping": [m.as_dict() for m in self.input_mapping],
"output_mapping": [m.as_dict() for m in self.output_mapping],
"input_mapping": [m.as_dict() for m in self.input_mapping] if self.input_mapping else None,
"output_mapping": [m.as_dict() for m in self.output_mapping] if self.output_mapping else None,
}