@bitfab/sdk 0.33.1 → 0.33.3

package/dist/index.cjs

@@ -315,14 +315,25 @@ function runWithReplayContext(ctx, fn) {
   }
   return fn();
 }
-var replayContextStorage, replayContextReady;
+var replayContextStorage, REPLAY_CONTEXT_STORAGE_SYMBOL, replayContextReady;
 var init_replayContext = __esm({
   "src/replayContext.ts"() {
     "use strict";
     init_asyncStorage();
     replayContextStorage = null;
+    REPLAY_CONTEXT_STORAGE_SYMBOL = /* @__PURE__ */ Symbol.for("bitfab.replayContextStorage");
     replayContextReady = asyncStorageReady.then(() => {
-      replayContextStorage = createAsyncLocalStorage();
+      const shared = globalThis;
+      const existing = shared[REPLAY_CONTEXT_STORAGE_SYMBOL];
+      if (existing) {
+        replayContextStorage = existing;
+        return;
+      }
+      const created = createAsyncLocalStorage();
+      if (created) {
+        shared[REPLAY_CONTEXT_STORAGE_SYMBOL] = created;
+        replayContextStorage = created;
+      }
     });
   }
 });
@@ -603,9 +614,10 @@ function buildMockTree(rootNode) {
   }
   return { spans };
 }
