@bitfab/sdk 0.36.12 → 0.38.0

package/dist/{chunk-VUYTATF5.js → chunk-SSNTMROV.js}

@@ -314,7 +314,7 @@ function encodeRequestBody(body) {
 }
 
 // src/version.generated.ts
-var __version__ = "0.36.12";
+var __version__ = "0.38.0";
 
 // src/constants.ts
 var DEFAULT_SERVICE_URL = "https://bitfab.ai";
@@ -2385,12 +2385,15 @@ async function processItem(httpClient, serverItem, fn, testRunId, mockStrategy,
   let lease = includeDbBranchLease ? serverItem.dbBranchLease : void 0;
   let leaseError = includeDbBranchLease ? serverItem.dbBranchLeaseError : void 0;
   let dbSnapshotRef = serverItem.dbSnapshotRef;
+  let dbBranchTimings = includeDbBranchLease ? serverItem.dbBranchTimings : void 0;
   let inputs = [];
   let originalOutput;
   let result;
   let error = null;
   let traceError = null;
   let replayError = null;
+  let replayDurationMs = null;
+  let replayStarted = null;
   const originalTraceId = serverItem.originalTraceId ?? serverItem.sourceTraceId;
   const originalSpanId = serverItem.originalSpanId ?? serverItem.sourceSpanId;
   try {
@@ -2413,6 +2416,7 @@ async function processItem(httpClient, serverItem, fn, testRunId, mockStrategy,
       lease = resolved.lease ?? void 0;
       leaseError = resolved.leaseError ?? void 0;
       dbSnapshotRef = resolved.dbSnapshotRef ?? dbSnapshotRef;
+      dbBranchTimings = resolved.timings ?? dbBranchTimings;
     }
     if (leaseError) {
       throw new DbBranchReplayError(
@@ -2446,20 +2450,17 @@ async function processItem(httpClient, serverItem, fn, testRunId, mockStrategy,
           includeOutputs,
           includeRootOutput: false
         });
-        if (treeResponse.root) {
-          mockTree = buildMockTree(treeResponse.root);
-        } else if (mockStrategy === "all" || hasOverrides) {
+        if (!treeResponse.root) {
           throw new BitfabError(
             `Replay mock strategy "${mockStrategy}"${hasOverrides ? " with overrides" : ""} requires a span tree root for original span ${originalSpanId}.`
           );
-        } else {
-          mockTree = void 0;
         }
-      } catch (e) {
-        if (mockStrategy === "all" || hasOverrides) {
-          throw e;
+        mockTree = buildMockTree(treeResponse.root);
+      } catch (error2) {
+        if (mockStrategy !== "marked" || hasOverrides) {
+          throw error2;
         }
-        mockTree = void 0;
+        mockTree = { spans: /* @__PURE__ */ new Map() };
       }
     }
     const outputCache = /* @__PURE__ */ new Map();
@@ -2476,6 +2477,7 @@ async function processItem(httpClient, serverItem, fn, testRunId, mockStrategy,
       return pending;
     } : void 0;
     try {
+      replayStarted = performance.now();
       const maybePromise = runWithReplayContext(
         {
           testRunId,
@@ -2488,16 +2490,24 @@ async function processItem(httpClient, serverItem, fn, testRunId, mockStrategy,
           mockStrategy,
           mockOverrides: hasOverrides ? resolvedOverrides : void 0,
           fetchSpanOutput,
-          dbBranchLease: lease
+          dbBranchLease: lease,
+          dbBranchTimings
         },
         () => fn(...inputs)
       );
       result = maybePromise instanceof Promise ? await maybePromise : maybePromise;
+      replayDurationMs = Math.round(performance.now() - replayStarted);
     } catch (e) {
+      if (replayStarted !== null) {
+        replayDurationMs = Math.round(performance.now() - replayStarted);
+      }
       traceError = e;
       error = errorMessage(e);
     }
   } catch (e) {
+    if (replayStarted !== null) {
+      replayDurationMs = Math.round(performance.now() - replayStarted);
+    }
     replayError = e;
     error = replayItemErrorMessage(e);
   } finally {
@@ -2514,6 +2524,8 @@ async function processItem(httpClient, serverItem, fn, testRunId, mockStrategy,
       }
     }
   }
+  const originalDurationMs = serverItem.originalDurationMs ?? serverItem.durationMs ?? null;
+  const originalModel = serverItem.originalModel ?? serverItem.model ?? null;
   return {
     // Written in by replay() from the complete-replay response once the server
     // has minted this replay trace's row. Null until then: the client-side
@@ -2530,13 +2542,17 @@ async function processItem(httpClient, serverItem, fn, testRunId, mockStrategy,
     error,
     traceError,
     replayError,
-    durationMs: serverItem.durationMs ?? null,
+    durationMs: replayDurationMs,
+    originalDurationMs,
+    originalTokens: serverItem.originalTokens ?? serverItem.tokens ?? null,
+    originalModel,
     // 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: dbSnapshotRef ?? null
+    model: originalModel,
+    dbSnapshotRef: dbSnapshotRef ?? null,
+    dbBranchTimings: dbBranchTimings ?? null
   };
 }
 async function waitForReplayPersistence(httpClient, testRunId, replayedTraceIds) {
@@ -2763,9 +2779,13 @@ async function replay(httpClient, serviceUrl, traceFunctionKey, fn, options, reg
             traceError: item.traceError,
             replayError: item.replayError,
             durationMs: item.durationMs,
+            originalDurationMs: item.originalDurationMs,
+            originalTokens: item.originalTokens,
+            originalModel: item.originalModel,
             tokens: item.tokens,
             model: item.model,
-            dbSnapshotRef: item.dbSnapshotRef
+            dbSnapshotRef: item.dbSnapshotRef,
+            dbBranchTimings: item.dbBranchTimings
           }
         });
       } catch {
@@ -2933,4 +2953,4 @@ export {
   sleepForReplayPersistence,
   replay
 };
-//# sourceMappingURL=chunk-VUYTATF5.js.map
+//# sourceMappingURL=chunk-SSNTMROV.js.map