bitfab 0.12.0 → 0.12.2

package/dist/index.d.cts

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

package/dist/index.d.ts

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

package/dist/index.js

@@ -6,13 +6,13 @@ import {
   BitfabOpenAITracingProcessor,
   getCurrentSpan,
   getCurrentTrace
-} from "./chunk-QUCK3IU2.js";
+} from "./chunk-3HBV4GQO.js";
 import {
   BitfabError,
   DEFAULT_SERVICE_URL,
   __version__,
   flushTraces
-} from "./chunk-VMJPNYAG.js";
+} from "./chunk-HUSTKOQI.js";
 export {
   Bitfab,
   BitfabClaudeAgentHandler,

package/dist/node.cjs

@@ -81,7 +81,7 @@ var __version__;
 var init_version_generated = __esm({
   "src/version.generated.ts"() {
     "use strict";
-    __version__ = "0.12.0";
+    __version__ = "0.12.2";
   }
 });
 
@@ -473,9 +473,11 @@ function buildMockTree(rootNode) {
   function walk(node) {
     const key = node.traceFunctionKey;
     if (key) {
-      const index = counters.get(key) ?? 0;
-      counters.set(key, index + 1);
-      spans.set(`${key}:${index}`, {
+      const name = node.spanName;
+      const counterKey = `${key}:${name}`;
+      const index = counters.get(counterKey) ?? 0;
+      counters.set(counterKey, index + 1);
+      spans.set(`${counterKey}:${index}`, {
         sourceSpanId: node.sourceSpanId,
         output: node.output,
         outputMeta: node.outputMeta
@@ -2617,6 +2619,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(
@@ -2710,11 +2721,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;
@@ -2725,6 +2737,9 @@ var Bitfab = class {
               });
             }
             void sendSpan({ result: output });
+            if (fnReturnsPromise) {
+              return Promise.resolve(output);
+            }
             return output;
           }
         }