@bian-womp/spark-graph 0.3.66 → 0.3.67

package/lib/cjs/index.cjs

@@ -2961,11 +2961,15 @@ class NodeExecutor {
         this.edgePropagator = edgePropagator;
         this.runtime = runtime;
         this.environment = {};
+        this.customNodeData = {};
         this.environment = environment ?? {};
     }
     setEnvironment(environment) {
         this.environment = environment;
     }
+    setCustomNodeData(customNodeData) {
+        this.customNodeData = customNodeData;
+    }
     /**
      * Compute effective inputs for a node by merging real inputs with defaults
      */
@@ -3034,6 +3038,7 @@ class NodeExecutor {
             },
             getInput: (handle) => inputs[handle],
             environment: this.environment,
+            customNodeData: this.customNodeData[nodeId],
             runId,
             abortSignal,
             reportProgress,
@@ -3652,6 +3657,7 @@ class GraphRuntime {
     constructor() {
         // State
         this.environment = {};
+        this.customNodeData = {};
         this.runMode = null;
         this.pauseRefCount = 0;
         this.persistentPauseToken = null;
@@ -3671,11 +3677,13 @@ class GraphRuntime {
         if (opts?.startPaused)
             gr.pause();
         gr.environment = opts?.environment ?? {};
+        gr.customNodeData = {};
         // Set registry and environment on components
         gr.graph.setRegistry(registry);
         gr.handleResolver.setRegistry(registry);
         gr.handleResolver.setEnvironment(gr.environment);
         gr.nodeExecutor.setEnvironment(gr.environment);
+        gr.nodeExecutor.setCustomNodeData(gr.customNodeData);
         gr.runtimeValidatorManager.setRegistry(registry);
         // Precompute per-node resolved handles (use def-provided overrides; do not compute dynamically here)
         const initial = gr.isPaused()
@@ -3897,6 +3905,13 @@ class GraphRuntime {
             this.handleResolver.scheduleRecomputeHandles(nodeId);
         }
     }
+    setCustomNodeData(customNodeData) {
+        this.customNodeData = { ...customNodeData };
+        this.nodeExecutor.setCustomNodeData(this.customNodeData);
+    }
+    getCustomNodeData() {
+        return { ...this.customNodeData };
+    }
     /**
      * Register a runtime validator that will be called before node execution.
      * Validators are called in registration order - if any returns true, execution is blocked.