bitfab 0.24.0 → 0.25.0

package/dist/index.cjs

@@ -414,18 +414,23 @@ async function processItem(httpClient, serverItem, fn, testRunId, mockStrategy,
     originalOutput,
     error,
     durationMs: serverItem.durationMs ?? null,
-    tokens: serverItem.tokens ?? null,
+    // Filled in by replay() from the complete-replay response once the
+    // replay traces are persisted and their spans aggregated server-side.
+    // Null here (and on older servers) means "replay tokens not known".
+    tokens: null,
     model: serverItem.model ?? null,
     dbSnapshotRef: serverItem.dbSnapshotRef ?? null
   };
 }
-async function mapWithConcurrency(tasks, maxConcurrency) {
+async function mapWithConcurrency(tasks, maxConcurrency, onSettled) {
   const results = new Array(tasks.length);
   let nextIndex = 0;
   async function worker() {
     while (nextIndex < tasks.length) {
       const index = nextIndex++;
-      results[index] = await tasks[index]();
+      const result = await tasks[index]();
+      results[index] = result;
+      onSettled?.(result, index);
     }
   }
   const workers = Array.from(
@@ -485,9 +490,29 @@ async function replay(httpClient, serviceUrl, traceFunctionKey, fn, options) {
       options?.adaptInputs
     )
   );
-  const resultItems = await mapWithConcurrency(tasks, maxConcurrency);
+  const total = tasks.length;
+  let completed = 0;
+  let succeeded = 0;
+  let errored = 0;
+  const resultItems = await mapWithConcurrency(
+    tasks,
+    maxConcurrency,
+    options?.onProgress ? (item) => {
+      completed += 1;
+      if (item.error === null) {
+        succeeded += 1;
+      } else {
+        errored += 1;
+      }
+      try {
+        options?.onProgress?.({ completed, total, succeeded, errored });
+      } catch {
+      }
+    } : void 0
+  );
   const completeResult = await httpClient.completeReplay(testRunId);
   const serverTraceIds = completeResult.traceIds;
+  const replayTokens = completeResult.tokens;
   if (serverTraceIds === void 0) {
     try {
       console.warn(
@@ -510,6 +535,9 @@ async function replay(httpClient, serviceUrl, traceFunctionKey, fn, options) {
             missing.push(item.traceId);
           }
         }
+        if (mapped !== void 0) {
+          item.tokens = replayTokens?.[mapped] ?? null;
+        }
         item.traceId = mapped ?? null;
       }
     }
@@ -568,7 +596,7 @@ __export(index_exports, {
 module.exports = __toCommonJS(index_exports);
 
 // src/version.generated.ts
-var __version__ = "0.24.0";
+var __version__ = "0.25.0";
 
 // src/constants.ts
 var DEFAULT_SERVICE_URL = "https://bitfab.ai";