bitfab 0.11.0 → 0.11.4

package/dist/index.cjs

@@ -35,7 +35,7 @@ var __version__;
 var init_version_generated = __esm({
   "src/version.generated.ts"() {
     "use strict";
-    __version__ = "0.11.0";
+    __version__ = "0.11.4";
   }
 });
 
@@ -434,7 +434,15 @@ async function processItem(httpClient, serverItem, fn, testRunId) {
   } catch (e) {
     error = e instanceof Error ? e.message : String(e);
   }
-  return { input: inputs, result, originalOutput, error };
+  return {
+    input: inputs,
+    result,
+    originalOutput,
+    error,
+    durationMs: serverItem.durationMs ?? null,
+    tokens: serverItem.tokens ?? null,
+    model: serverItem.model ?? null
+  };
 }
 async function mapWithConcurrency(tasks, maxConcurrency) {
   const results = new Array(tasks.length);
@@ -1981,6 +1989,65 @@ function runWithSpanStack(stack, fn) {
     throw error;
   }
 }
+function isAsyncGenerator(value) {
+  if (value === null || typeof value !== "object") {
+    return false;
+  }
+  const candidate = value;
+  return typeof candidate.next === "function" && typeof candidate.return === "function" && typeof candidate.throw === "function" && typeof candidate[Symbol.asyncIterator] === "function";
+}
+function wrapAsyncGenerator(source, spanStack, sendSpan) {
+  const yielded = [];
+  let returnValue;
+  let finalized = false;
+  const finalize = (errorMsg) => {
+    if (finalized) {
+      return;
+    }
+    finalized = true;
+    void sendSpan({
+      result: { yielded, return: returnValue },
+      ...errorMsg && { error: errorMsg }
+    });
+  };
+  const step = (method, arg) => runWithSpanStack(spanStack, () => {
+    const op = source[method];
+    return op.call(source, arg);
+  });
+  const handle = async (method, arg) => {
+    try {
+      const result = await step(method, arg);
+      if (result.done) {
+        returnValue = result.value;
+        finalize();
+      } else {
+        yielded.push(result.value);
+      }
+      return result;
+    } catch (error) {
+      finalize(error instanceof Error ? error.message : String(error));
+      throw error;
+    }
+  };
+  const wrapped = {
+    next(arg) {
+      return handle("next", arg);
+    },
+    return(value) {
+      return handle("return", value);
+    },
+    throw(err) {
+      return handle("throw", err);
+    },
+    [Symbol.asyncIterator]() {
+      return wrapped;
+    },
+    [Symbol.asyncDispose]() {
+      return handle("return", void 0).then(() => void 0);
+    }
+  };
+  return wrapped;
+}
 var cachedCollectorClass;
 async function loadCollectorClass() {
   if (cachedCollectorClass !== void 0) {
@@ -2473,10 +2540,14 @@ var Bitfab = class {
       const inputs = args;
       const startedAt = (/* @__PURE__ */ new Date()).toISOString();
       if (isRootSpan && !activeTraceStates.has(traceId)) {
+        const replayCtxAtRoot = getReplayContext();
         activeTraceStates.set(traceId, {
           traceId,
           startedAt,
-          contexts: []
+          contexts: [],
+          ...replayCtxAtRoot?.testRunId && {
+            testRunId: replayCtxAtRoot.testRunId
+          }
         });
         pendingSpanPromises.set(traceId, []);
       }
@@ -2523,7 +2594,8 @@ var Bitfab = class {
               endedAt,
               sessionId: traceState?.sessionId,
               metadata: traceState?.metadata,
-              contexts: traceState?.contexts ?? []
+              contexts: traceState?.contexts ?? [],
+              testRunId: traceState?.testRunId
             });
             activeTraceStates.delete(traceId);
           } else {
@@ -2551,6 +2623,9 @@ var Bitfab = class {
             throw error;
           });
         }
+        if (isAsyncGenerator(result)) {
+          return wrapAsyncGenerator(result, newStack, sendSpan);
+        }
         void sendSpan({ result });
         return result;
       };
@@ -2603,7 +2678,8 @@ var Bitfab = class {
       traceFunctionKey: params.traceFunctionKey,
       externalTrace: rawTrace,
       completed: true,
-      ...params.sessionId && { sessionId: params.sessionId }
+      ...params.sessionId && { sessionId: params.sessionId },
+      ...params.testRunId && { testRunId: params.testRunId }
     });
   }
   /**