-async function processItem(httpClient, serverItem, fn, testRunId, mockStrategy, resolvedOverrides, replayedTraceId, includeDbBranchLease, adaptInputs) {
-  const lease = includeDbBranchLease ? serverItem.dbBranchLease : void 0;
-  const leaseError = includeDbBranchLease ? serverItem.dbBranchLeaseError : void 0;
+async function processItem(httpClient, serverItem, fn, testRunId, mockStrategy, resolvedOverrides, replayedTraceId, includeDbBranchLease, dbBranchSettings, adaptInputs) {
+  let lease = includeDbBranchLease ? serverItem.dbBranchLease : void 0;
+  let leaseError = includeDbBranchLease ? serverItem.dbBranchLeaseError : void 0;
+  let dbSnapshotRef = serverItem.dbSnapshotRef;
   let inputs = [];
   let originalOutput;
   let result;
@@ -614,6 +626,16 @@ async function processItem(httpClient, serverItem, fn, testRunId, mockStrategy,
   const originalTraceId = serverItem.originalTraceId ?? serverItem.sourceTraceId;
   const originalSpanId = serverItem.originalSpanId ?? serverItem.sourceSpanId;
   try {
+    if (includeDbBranchLease && !lease && !leaseError) {
+      const resolved = await httpClient.resolveDbBranchLease(
+        testRunId,
+        originalTraceId,
+        dbBranchSettings
+      );
+      lease = resolved.lease ?? void 0;
+      leaseError = resolved.leaseError ?? void 0;
+      dbSnapshotRef = resolved.dbSnapshotRef ?? dbSnapshotRef;
+    }
     if (leaseError) {
       throw new BitfabError(
         `Replay requested a database branch for trace ${originalTraceId} but it could not be resolved (${leaseError.code}): ${leaseError.message}. The function was not run, because replaying it against the live database would produce a result that looks valid but did not use the historical data you asked for.`
@@ -725,7 +747,7 @@ async function processItem(httpClient, serverItem, fn, testRunId, mockStrategy,
     // Null here (and on older servers) means "replay tokens not known".
     tokens: null,
     model: serverItem.model ?? null,
-    dbSnapshotRef: serverItem.dbSnapshotRef ?? null
+    dbSnapshotRef: dbSnapshotRef ?? null
   };
 }
 async function mapWithConcurrency(tasks, maxConcurrency, onSettled) {
@@ -768,11 +790,13 @@ async function replay(httpClient, serviceUrl, traceFunctionKey, fn, options, reg
   await replayContextReady;
   let codeChangeDescription = options?.codeChangeDescription;
   let codeChangeFiles = options?.codeChangeFiles;
-  if (codeChangeDescription === void 0 && codeChangeFiles === void 0) {
+  if (codeChangeFiles === void 0) {
     const captured = await resolveAutoCodeChange(options?.name);
     if (captured) {
-      codeChangeDescription = captured.description;
       codeChangeFiles = captured.files;
+      if (codeChangeDescription === void 0) {
+        codeChangeDescription = captured.description;
+      }
     }
   }
   const {
@@ -812,6 +836,7 @@ async function replay(httpClient, serviceUrl, traceFunctionKey, fn, options, reg
       resolvedOverrides,
       replayedTraceIds[index],
       options?.dbBranch !== void 0,
+      resolveDbBranchSettings(options?.dbBranch),
       options?.adaptInputs
     )
   );
@@ -980,7 +1005,7 @@ __export(index_exports, {
 module.exports = __toCommonJS(index_exports);
 
 // src/version.generated.ts
-var __version__ = "0.33.1";
+var __version__ = "0.33.3";
 
 // src/constants.ts
 var DEFAULT_SERVICE_URL = "https://bitfab.ai";
@@ -998,6 +1023,7 @@ function unrefTimer(timer) {
 
 // src/http.ts
 init_warnOnce();
+var REPLAY_DB_BRANCH_REQUEST_TIMEOUT_MS = 3e5;
 function serializePayloadBody(payload) {
   try {
     return { body: JSON.stringify(payload), dropped: [] };
@@ -1358,6 +1384,7 @@ var HttpClient = class {
     }
     if (includeDbBranchLease) {
       payload.includeDbBranchLease = true;
+      payload.lazyDbBranchLease = true;
     }
     if (experimentGroupId !== void 0) {
       payload.experimentGroupId = experimentGroupId;
@@ -1371,7 +1398,7 @@ var HttpClient = class {
     if (dbBranchSettings !== void 0) {
       payload.dbBranchSettings = dbBranchSettings;
     }
-    const timeout = includeDbBranchLease ? 3e5 : 3e4;
+    const timeout = includeDbBranchLease ? REPLAY_DB_BRANCH_REQUEST_TIMEOUT_MS : 3e4;
     return this.request("/api/sdk/replay/start", payload, {
       timeout
     });
@@ -1470,11 +1497,11 @@ var HttpClient = class {
    * snapshot + preview branch and polls operations to readiness, which
    * can take seconds.
    */
-  async resolveDbBranchLease(testRunId, traceId, dbSnapshotRef) {
+  async resolveDbBranchLease(testRunId, traceId, dbBranchSettings) {
     return this.request(
       "/api/sdk/replay/resolveDbBranchLease",
-      { testRunId, traceId, dbSnapshotRef },
-      { timeout: 9e4 }
+      { testRunId, traceId, dbBranchSettings },
+      { timeout: REPLAY_DB_BRANCH_REQUEST_TIMEOUT_MS }
     );
   }
   /** Release a previously-resolved DB branch by deleting its Neon branch. Idempotent server-side. */
@@ -3471,8 +3498,22 @@ init_warnOnce();
 var activeTraceStates = /* @__PURE__ */ new Map();
 var pendingSpanPromises = /* @__PURE__ */ new Map();
 var asyncLocalStorage = null;
+var SPAN_CONTEXT_STORAGE_SYMBOL = /* @__PURE__ */ Symbol.for("bitfab.spanContextStorage");
 var initializeAsyncContext = () => {
-  asyncLocalStorage ?? (asyncLocalStorage = createAsyncLocalStorage());
+  if (asyncLocalStorage) {
+    return;
+  }
+  const shared = globalThis;
+  const existing = shared[SPAN_CONTEXT_STORAGE_SYMBOL];
+  if (existing) {
+    asyncLocalStorage = existing;
+    return;
+  }
+  const created = createAsyncLocalStorage();
+  if (created) {
+    shared[SPAN_CONTEXT_STORAGE_SYMBOL] = created;
+    asyncLocalStorage = created;
+  }
 };
 var asyncLocalStorageReady = asyncStorageReady.then(() => {
   initializeAsyncContext();