From 3caf8115b0bdf2658e5c3158a04c9e7295cb0cdd Mon Sep 17 00:00:00 2001
From: Jedrzej Kosinski <kosinkadink1@gmail.com>
Date: Sat, 14 Feb 2026 19:58:40 -0800
Subject: [PATCH] Make the API call work, not sure if this is the intended way

---
 comfy_extras/nodes_replacements.py | 18 ++++++++++--------
 1 file changed, 10 insertions(+), 8 deletions(-)

diff --git a/comfy_extras/nodes_replacements.py b/comfy_extras/nodes_replacements.py
index b39a1de58..7e99b802c 100644
--- a/comfy_extras/nodes_replacements.py
+++ b/comfy_extras/nodes_replacements.py
@@ -1,5 +1,7 @@
 from comfy_api.latest import ComfyExtension, io, node_replace, ComfyAPI
 
+api = ComfyAPI()
+
 
 async def register_replacements():
     """Register all built-in node replacements."""
@@ -14,7 +16,7 @@ async def register_replacements():
 
 async def register_replacements_longeredge():
     # No dynamic inputs here
-    await ComfyAPI.node_replacement.register(node_replace.NodeReplace(
+    await api.NodeReplacement().register(node_replace.NodeReplace(
             new_node_id="ImageScaleToMaxDimension",
             old_node_id="ResizeImagesByLongerEdge",
             old_widget_ids=["longer_edge"],
@@ -29,7 +31,7 @@ async def register_replacements_longeredge():
 
 async def register_replacements_batchimages():
     # BatchImages node uses Autogrow
-    await ComfyAPI.node_replacement.register(node_replace.NodeReplace(
+    await api.NodeReplacement().register(node_replace.NodeReplace(
             new_node_id="BatchImagesNode",
             old_node_id="ImageBatch",
             input_mapping=[
@@ -40,7 +42,7 @@ async def register_replacements_batchimages():
 
 async def register_replacements_upscaleimage():
     # ResizeImageMaskNode uses DynamicCombo
-    await ComfyAPI.node_replacement.register(node_replace.NodeReplace(
+    await api.NodeReplacement().register(node_replace.NodeReplace(
             new_node_id="ResizeImageMaskNode",
             old_node_id="ImageScaleBy",
             old_widget_ids=["upscale_method", "scale_by"],
@@ -54,7 +56,7 @@ async def register_replacements_upscaleimage():
 
 async def register_replacements_controlnet():
     # T2IAdapterLoader → ControlNetLoader
-    await ComfyAPI.node_replacement.register(node_replace.NodeReplace(
+    await api.NodeReplacement().register(node_replace.NodeReplace(
             new_node_id="ControlNetLoader",
             old_node_id="T2IAdapterLoader",
             input_mapping=[
@@ -64,28 +66,28 @@ async def register_replacements_controlnet():
 
 async def register_replacements_load3d():
     # Load3DAnimation merged into Load3D
-    await ComfyAPI.node_replacement.register(node_replace.NodeReplace(
+    await api.NodeReplacement().register(node_replace.NodeReplace(
             new_node_id="Load3D",
             old_node_id="Load3DAnimation",
         ))
 
 async def register_replacements_preview3d():
     # Preview3DAnimation merged into Preview3D
-    await ComfyAPI.node_replacement.register(node_replace.NodeReplace(
+    await api.NodeReplacement().register(node_replace.NodeReplace(
             new_node_id="Preview3D",
             old_node_id="Preview3DAnimation",
         ))
 
 async def register_replacements_svdimg2vid():
     # 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",
             old_node_id="SDV_img2vid_Conditioning",
         ))
 
 async def register_replacements_conditioningavg():
     # 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",
             old_node_id="ConditioningAverage ",
         ))