bitfab 0.33.1 → 0.33.2

package/dist/index.d.cts

@@ -1963,7 +1963,7 @@ declare class BitfabFunction {
 /**
  * SDK version from package.json (injected at build time)
  */
-declare const __version__ = "0.33.1";
+declare const __version__ = "0.33.2";
 
 /**
  * Constants for the Bitfab SDK.

package/dist/index.d.ts

@@ -1963,7 +1963,7 @@ declare class BitfabFunction {
 /**
  * SDK version from package.json (injected at build time)
  */
-declare const __version__ = "0.33.1";
+declare const __version__ = "0.33.2";
 
 /**
  * Constants for the Bitfab SDK.

package/dist/index.js

@@ -23,12 +23,12 @@ import {
   getCurrentReplayBranch,
   getCurrentSpan,
   getCurrentTrace
-} from "./chunk-7UHJ67MH.js";
+} from "./chunk-A3WLV5VS.js";
 import {
   BITFAB_PROGRESS_PREFIX,
   BitfabError,
   reportReplayProgress
-} from "./chunk-ADUWXWWI.js";
+} from "./chunk-GT2GSQM2.js";
 export {
   BITFAB_PROGRESS_PREFIX,
   Bitfab,

package/dist/node.cjs

@@ -331,14 +331,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;
+      }
     });
   }
 });
@@ -619,9 +630,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;
@@ -630,6 +642,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.`
@@ -741,7 +763,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) {
@@ -828,6 +850,7 @@ async function replay(httpClient, serviceUrl, traceFunctionKey, fn, options, reg
       resolvedOverrides,
       replayedTraceIds[index],
       options?.dbBranch !== void 0,
+      resolveDbBranchSettings(options?.dbBranch),
       options?.adaptInputs
     )
   );
@@ -1003,7 +1026,7 @@ registerAsyncLocalStorageClass(
 );
 
 // src/version.generated.ts
-var __version__ = "0.33.1";
+var __version__ = "0.33.2";
 
 // src/constants.ts
 var DEFAULT_SERVICE_URL = "https://bitfab.ai";
@@ -1021,6 +1044,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: [] };
@@ -1381,6 +1405,7 @@ var HttpClient = class {
     }
     if (includeDbBranchLease) {
       payload.includeDbBranchLease = true;
+      payload.lazyDbBranchLease = true;
     }
     if (experimentGroupId !== void 0) {
       payload.experimentGroupId = experimentGroupId;
@@ -1394,7 +1419,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
     });
@@ -1493,11 +1518,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. */
@@ -3494,8 +3519,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();