bitfab 0.41.0 → 0.42.0

package/dist/index.cjs

@@ -51,7 +51,7 @@ var __version__, __packageName__;
 var init_version_generated = __esm({
   "src/version.generated.ts"() {
     "use strict";
-    __version__ = "0.41.0";
+    __version__ = "0.42.0";
     __packageName__ = "bitfab";
   }
 });
@@ -1346,7 +1346,7 @@ function sourceTraceIdOf(payload) {
   const id = rawTrace?.id;
   return typeof id === "string" ? id : void 0;
 }
-var REPLAY_DB_BRANCH_REQUEST_TIMEOUT_MS, OTLP_TRACES_ENDPOINT, RETRYABLE_STATUSES, EXIT_FLUSH_TIMEOUT_MS, DEFAULT_LIFECYCLE_TIMEOUT_MS2, pendingTracePromises, carrierSeq, HttpClient;
+var REPLAY_DB_BRANCH_REQUEST_TIMEOUT_MS, REPLAY_COMPLETE_REQUEST_TIMEOUT_MS, OTLP_TRACES_ENDPOINT, RETRYABLE_STATUSES, EXIT_FLUSH_TIMEOUT_MS, DEFAULT_LIFECYCLE_TIMEOUT_MS2, pendingTracePromises, carrierSeq, HttpClient;
 var init_http = __esm({
   "src/http.ts"() {
     "use strict";
@@ -1360,6 +1360,7 @@ var init_http = __esm({
     init_unrefTimer();
     init_warnOnce();
     REPLAY_DB_BRANCH_REQUEST_TIMEOUT_MS = 3e5;
+    REPLAY_COMPLETE_REQUEST_TIMEOUT_MS = 12e4;
     OTLP_TRACES_ENDPOINT = "/api/sdk/otel/v1/traces";
     RETRYABLE_STATUSES = /* @__PURE__ */ new Set([429, 500, 502, 503, 504]);
     EXIT_FLUSH_TIMEOUT_MS = 5e3;
@@ -1849,7 +1850,7 @@ var init_http = __esm({
        * Start a replay session by fetching historical traces.
        * Blocking call - creates a test run and returns lightweight item references.
        */
-      async startReplay(traceFunctionKey, limit, traceIds, name, codeChangeDescription, codeChangeFiles, includeDbBranchLease, experimentGroupId, datasetId, graderIds, dbBranchSettings) {
+      async startReplay(traceFunctionKey, limit, traceIds, name, codeChangeDescription, codeChangeFiles, includeDbBranchLease, experimentGroupId, datasetId, graderIds, dbBranchSettings, attempts) {
         const payload = { traceFunctionKey };
         if (limit !== void 0) {
           payload.limit = limit;
@@ -1882,6 +1883,9 @@ var init_http = __esm({
         if (dbBranchSettings !== void 0) {
           payload.dbBranchSettings = dbBranchSettings;
         }
+        if (attempts !== void 0 && attempts > 1) {
+          payload.attempts = attempts;
+        }
         const timeout = includeDbBranchLease ? REPLAY_DB_BRANCH_REQUEST_TIMEOUT_MS : 3e4;
         return this.request("/api/sdk/replay/start", payload, {
           timeout
@@ -1997,7 +2001,7 @@ var init_http = __esm({
         return this.request(
           "/api/sdk/replay/complete",
           { testRunId },
-          { timeout: 3e4 }
+          { timeout: REPLAY_COMPLETE_REQUEST_TIMEOUT_MS }
         );
       }
       /**
@@ -2006,10 +2010,15 @@ var init_http = __esm({
        * snapshot + preview branch and polls operations to readiness, which
        * can take seconds.
        */
-      async resolveDbBranchLease(testRunId, traceId, dbBranchSettings) {
+      async resolveDbBranchLease(testRunId, traceId, dbBranchSettings, attempt) {
         return this.request(
           "/api/sdk/replay/resolveDbBranchLease",
-          { testRunId, traceId, dbBranchSettings },
+          {
+            testRunId,
+            traceId,
+            dbBranchSettings,
+            ...attempt !== void 0 && attempt > 0 ? { attempt } : {}
+          },
           { timeout: REPLAY_DB_BRANCH_REQUEST_TIMEOUT_MS }
         );
       }
@@ -2445,6 +2454,23 @@ __export(replay_exports, {
   sleepForReplayPersistence: () => sleepForReplayPersistence,
   waitForReplayPersistence: () => waitForReplayPersistence
 });
+function expandReplayWork(serverItems, attempts) {
+  return Array.from(
+    { length: attempts },
+    (_, attempt) => serverItems.map((serverItem) => ({ serverItem, attempt }))
+  ).flat();
+}
+function resolveAttempts(attempts) {
+  if (attempts === void 0) {
+    return 1;
+  }
+  if (!Number.isInteger(attempts) || attempts < 1 || attempts > MAX_REPLAY_ATTEMPTS) {
+    throw new BitfabError(
+      `attempts must be an integer from 1 to ${MAX_REPLAY_ATTEMPTS} (got ${attempts}).`
+    );
+  }
+  return attempts;
+}
 function dbBranchEnabled(dbBranch) {
   return dbBranch !== void 0 && dbBranch !== false;
 }
@@ -2568,7 +2594,7 @@ function buildMockTree(rootNode) {
   }
   return { spans };
 }
-async function processItem(httpClient, serverItem, fn, testRunId, mockStrategy, resolvedOverrides, replayedTraceId, includeDbBranchLease, dbBranchSettings, adaptInputs, dryRun) {
+async function processItem(httpClient, serverItem, fn, testRunId, mockStrategy, resolvedOverrides, replayedTraceId, attempt, includeDbBranchLease, dbBranchSettings, adaptInputs, dryRun) {
   let lease = includeDbBranchLease ? serverItem.dbBranchLease : void 0;
   let leaseError = includeDbBranchLease ? serverItem.dbBranchLeaseError : void 0;
   let dbSnapshotRef = serverItem.dbSnapshotRef;
@@ -2590,7 +2616,8 @@ async function processItem(httpClient, serverItem, fn, testRunId, mockStrategy,
         resolved = await httpClient.resolveDbBranchLease(
           testRunId,
           originalTraceId,
-          dbBranchSettings
+          dbBranchSettings,
+          attempt
         );
       } catch (cause) {
         throw new DbBranchReplayError(
@@ -2670,6 +2697,7 @@ async function processItem(httpClient, serverItem, fn, testRunId, mockStrategy,
         originalSpanId,
         sourceTraceId: originalTraceId,
         sourceSpanId: originalSpanId,
+        attempt,
         input: inputs,
         result: void 0,
         originalOutput,
@@ -2698,6 +2726,7 @@ async function processItem(httpClient, serverItem, fn, testRunId, mockStrategy,
           inputSourceSpanId: span.id,
           inputSourceTraceId: span.externalTraceId,
           sourceBitfabTraceId: originalTraceId,
+          replayAttempt: attempt,
           mockTree,
           callCounters: mockTree ? /* @__PURE__ */ new Map() : void 0,
           mockStrategy,
@@ -2749,6 +2778,7 @@ async function processItem(httpClient, serverItem, fn, testRunId, mockStrategy,
     // Deprecated aliases for originalTraceId/originalSpanId.
     sourceTraceId: originalTraceId,
     sourceSpanId: originalSpanId,
+    attempt,
     input: inputs,
     result,
     originalOutput,
@@ -2877,6 +2907,7 @@ async function replay(httpClient, serviceUrl, traceFunctionKey, fn, options, reg
     } catch {
     }
   }
+  const attempts = resolveAttempts(options?.attempts);
   await replayContextReady;
   let codeChangeDescription = options?.codeChangeDescription;
   let codeChangeFiles = options?.codeChangeFiles;
@@ -2909,7 +2940,8 @@ async function replay(httpClient, serviceUrl, traceFunctionKey, fn, options, reg
     options?.experimentGroupId,
     options?.datasetId,
     options?.graderIds,
-    resolveDbBranchSettings(options?.dbBranch)
+    resolveDbBranchSettings(options?.dbBranch),
+    attempts
   );
   if (serverItems.length === 0) {
     try {
@@ -2926,10 +2958,11 @@ async function replay(httpClient, serviceUrl, traceFunctionKey, fn, options, reg
     ...normalizeMockOverrides(options?.mockOverride),
     ...registeredOverrides
   ];
-  const replayedTraceIds = serverItems.map(() => randomUuid());
+  const workItems = expandReplayWork(serverItems, attempts);
+  const replayedTraceIds = workItems.map(() => randomUuid());
   httpClient.trackTraceDeliveries(replayedTraceIds);
-  const tasks = serverItems.map(
-    (serverItem, index) => () => processItem(
+  const tasks = workItems.map(
+    ({ serverItem, attempt }, index) => () => processItem(
       httpClient,
       serverItem,
       fn,
@@ -2937,6 +2970,7 @@ async function replay(httpClient, serviceUrl, traceFunctionKey, fn, options, reg
       mockStrategy,
       resolvedOverrides,
       replayedTraceIds[index],
+      attempt,
       dbBranchEnabled(options?.dbBranch) && options?.dryRun !== true,
       resolveDbBranchSettings(options?.dbBranch),
       options?.adaptInputs,
@@ -3004,6 +3038,7 @@ async function replay(httpClient, serviceUrl, traceFunctionKey, fn, options, reg
             // Deprecated aliases for originalTraceId/originalSpanId.
             sourceTraceId: item.originalTraceId ?? null,
             sourceSpanId: item.originalSpanId ?? null,
+            attempt: item.attempt,
             input: item.input,
             result: item.result,
             originalOutput: item.originalOutput,
@@ -3025,7 +3060,7 @@ async function replay(httpClient, serviceUrl, traceFunctionKey, fn, options, reg
     },
     options?.onItemStart ? (index) => {
       started += 1;
-      const serverItem = serverItems[index];
+      const { serverItem, attempt } = workItems[index];
       const originalTraceId = serverItem.originalTraceId ?? serverItem.sourceTraceId;
       const originalSpanId = serverItem.originalSpanId ?? serverItem.sourceSpanId;
       try {
@@ -3041,7 +3076,8 @@ async function replay(httpClient, serviceUrl, traceFunctionKey, fn, options, reg
             originalTraceId,
             originalSpanId,
             sourceTraceId: originalTraceId,
-            sourceSpanId: originalSpanId
+            sourceSpanId: originalSpanId,
+            attempt
           }
         });
       } catch {
@@ -3053,7 +3089,8 @@ async function replay(httpClient, serviceUrl, traceFunctionKey, fn, options, reg
     const dryResult = {
       items: resultItems,
       testRunId,
-      testRunUrl: fullTestRunUrl
+      testRunUrl: fullTestRunUrl,
+      attempts
     };
     await writeReplayResultFile(dryResult);
     return dryResult;
@@ -3122,7 +3159,8 @@ async function replay(httpClient, serviceUrl, traceFunctionKey, fn, options, reg
   const result = {
     items: resultItems,
     testRunId,
-    testRunUrl: fullTestRunUrl
+    testRunUrl: fullTestRunUrl,
+    attempts
   };
   await writeReplayResultFile(result);
   if (!options?.onItemFinish) {
@@ -3163,7 +3201,7 @@ async function writeReplayResultFile(result) {
     }
   }
 }
-var REPLAY_PERSISTENCE_TIMEOUT_MS, BITFAB_PROGRESS_PREFIX, ReplayError, DbBranchReplayError;
+var REPLAY_PERSISTENCE_TIMEOUT_MS, MAX_REPLAY_ATTEMPTS, BITFAB_PROGRESS_PREFIX, ReplayError, DbBranchReplayError;
 var init_replay = __esm({
   "src/replay.ts"() {
     "use strict";
@@ -3175,6 +3213,7 @@ var init_replay = __esm({
     init_replayContext();
     init_serialize();
     REPLAY_PERSISTENCE_TIMEOUT_MS = 3e4;
+    MAX_REPLAY_ATTEMPTS = 100;
     BITFAB_PROGRESS_PREFIX = "@@bitfab:progress ";
     ReplayError = class extends BitfabError {
       constructor(message, items, testRunId, testRunUrl, cause) {
@@ -7021,6 +7060,9 @@ var Bitfab = class {
             ...replayCtxAtStart?.inputSourceTraceId && {
               inputSourceTraceId: replayCtxAtStart.inputSourceTraceId
             },
+            ...replayCtxAtStart?.replayAttempt !== void 0 && {
+              replayAttempt: replayCtxAtStart.replayAttempt
+            },
             dbSnapshotRef
           });
           registeredTraceId = traceId;
@@ -7071,6 +7113,7 @@ var Bitfab = class {
                 contexts: traceState?.contexts ?? [],
                 testRunId: traceState?.testRunId,
                 inputSourceTraceId: traceState?.inputSourceTraceId,
+                replayAttempt: traceState?.replayAttempt,
                 dbSnapshotRef: traceState?.dbSnapshotRef,
                 dropped: traceState?.dropped,
                 ingestionType: traceState?.ingestionType,
@@ -7468,6 +7511,9 @@ var Bitfab = class {
     if (params.inputSourceTraceId) {
       rawTrace.input_source_trace_id = params.inputSourceTraceId;
     }
+    if (params.replayAttempt !== void 0) {
+      rawTrace.replay_attempt = params.replayAttempt;
+    }
     if (params.dbSnapshotRef) {
       rawTrace.db_snapshot_ref = params.dbSnapshotRef;
     }