bitfab 0.61.6 → 0.62.1

package/dist/index.cjs

@@ -51,7 +51,7 @@ var __version__, __packageName__;
 var init_version_generated = __esm({
   "src/version.generated.ts"() {
     "use strict";
-    __version__ = "0.61.6";
+    __version__ = "0.62.1";
     __packageName__ = "bitfab";
   }
 });
@@ -3706,12 +3706,13 @@ function installReplayInterruptHandlers(target) {
   const removeHandlers = () => {
     for (const [signal, handler] of handlers) {
       host.removeListener(signal, handler);
+      replayListeners.delete(handler);
     }
     handlers.clear();
   };
   let interrupting = false;
   const handleSignal = async (signal) => {
-    const hostHandlesSignal = host.listenerCount(signal) > 1;
+    const hostHandlesSignal = host.listeners(signal).some((listener) => !replayListeners.has(listener));
     const markedInterrupted = await target.interrupt();
     removeHandlers();
     if (target.onInterrupt !== void 0) {
@@ -3740,11 +3741,12 @@ function installReplayInterruptHandlers(target) {
       void handleSignal(signal);
     };
     handlers.set(signal, handler);
+    replayListeners.add(handler);
     host.on(signal, handler);
   }
   return { remove: removeHandlers, interrupted: () => interrupting };
 }
-var REPLAY_INTERRUPT_SIGNALS;
+var REPLAY_INTERRUPT_SIGNALS, replayListeners;
 var init_replayInterrupt = __esm({
   "src/replayInterrupt.ts"() {
     "use strict";
@@ -3752,6 +3754,7 @@ var init_replayInterrupt = __esm({
       "SIGINT",
       "SIGTERM"
     ];
+    replayListeners = /* @__PURE__ */ new Set();
   }
 });
 
@@ -4136,8 +4139,6 @@ function carriedOverReplayItem(finished) {
     model: null,
     dbSnapshotRef: null,
     dbBranchTimings: null,
-    traceOutline: null,
-    originalTraceOutline: null,
     carriedOver: true
   };
 }
@@ -4430,9 +4431,7 @@ async function processItem(httpClient, serverItem, fn, experimentId, mockStrateg
         tokens: null,
         model: serverItem.originalModel ?? serverItem.model ?? null,
         dbSnapshotRef: dbSnapshotRef ?? null,
-        dbBranchTimings: dbBranchTimings ?? null,
-        traceOutline: null,
-        originalTraceOutline: null
+        dbBranchTimings: dbBranchTimings ?? null
       };
     }
     try {
@@ -4520,9 +4519,7 @@ async function processItem(httpClient, serverItem, fn, experimentId, mockStrateg
     tokens: null,
     model: originalModel,
     dbSnapshotRef: dbSnapshotRef ?? null,
-    dbBranchTimings: dbBranchTimings ?? null,
-    traceOutline: null,
-    originalTraceOutline: null
+    dbBranchTimings: dbBranchTimings ?? null
   };
 }
 async function waitForReplayPersistence(httpClient, experimentId, replayedTraceIds) {
@@ -4589,11 +4586,11 @@ function sleepForReplayPersistence(ms) {
     setTimeout(resolve, ms);
   });
 }
-async function mapWithConcurrency2(tasks, maxConcurrency, onSettled, onStarted) {
+async function mapWithConcurrency2(tasks, maxConcurrency, onSettled, onStarted, shouldStop) {
   const results = new Array(tasks.length);
   let nextIndex = 0;
   async function worker() {
-    while (nextIndex < tasks.length) {
+    while (nextIndex < tasks.length && shouldStop?.() !== true) {
       const index = nextIndex++;
       onStarted?.(index);
       const result = await tasks[index]();
@@ -4606,7 +4603,7 @@ async function mapWithConcurrency2(tasks, maxConcurrency, onSettled, onStarted)
     () => worker()
   );
   await Promise.all(workers);
-  return results;
+  return results.slice(0, nextIndex);
 }
 async function replay(httpClient, serviceUrl, traceFunctionKey, fn, options, registeredOverrides = []) {
   if (options?.resume !== void 0) {
@@ -4890,9 +4887,7 @@ async function runReplaySession(httpClient, fn, session, fullExperimentUrl, maxC
             tokens: item.tokens,
             model: item.model,
             dbSnapshotRef: item.dbSnapshotRef,
-            dbBranchTimings: item.dbBranchTimings,
-            traceOutline: item.traceOutline,
-            originalTraceOutline: item.originalTraceOutline
+            dbBranchTimings: item.dbBranchTimings
           }
         });
       } catch {
@@ -4923,9 +4918,11 @@ async function runReplaySession(httpClient, fn, session, fullExperimentUrl, maxC
         });
       } catch {
       }
-    } : void 0
+    } : void 0,
+    interrupted
   );
   const resultItems = [...carriedOverItems, ...ranItems];
+  throwIfInterrupted(interrupted, resultItems, experimentId, fullExperimentUrl);
   if (options?.dryRun === true) {
     throwIfInterrupted(
       interrupted,
@@ -4987,9 +4984,7 @@ async function runReplaySession(httpClient, fn, session, fullExperimentUrl, maxC
       }
       if (mapped !== void 0) {
         item.tokens = replayTokens?.[mapped] ?? null;
-        item.traceOutline = completeResult.traceOutlines?.[mapped] ?? null;
       }
-      item.originalTraceOutline = completeResult.originalTraceOutlines?.[item.originalTraceId] ?? null;
     }
     if (completedCount > 0 && missing.length === completedCount) {
       const serverCount = completeResult.traceCount !== void 0 ? ` The server persisted ${completeResult.traceCount} trace(s) for this run.` : "";