@bian-womp/spark-graph 0.3.33 → 0.3.35

package/lib/cjs/index.cjs

@@ -2380,7 +2380,7 @@ class NodeExecutor {
     /**
      * Internal method for executing inputs changed (also used by GraphRuntime)
      */
-    execute(nodeId, runContextIds) {
+    execute(nodeId, runContextIds, canSkipHandleResolution) {
         const node = this.graph.getNode(nodeId);
         if (!node)
             return;
@@ -2430,6 +2430,10 @@ class NodeExecutor {
         if (runContextIds) {
             this.graph.addNodeRunContextIds(nodeId, runContextIds);
         }
+        if (!canSkipHandleResolution &&
+            !this.handleResolver.getPendingResolution(nodeId)) {
+            this.handleResolver.scheduleRecomputeHandles(nodeId);
+        }
         // Check if handles are being resolved - wait for resolution before executing
         // Do this AFTER setting up run contexts so handle resolution can track them
         const pendingResolution = this.handleResolver.getPendingResolution(nodeId);
@@ -2444,7 +2448,7 @@ class NodeExecutor {
                 // Re-check node still exists and conditions
                 const nodeAfter = this.graph.getNode(nodeId);
                 if (nodeAfter) {
-                    this.execute(nodeId, runContextIds);
+                    this.execute(nodeId, runContextIds, true);
                 }
                 if (runContextIds && runContextIds.size > 0) {
                     for (const id of runContextIds) {
@@ -3322,6 +3326,7 @@ class GraphRuntime {
         if (!fromNode?.outputs)
             return;
         this.hydrate({ outputs: { [toNodeId]: { ...fromNode.outputs } } }, { invalidate: !options?.dry });
+        this.handleResolver.scheduleRecomputeHandles(toNodeId);
     }
     hydrate(payload, opts) {
         const releasePause = this.requestPause();
@@ -3638,8 +3643,8 @@ class GraphRuntime {
         });
         this.graph.clear();
     }
-    execute(nodeId, runContextIds) {
-        this.nodeExecutor.execute(nodeId, runContextIds);
+    execute(nodeId, runContextIds, resolveHandles) {
+        this.nodeExecutor.execute(nodeId, runContextIds, resolveHandles);
     }
     propagate(srcNodeId, srcHandle, value, runContextIds) {
         this.edgePropagator.propagate(srcNodeId, srcHandle, value, runContextIds);