fix: suppress link rendering during slot sync after graph reconfigure

src/lib/litegraph/src/LGraphCanvas.ts

@@ -5614,6 +5614,12 @@ export class LGraphCanvas implements CustomEventDispatcher<LGraphCanvasEventMap>
     this.renderedPaths.clear()
     if (this.links_render_mode === LinkRenderType.HIDDEN_LINK) return
 
+    // Skip link rendering while waiting for slot positions to sync after reconfigure
+    if (LiteGraph.vueNodesMode && layoutStore.pendingSlotSync) {
+      this.#visibleReroutes.clear()
+      return
+    }
+
     const { graph, subgraph } = this
     if (!graph) throw new NullGraphError()
 

src/renderer/core/layout/store/layoutStore.ts

@@ -141,6 +141,20 @@ class LayoutStoreImpl implements LayoutStore {
   // Vue resizing state to prevent drag from activating during resize
   public isResizingVueNodes = ref(false)
 
+  /**
+   * Flag indicating slot positions are pending sync after graph reconfiguration.
+   * When true, link rendering should be skipped to avoid drawing with stale positions.
+   */
+  private _pendingSlotSync = false
+
+  get pendingSlotSync(): boolean {
+    return this._pendingSlotSync
+  }
+
+  setPendingSlotSync(value: boolean): void {
+    this._pendingSlotSync = value
+  }
+
   constructor() {
     // Initialize Yjs data structures
     this.ynodes = this.ydoc.getMap('nodes')

src/renderer/extensions/vueNodes/composables/useSlotElementTracking.ts

@@ -28,16 +28,26 @@ const raf = createRafBatch(() => {
 
 function scheduleSlotLayoutSync(nodeId: string) {
   pendingNodes.add(nodeId)
+  // Re-assert pending flag for late mounts (Vue components mounting after
+  // flushScheduledSlotLayoutSync was called synchronously in onConfigure)
+  layoutStore.setPendingSlotSync(true)
   raf.schedule()
 }
 
-function flushScheduledSlotLayoutSync() {
-  if (pendingNodes.size === 0) return
+export function flushScheduledSlotLayoutSync() {
+  if (pendingNodes.size === 0) {
+    // No pending nodes - don't clear the flag here as late mounts may still
+    // call scheduleSlotLayoutSync. The RAF callback will clear it after
+    // processing actual nodes.
+    return
+  }
   const conv = useSharedCanvasPositionConversion()
   for (const nodeId of Array.from(pendingNodes)) {
     pendingNodes.delete(nodeId)
     syncNodeSlotLayoutsFromDOM(nodeId, conv)
   }
+  // Clear the pending sync flag - slots are now synced
+  layoutStore.setPendingSlotSync(false)
 }
 
 export function syncNodeSlotLayoutsFromDOM(

src/scripts/app.ts

@@ -5,6 +5,8 @@ import { reactive, unref } from 'vue'
 import { shallowRef } from 'vue'
 
 import { useCanvasPositionConversion } from '@/composables/element/useCanvasPositionConversion'
+import { layoutStore } from '@/renderer/core/layout/store/layoutStore'
+import { flushScheduledSlotLayoutSync } from '@/renderer/extensions/vueNodes/composables/useSlotElementTracking'
 import { registerProxyWidgets } from '@/core/graph/subgraph/proxyWidget'
 import { st, t } from '@/i18n'
 import type { IContextMenuValue } from '@/lib/litegraph/src/interfaces'
@@ -730,17 +732,31 @@ export class ComfyApp {
   private addAfterConfigureHandler(graph: LGraph) {
     const { onConfigure } = graph
     graph.onConfigure = function (...args) {
-      fixLinkInputSlots(this)
+      // Set pending sync flag to suppress link rendering until slots are synced
+      if (LiteGraph.vueNodesMode) {
+        layoutStore.setPendingSlotSync(true)
+      }
+
+      try {
+        fixLinkInputSlots(this)
 
-      // Fire callbacks before the onConfigure, this is used by widget inputs to setup the config
-      triggerCallbackOnAllNodes(this, 'onGraphConfigured')
+        // Fire callbacks before the onConfigure, this is used by widget inputs to setup the config
+        triggerCallbackOnAllNodes(this, 'onGraphConfigured')
 
-      const r = onConfigure?.apply(this, args)
+        const r = onConfigure?.apply(this, args)
 
-      // Fire after onConfigure, used by primitives to generate widget using input nodes config
-      triggerCallbackOnAllNodes(this, 'onAfterGraphConfigured')
+        // Fire after onConfigure, used by primitives to generate widget using input nodes config
+        triggerCallbackOnAllNodes(this, 'onAfterGraphConfigured')
 
-      return r
+        return r
+      } finally {
+        // Flush pending slot layout syncs to fix link alignment after undo/redo
+        // Using finally ensures links aren't permanently suppressed if an error occurs
+        if (LiteGraph.vueNodesMode) {
+          flushScheduledSlotLayoutSync()
+          app.canvas?.setDirty(true, true)
+        }
+      }
     }
   }