From 9c7d5f1fddfa64aa5a92bd63e82c28584f7ed384 Mon Sep 17 00:00:00 2001
From: Jedrzej Kosinski <kosinkadink1@gmail.com>
Date: Thu, 22 Jan 2026 03:26:52 -0800
Subject: [PATCH] Added old_widget_ids param to NodeReplace

---
 comfy_api/latest/_node_replace.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/comfy_api/latest/_node_replace.py b/comfy_api/latest/_node_replace.py
index b8278d09d..f179d9809 100644
--- a/comfy_api/latest/_node_replace.py
+++ b/comfy_api/latest/_node_replace.py
@@ -19,11 +19,13 @@ class NodeReplace:
     def __init__(self,
         new_node_id: str,
         old_node_id: str,
+        old_widget_ids: list[str] | None=None,
         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
+        self.old_widget_ids = old_widget_ids
         self.input_mapping = input_mapping
         self.output_mapping = output_mapping
 
@@ -34,6 +36,7 @@ class NodeReplace:
         return {
             "new_node_id": self.new_node_id,
             "old_node_id": self.old_node_id,
+            "old_widget_ids": self.old_widget_ids,
             "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,
         }