bitfab 0.12.0 → 0.12.2

package/dist/{chunk-QUCK3IU2.js → chunk-3HBV4GQO.js}

@@ -8,7 +8,7 @@ import {
   getReplayContext,
   isAsyncStorageInitDone,
   serializeValue
-} from "./chunk-VMJPNYAG.js";
+} from "./chunk-HUSTKOQI.js";
 
 // src/claudeAgentSdk.ts
 function nowIso() {
@@ -1999,6 +1999,15 @@ var Bitfab = class {
     const options = typeof optionsOrFn === "function" ? {} : optionsOrFn;
     const fn = typeof optionsOrFn === "function" ? optionsOrFn : maybeFn;
     const self = this;
+    const fnIsAsyncFunction = fn.constructor.name === "AsyncFunction";
+    const fnReturnsPromise = fnIsAsyncFunction || (() => {
+      try {
+        const src = fn.toString();
+        return /\b(?:Promise|await)\b/.test(src);
+      } catch {
+        return false;
+      }
+    })();
     const wrappedFn = function(...args) {
       if (!asyncLocalStorage && !isAsyncStorageInitDone()) {
         return asyncLocalStorageReady.then(
@@ -2092,11 +2101,12 @@ var Bitfab = class {
       const replayCtxForMock = getReplayContext();
       if (replayCtxForMock?.mockTree && !isRootSpan) {
         const counters = replayCtxForMock.callCounters;
-        const callIndex = counters.get(traceFunctionKey) ?? 0;
-        counters.set(traceFunctionKey, callIndex + 1);
+        const counterKey = `${traceFunctionKey}:${baseSpanParams.spanName}`;
+        const callIndex = counters.get(counterKey) ?? 0;
+        counters.set(counterKey, callIndex + 1);
         const shouldMock = replayCtxForMock.mockStrategy === "all" || replayCtxForMock.mockStrategy === "marked" && options.mockOnReplay === true;
         if (shouldMock) {
-          const mockKey = `${traceFunctionKey}:${callIndex}`;
+          const mockKey = `${counterKey}:${callIndex}`;
           const mockSpan = replayCtxForMock.mockTree.spans.get(mockKey);
           if (mockSpan) {
             let output = mockSpan.output;
@@ -2107,6 +2117,9 @@ var Bitfab = class {
               });
             }
             void sendSpan({ result: output });
+            if (fnReturnsPromise) {
+              return Promise.resolve(output);
+            }
             return output;
           }
         }
@@ -2252,7 +2265,7 @@ var Bitfab = class {
    * @returns ReplayResult with items, testRunId, and testRunUrl
    */
   async replay(traceFunctionKey, fn, options) {
-    const { replay: doReplay } = await import("./replay-6FKD2UEU.js");
+    const { replay: doReplay } = await import("./replay-RJZGLEHI.js");
     return doReplay(
       this.httpClient,
       this.serviceUrl,
@@ -2320,4 +2333,4 @@ export {
   Bitfab,
   BitfabFunction
 };
-//# sourceMappingURL=chunk-QUCK3IU2.js.map
+//# sourceMappingURL=chunk-3HBV4GQO.js.map