From db452c1e63342cd536da031bcd399da2e7578c4c Mon Sep 17 00:00:00 2001
From: AustinMroz <austin@comfy.org>
Date: Sat, 9 Aug 2025 02:45:52 -0500
Subject: [PATCH] Fix disconnection from subgraph inputs (#4800)

---
 src/lib/litegraph/src/canvas/LinkConnector.ts         | 5 ++++-
 src/lib/litegraph/src/canvas/ToInputFromIoNodeLink.ts | 5 +++++
 2 files changed, 9 insertions(+), 1 deletion(-)

diff --git a/src/lib/litegraph/src/canvas/LinkConnector.ts b/src/lib/litegraph/src/canvas/LinkConnector.ts
index b42c12b2d..8dea4c7a0 100644
--- a/src/lib/litegraph/src/canvas/LinkConnector.ts
+++ b/src/lib/litegraph/src/canvas/LinkConnector.ts
@@ -795,7 +795,10 @@ export class LinkConnector {
    */
   disconnectLinks(): void {
     for (const link of this.renderLinks) {
-      if (link instanceof MovingLinkBase) {
+      if (
+        link instanceof MovingLinkBase ||
+        link instanceof ToInputFromIoNodeLink
+      ) {
         link.disconnect()
       }
     }
diff --git a/src/lib/litegraph/src/canvas/ToInputFromIoNodeLink.ts b/src/lib/litegraph/src/canvas/ToInputFromIoNodeLink.ts
index 14dc5b05c..f056ef687 100644
--- a/src/lib/litegraph/src/canvas/ToInputFromIoNodeLink.ts
+++ b/src/lib/litegraph/src/canvas/ToInputFromIoNodeLink.ts
@@ -135,4 +135,9 @@ export class ToInputFromIoNodeLink implements RenderLink {
   connectToRerouteOutput() {
     throw new Error('ToInputRenderLink cannot connect to an output.')
   }
+  disconnect(): boolean {
+    if (!this.existingLink) return false
+    this.existingLink.disconnect(this.network, 'input')
+    return true
+  }
 }