Make the API call work, not sure if this is the intended way

This commit is contained in:
Jedrzej Kosinski
2026-02-14 19:58:40 -08:00
parent 5f409b6cd6
commit 3caf8115b0

View File

@@ -1,5 +1,7 @@
from comfy_api.latest import ComfyExtension, io, node_replace, ComfyAPI from comfy_api.latest import ComfyExtension, io, node_replace, ComfyAPI
api = ComfyAPI()
async def register_replacements(): async def register_replacements():
"""Register all built-in node replacements.""" """Register all built-in node replacements."""
@@ -14,7 +16,7 @@ async def register_replacements():
async def register_replacements_longeredge(): async def register_replacements_longeredge():
# No dynamic inputs here # No dynamic inputs here
await ComfyAPI.node_replacement.register(node_replace.NodeReplace( await api.NodeReplacement().register(node_replace.NodeReplace(
new_node_id="ImageScaleToMaxDimension", new_node_id="ImageScaleToMaxDimension",
old_node_id="ResizeImagesByLongerEdge", old_node_id="ResizeImagesByLongerEdge",
old_widget_ids=["longer_edge"], old_widget_ids=["longer_edge"],
@@ -29,7 +31,7 @@ async def register_replacements_longeredge():
async def register_replacements_batchimages(): async def register_replacements_batchimages():
# BatchImages node uses Autogrow # BatchImages node uses Autogrow
await ComfyAPI.node_replacement.register(node_replace.NodeReplace( await api.NodeReplacement().register(node_replace.NodeReplace(
new_node_id="BatchImagesNode", new_node_id="BatchImagesNode",
old_node_id="ImageBatch", old_node_id="ImageBatch",
input_mapping=[ input_mapping=[
@@ -40,7 +42,7 @@ async def register_replacements_batchimages():
async def register_replacements_upscaleimage(): async def register_replacements_upscaleimage():
# ResizeImageMaskNode uses DynamicCombo # ResizeImageMaskNode uses DynamicCombo
await ComfyAPI.node_replacement.register(node_replace.NodeReplace( await api.NodeReplacement().register(node_replace.NodeReplace(
new_node_id="ResizeImageMaskNode", new_node_id="ResizeImageMaskNode",
old_node_id="ImageScaleBy", old_node_id="ImageScaleBy",
old_widget_ids=["upscale_method", "scale_by"], old_widget_ids=["upscale_method", "scale_by"],
@@ -54,7 +56,7 @@ async def register_replacements_upscaleimage():
async def register_replacements_controlnet(): async def register_replacements_controlnet():
# T2IAdapterLoader → ControlNetLoader # T2IAdapterLoader → ControlNetLoader
await ComfyAPI.node_replacement.register(node_replace.NodeReplace( await api.NodeReplacement().register(node_replace.NodeReplace(
new_node_id="ControlNetLoader", new_node_id="ControlNetLoader",
old_node_id="T2IAdapterLoader", old_node_id="T2IAdapterLoader",
input_mapping=[ input_mapping=[
@@ -64,28 +66,28 @@ async def register_replacements_controlnet():
async def register_replacements_load3d(): async def register_replacements_load3d():
# Load3DAnimation merged into Load3D # Load3DAnimation merged into Load3D
await ComfyAPI.node_replacement.register(node_replace.NodeReplace( await api.NodeReplacement().register(node_replace.NodeReplace(
new_node_id="Load3D", new_node_id="Load3D",
old_node_id="Load3DAnimation", old_node_id="Load3DAnimation",
)) ))
async def register_replacements_preview3d(): async def register_replacements_preview3d():
# Preview3DAnimation merged into Preview3D # Preview3DAnimation merged into Preview3D
await ComfyAPI.node_replacement.register(node_replace.NodeReplace( await api.NodeReplacement().register(node_replace.NodeReplace(
new_node_id="Preview3D", new_node_id="Preview3D",
old_node_id="Preview3DAnimation", old_node_id="Preview3DAnimation",
)) ))
async def register_replacements_svdimg2vid(): async def register_replacements_svdimg2vid():
# Typo fix: SDV → SVD # Typo fix: SDV → SVD
await ComfyAPI.node_replacement.register(node_replace.NodeReplace( await api.NodeReplacement().register(node_replace.NodeReplace(
new_node_id="SVD_img2vid_Conditioning", new_node_id="SVD_img2vid_Conditioning",
old_node_id="SDV_img2vid_Conditioning", old_node_id="SDV_img2vid_Conditioning",
)) ))
async def register_replacements_conditioningavg(): async def register_replacements_conditioningavg():
# Typo fix: trailing space in node name # Typo fix: trailing space in node name
await ComfyAPI.node_replacement.register(node_replace.NodeReplace( await api.NodeReplacement().register(node_replace.NodeReplace(
new_node_id="ConditioningAverage", new_node_id="ConditioningAverage",
old_node_id="ConditioningAverage ", old_node_id="ConditioningAverage ",
)) ))