Fix reroute to wildcard & multi-typed slots (#769)

Use the same type check the rest of the connection process uses.
Comfy-Org · Sep 9, 2024 · 0e01bb3 · 0e01bb3
1 parent 8b77dde
commit 0e01bb3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/extensions/core/rerouteNode.ts b/src/extensions/core/rerouteNode.ts
@@ -128,8 +128,8 @@ app.registerExtension({
                       : null
                   if (
                     inputType &&
-                    inputType !== '*' &&
-                    nodeOutType !== inputType
+                    // @ts-expect-error Will self-resolve when LiteGraph types are generated
+                    !LiteGraph.isValidConnection(inputType, nodeOutType)
                   ) {
                     // The output doesnt match our input so disconnect it
                     node.disconnectInput(link.target_slot)