@bitfab/sdk 0.13.4 → 0.13.6

package/dist/index.d.cts

@@ -394,8 +394,12 @@ interface ReplayOptions {
      * environment to pick up the per-trace branch URL.
      */
     environment?: ReplayEnvironment;
+    /** Group ID to associate this replay with an experiment group for live streaming in Studio. */
+    experimentGroupId?: string;
 }
 interface ReplayItem<T> {
+    /** Trace ID of the new trace created during replay. */
+    traceId: string | null;
     /** Deserialized inputs from the original trace. */
     input: unknown[];
     /** The result returned by the function during replay, or undefined on error. */
@@ -1006,7 +1010,7 @@ declare class BitfabFunction {
 /**
  * SDK version from package.json (injected at build time)
  */
-declare const __version__ = "0.13.4";
+declare const __version__ = "0.13.6";
 
 /**
  * Constants for the Bitfab SDK.

package/dist/index.d.ts

@@ -394,8 +394,12 @@ interface ReplayOptions {
      * environment to pick up the per-trace branch URL.
      */
     environment?: ReplayEnvironment;
+    /** Group ID to associate this replay with an experiment group for live streaming in Studio. */
+    experimentGroupId?: string;
 }
 interface ReplayItem<T> {
+    /** Trace ID of the new trace created during replay. */
+    traceId: string | null;
     /** Deserialized inputs from the original trace. */
     input: unknown[];
     /** The result returned by the function during replay, or undefined on error. */
@@ -1006,7 +1010,7 @@ declare class BitfabFunction {
 /**
  * SDK version from package.json (injected at build time)
  */
-declare const __version__ = "0.13.4";
+declare const __version__ = "0.13.6";
 
 /**
  * Constants for the Bitfab SDK.

package/dist/index.js

@@ -8,13 +8,13 @@ import {
   SUPPORTED_PROVIDERS,
   getCurrentSpan,
   getCurrentTrace
-} from "./chunk-XUW46356.js";
+} from "./chunk-6757PVPS.js";
 import {
   BitfabError,
   DEFAULT_SERVICE_URL,
   __version__,
   flushTraces
-} from "./chunk-SKJWF5VX.js";
+} from "./chunk-TMSVIA5J.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.13.4";
+    __version__ = "0.13.6";
   }
 });
 
@@ -312,7 +312,7 @@ var init_http = __esm({
        * Start a replay session by fetching historical traces.
        * Blocking call — creates a test run and returns lightweight item references.
        */
-      async startReplay(traceFunctionKey, limit, traceIds, codeChangeDescription, codeChangeFiles, includeDbBranchLease) {
+      async startReplay(traceFunctionKey, limit, traceIds, codeChangeDescription, codeChangeFiles, includeDbBranchLease, experimentGroupId) {
         const payload = { traceFunctionKey, limit };
         if (traceIds) {
           payload.traceIds = traceIds;
@@ -326,6 +326,9 @@ var init_http = __esm({
         if (includeDbBranchLease) {
           payload.includeDbBranchLease = true;
         }
+        if (experimentGroupId !== void 0) {
+          payload.experimentGroupId = experimentGroupId;
+        }
         const timeout = includeDbBranchLease ? 18e4 : 3e4;
         return this.request("/api/sdk/replay/start", payload, {
           timeout
@@ -579,6 +582,7 @@ async function processItem(httpClient, serverItem, fn, testRunId, mockStrategy,
   let originalOutput;
   let result;
   let error = null;
+  const replayedTraceId = crypto.randomUUID();
   try {
     const span = await httpClient.getExternalSpan(serverItem.externalSpanId);
     const spanData = span.rawData?.span_data ?? {};
@@ -594,6 +598,7 @@ async function processItem(httpClient, serverItem, fn, testRunId, mockStrategy,
     const maybePromise = runWithReplayContext(
       {
         testRunId,
+        traceId: replayedTraceId,
         inputSourceSpanId: span.id,
         inputSourceTraceId: span.externalTraceId,
         sourceBitfabTraceId: serverItem.traceId,
@@ -622,6 +627,7 @@ async function processItem(httpClient, serverItem, fn, testRunId, mockStrategy,
     }
   }
   return {
+    traceId: replayedTraceId,
     input: inputs,
     result,
     originalOutput,
@@ -660,8 +666,9 @@ async function replay(httpClient, serviceUrl, traceFunctionKey, fn, options) {
     options?.traceIds,
     options?.codeChangeDescription,
     options?.codeChangeFiles,
-    options?.environment !== void 0
+    options?.environment !== void 0,
     // includeDbBranchLease
+    options?.experimentGroupId
   );
   const mockStrategy = options?.mock ?? "none";
   const maxConcurrency = options?.maxConcurrency ?? 10;
@@ -677,14 +684,21 @@ async function replay(httpClient, serviceUrl, traceFunctionKey, fn, options) {
   );
   const resultItems = await mapWithConcurrency(tasks, maxConcurrency);
   await flushTraces();
+  let serverTraceIds = {};
   try {
-    await httpClient.completeReplay(testRunId);
+    const completeResult = await httpClient.completeReplay(testRunId);
+    serverTraceIds = completeResult.traceIds ?? {};
   } catch (e) {
     try {
       console.error("Bitfab: Failed to complete replay:", e);
     } catch {
     }
   }
+  for (const item of resultItems) {
+    if (item.traceId) {
+      item.traceId = serverTraceIds[item.traceId] ?? null;
+    }
+  }
   return {
     items: resultItems,
     testRunId,
@@ -2876,7 +2890,8 @@ var Bitfab = class {
       }
       const currentStack = getSpanStack();
       const parentContext = currentStack[currentStack.length - 1];
-      const traceId = parentContext?.traceId ?? crypto.randomUUID();
+      const replayCtxForTraceId = parentContext ? null : getReplayContext();
+      const traceId = parentContext?.traceId ?? replayCtxForTraceId?.traceId ?? crypto.randomUUID();
       const spanId = crypto.randomUUID();
       const parentSpanId = parentContext?.spanId ?? null;
       const isRootSpan = parentSpanId === null;