@workglow/task-graph 0.2.18 → 0.2.20

package/dist/bun.js

@@ -1436,6 +1436,7 @@ class TaskRunner {
   timeoutTimer;
   pendingTimeoutError;
   shouldAccumulate = true;
+  runWithPreviews = false;
   telemetrySpan;
   constructor(task) {
     this.task = task;
@@ -1718,6 +1719,7 @@ class TaskRunner {
       this.outputCache = cache;
     }
     this.shouldAccumulate = config.shouldAccumulate !== false;
+    this.runWithPreviews = config.runWithPreviews === true;
     if (config.updateProgress) {
       this.updateProgress = config.updateProgress;
     }
@@ -2889,6 +2891,7 @@ class TaskGraphRunner {
   graph;
   outputCache;
   accumulateLeafOutputs = true;
+  runWithPreviews = false;
   registry = globalServiceRegistry3;
   resourceScope;
   abortController;
@@ -3132,12 +3135,13 @@ class TaskGraphRunner {
         consumerCounts.set(port, (consumerCounts.get(port) ?? 0) + 1);
       }
       for (const [port, count] of consumerCounts) {
-        if (count <= 1)
+        const extra = this.runWithPreviews ? count : count - 1;
+        if (extra <= 0)
           continue;
         const value = results[port];
         const ref = asRefcountable(value);
         if (ref)
-          ref.retain(count - 1);
+          ref.retain(extra);
       }
     }
     for (const dataflow of dataflows) {
@@ -3309,7 +3313,8 @@ class TaskGraphRunner {
       outputCache: this.outputCache ?? false,
       updateProgress: async (task2, progress, message, ...args) => await this.handleProgress(task2, progress, message, ...args),
       registry: this.registry,
-      resourceScope: this.resourceScope
+      resourceScope: this.resourceScope,
+      runWithPreviews: this.runWithPreviews
     });
     await this.pushOutputFromNodeToEdges(task, results);
     return {
@@ -3359,7 +3364,8 @@ class TaskGraphRunner {
         shouldAccumulate,
         updateProgress: async (task2, progress, message, ...args) => await this.handleProgress(task2, progress, message, ...args),
         registry: this.registry,
-        resourceScope: this.resourceScope
+        resourceScope: this.resourceScope,
+        runWithPreviews: this.runWithPreviews
       });
       await this.pushOutputFromNodeToEdges(task, results);
       return {
@@ -3434,6 +3440,17 @@ class TaskGraphRunner {
     }
   }
   resetTask(graph, task, runId) {
+    const previous = task.runOutputData;
+    if (previous) {
+      for (const port of Object.keys(previous)) {
+        const ref = asRefcountable(previous[port]);
+        if (!ref)
+          continue;
+        try {
+          ref.release();
+        } catch {}
+      }
+    }
     task.status = TaskStatus.PENDING;
     task.resetInputData();
     task.runOutputData = {};
@@ -3467,6 +3484,7 @@ class TaskGraphRunner {
       this.resourceScope = config.resourceScope;
     }
     this.accumulateLeafOutputs = config?.accumulateLeafOutputs !== false;
+    this.runWithPreviews = config?.runWithPreviews === true;
     if (config?.outputCache !== undefined) {
       if (typeof config.outputCache === "boolean") {
         if (config.outputCache === true) {
@@ -3566,6 +3584,7 @@ class TaskGraphRunner {
         throw new TaskConfigurationError(`Graph has ${taskCount} tasks, exceeding the limit of ${config.maxTasks}`);
       }
     }
+    this.runWithPreviews = false;
     this.previewScheduler.reset();
     this.previewRunning = true;
   }
@@ -3665,7 +3684,8 @@ class GraphAsTaskRunner extends TaskRunner {
       parentSignal: this.abortController?.signal,
       outputCache: this.outputCache,
       registry: this.registry,
-      resourceScope: this.resourceScope
+      resourceScope: this.resourceScope,
+      runWithPreviews: this.runWithPreviews
     });
     unsubscribe();
     return results;
@@ -4128,7 +4148,8 @@ class TaskGraph {
       registry: config?.registry,
       timeout: config?.timeout,
       maxTasks: config?.maxTasks,
-      resourceScope: config?.resourceScope
+      resourceScope: config?.resourceScope,
+      runWithPreviews: config?.runWithPreviews
     });
   }
   runPreview(input = {}, config = {}) {
@@ -4981,7 +5002,8 @@ class Workflow {
         parentSignal: this._abortController.signal,
         outputCache: this._outputCache,
         registry: config?.registry ?? this._registry,
-        resourceScope: config?.resourceScope
+        resourceScope: config?.resourceScope,
+        runWithPreviews: config?.runWithPreviews
       });
       const results = this.graph.mergeExecuteOutputsToRunOutput(output, PROPERTY_ARRAY);
       this.events.emit("complete");
@@ -8408,4 +8430,4 @@ export {
   BROWSER_GRANTS
 };
 
-//# debugId=791136A32061C4D664756E2164756E21
+//# debugId=AA782C3ACDE0213064756E2164756E21