@bitfab/sdk 0.38.2 → 0.38.4

package/dist/{chunk-GFBQ2AMO.js → chunk-J47KPS77.js}

@@ -28,6 +28,29 @@ function runWithAutoTraceContext(context, fn, depth = 0) {
     autoTraceState.browserScope = previous;
   }
 }
+function runWithAutoTraceNodeConfiguration(nodeConfiguration, fn) {
+  const scope = currentAutoTraceScope();
+  if (!scope) {
+    return fn();
+  }
+  const configuredScope = { ...scope, nodeConfiguration };
+  let result;
+  if (autoTraceState.storage) {
+    result = autoTraceState.storage.run(configuredScope, fn);
+  } else {
+    const previous = autoTraceState.browserScope;
+    autoTraceState.browserScope = configuredScope;
+    try {
+      result = fn();
+    } finally {
+      autoTraceState.browserScope = previous;
+    }
+  }
+  if (isAutoTraceAsyncGenerator(result)) {
+    return wrapAutoTraceNodeAsyncGenerator(nodeConfiguration, result);
+  }
+  return result;
+}
 function runWithAutoTraceRootContext(context, fn) {
   autoTraceState.activeRoots += 1;
   let result;
@@ -66,12 +89,34 @@ function wrapAutoTraceAsyncGenerator(context, source) {
   };
   return wrapped;
 }
+function wrapAutoTraceNodeAsyncGenerator(nodeConfiguration, source) {
+  const step = (method, value) => runWithAutoTraceNodeConfiguration(
+    nodeConfiguration,
+    () => source[method](value)
+  );
+  const wrapped = {
+    next: (value) => step("next", value),
+    return: (value) => step("return", value),
+    throw: (error) => step("throw", error),
+    [Symbol.asyncIterator]: () => wrapped
+  };
+  return wrapped;
+}
 function __bitfabAutoSpan(definition, inputs, fn) {
   const scope = currentAutoTraceScope();
   if (!scope) {
     return fn();
   }
-  return scope.context.invoke(definition, inputs, fn, scope.depth);
+  const nameParts = definition.name.split(".");
+  const simpleName = nameParts[nameParts.length - 1];
+  const nodeConfiguration = simpleName === scope.nodeConfiguration?.functionName ? scope.nodeConfiguration : void 0;
+  return scope.context.invoke(
+    definition,
+    inputs,
+    fn,
+    scope.depth,
+    nodeConfiguration
+  );
 }
 function __bitfabAutoWrap(definition, fn) {
   if (fn.name === "") {
@@ -119,6 +164,7 @@ function getAutoTraceCapturePolicy(client, traceFunctionKey) {
 
 export {
   runWithAutoTraceContext,
+  runWithAutoTraceNodeConfiguration,
   runWithAutoTraceRootContext,
   __bitfabAutoSpan,
   __bitfabAutoWrap,
@@ -126,4 +172,4 @@ export {
   __setBitfabAutoTraceCapturePolicy,
   getAutoTraceCapturePolicy
 };
-//# sourceMappingURL=chunk-GFBQ2AMO.js.map
+//# sourceMappingURL=chunk-J47KPS77.js.map

package/dist/chunk-J47KPS77.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/autoTrace.ts"],"sourcesContent":["import {\n  type AsyncLocalStorageLike,\n  createAsyncLocalStorage,\n} from \"./asyncStorage.js\"\n\nexport interface AutoTraceFunctionDefinition {\n  id: string\n  name: string\n  file: string\n  line: number\n  column: number\n  async?: boolean\n  wrapper?: boolean\n}\n\nexport interface AutoTraceNodeConfiguration {\n  functionName: string\n  name?: string\n  type?: \"llm\" | \"agent\" | \"function\" | \"guardrail\" | \"handoff\" | \"custom\"\n  capture: boolean\n  testRunId?: string\n  mockOnReplay?: boolean\n  // biome-ignore lint/suspicious/noExplicitAny: node finalizers receive the configured function's result, whose type is owned by the caller\n  finalize?: (result: any) => unknown | Promise<unknown>\n}\n\nexport interface AutoTraceContext {\n  invoke<T>(\n    definition: AutoTraceFunctionDefinition,\n    inputs: unknown[],\n    fn: () => T,\n    depth: number,\n    nodeConfiguration?: AutoTraceNodeConfiguration,\n  ): T\n}\n\ninterface AutoTraceScope {\n  context: AutoTraceContext\n  depth: number\n  nodeConfiguration?: AutoTraceNodeConfiguration\n}\n\ninterface AutoTraceState {\n  storage: AsyncLocalStorageLike<AutoTraceScope> | null\n  browserScope: AutoTraceScope | undefined\n  capturePolicies: WeakMap<object, Map<string, ReadonlySet<string>>>\n  activeRoots: number\n}\n\ninterface AutoTraceGlobal {\n  __bitfabAutoTraceStateV3?: AutoTraceState\n}\n\ninterface AutoTraceAsyncGenerator {\n  next(value?: unknown): Promise<IteratorResult<unknown, unknown>>\n  return(value?: unknown): Promise<IteratorResult<unknown, unknown>>\n  throw(error?: unknown): Promise<IteratorResult<unknown, unknown>>\n  [Symbol.asyncIterator](): AutoTraceAsyncGenerator\n}\n\nconst autoTraceGlobal = globalThis as unknown as AutoTraceGlobal\nconst autoTraceState: AutoTraceState =\n  autoTraceGlobal.__bitfabAutoTraceStateV3 ?? {\n    storage: null,\n    browserScope: undefined,\n    capturePolicies: new WeakMap(),\n    activeRoots: 0,\n  }\nautoTraceGlobal.__bitfabAutoTraceStateV3 = autoTraceState\n\nfunction initializeAutoTraceStorage(): void {\n  autoTraceState.storage ??= createAsyncLocalStorage<AutoTraceScope>()\n}\n\nexport function runWithAutoTraceContext<T>(\n  context: AutoTraceContext,\n  fn: () => T,\n  depth = 0,\n): T {\n  initializeAutoTraceStorage()\n  const scope = { context, depth }\n  if (autoTraceState.storage) {\n    return autoTraceState.storage.run(scope, fn)\n  }\n\n  const previous = autoTraceState.browserScope\n  autoTraceState.browserScope = scope\n  try {\n    return fn()\n  } finally {\n    autoTraceState.browserScope = previous\n  }\n}\n\nexport function runWithAutoTraceNodeConfiguration<T>(\n  nodeConfiguration: AutoTraceNodeConfiguration,\n  fn: () => T,\n): T {\n  const scope = currentAutoTraceScope()\n  if (!scope) {\n    return fn()\n  }\n\n  const configuredScope = { ...scope, nodeConfiguration }\n  let result: T\n  if (autoTraceState.storage) {\n    result = autoTraceState.storage.run(configuredScope, fn)\n  } else {\n    const previous = autoTraceState.browserScope\n    autoTraceState.browserScope = configuredScope\n    try {\n      result = fn()\n    } finally {\n      autoTraceState.browserScope = previous\n    }\n  }\n\n  if (isAutoTraceAsyncGenerator(result)) {\n    return wrapAutoTraceNodeAsyncGenerator(nodeConfiguration, result) as T\n  }\n  return result\n}\n\nexport function runWithAutoTraceRootContext<T>(\n  context: AutoTraceContext,\n  fn: () => T,\n): T {\n  autoTraceState.activeRoots += 1\n  let result: T\n  try {\n    result = runWithAutoTraceContext(context, fn)\n  } catch (error) {\n    autoTraceState.activeRoots -= 1\n    throw error\n  }\n\n  if (isAutoTraceAsyncGenerator(result)) {\n    autoTraceState.activeRoots -= 1\n    return wrapAutoTraceAsyncGenerator(context, result) as T\n  }\n\n  if (result instanceof Promise) {\n    return result.finally(() => {\n      autoTraceState.activeRoots -= 1\n    }) as T\n  }\n\n  autoTraceState.activeRoots -= 1\n  return result\n}\n\nfunction isAutoTraceAsyncGenerator(\n  value: unknown,\n): value is AutoTraceAsyncGenerator {\n  if (value === null || typeof value !== \"object\") {\n    return false\n  }\n  const candidate = value as Record<PropertyKey, unknown>\n  return (\n    typeof candidate.next === \"function\" &&\n    typeof candidate.return === \"function\" &&\n    typeof candidate.throw === \"function\" &&\n    typeof candidate[Symbol.asyncIterator] === \"function\"\n  )\n}\n\nfunction wrapAutoTraceAsyncGenerator(\n  context: AutoTraceContext,\n  source: AutoTraceAsyncGenerator,\n): AutoTraceAsyncGenerator {\n  const step = (\n    method: \"next\" | \"return\" | \"throw\",\n    value?: unknown,\n  ): Promise<IteratorResult<unknown, unknown>> =>\n    runWithAutoTraceRootContext(context, () => source[method](value))\n  const wrapped: AutoTraceAsyncGenerator = {\n    next: (value) => step(\"next\", value),\n    return: (value) => step(\"return\", value),\n    throw: (error) => step(\"throw\", error),\n    [Symbol.asyncIterator]: () => wrapped,\n  }\n  return wrapped\n}\n\nfunction wrapAutoTraceNodeAsyncGenerator(\n  nodeConfiguration: AutoTraceNodeConfiguration,\n  source: AutoTraceAsyncGenerator,\n): AutoTraceAsyncGenerator {\n  const step = (\n    method: \"next\" | \"return\" | \"throw\",\n    value?: unknown,\n  ): Promise<IteratorResult<unknown, unknown>> =>\n    runWithAutoTraceNodeConfiguration(nodeConfiguration, () =>\n      source[method](value),\n    )\n  const wrapped: AutoTraceAsyncGenerator = {\n    next: (value) => step(\"next\", value),\n    return: (value) => step(\"return\", value),\n    throw: (error) => step(\"throw\", error),\n    [Symbol.asyncIterator]: () => wrapped,\n  }\n  return wrapped\n}\n\nexport function __bitfabAutoSpan<T>(\n  definition: AutoTraceFunctionDefinition,\n  inputs: unknown[],\n  fn: () => T,\n): T {\n  const scope = currentAutoTraceScope()\n  if (!scope) {\n    return fn()\n  }\n  const nameParts = definition.name.split(\".\")\n  const simpleName = nameParts[nameParts.length - 1]\n  const nodeConfiguration =\n    simpleName === scope.nodeConfiguration?.functionName\n      ? scope.nodeConfiguration\n      : undefined\n  return scope.context.invoke(\n    definition,\n    inputs,\n    fn,\n    scope.depth,\n    nodeConfiguration,\n  )\n}\n\n/**\n * Preserve a function's original call arguments for transform cases where\n * parameter bindings discard them, such as destructured arrow parameters.\n *\n * This helper is internal transform/runtime protocol. The proxy preserves the\n * target's callability, arity, async identity, and non-constructibility while\n * only allocating a trace closure beneath an active automatic trace root.\n *\n * @experimental The automatic tracing protocol may change.\n */\nexport function __bitfabAutoWrap<T extends (...args: never[]) => unknown>(\n  definition: AutoTraceFunctionDefinition,\n  fn: T,\n): T {\n  if (fn.name === \"\") {\n    const nameParts = definition.name.split(\".\")\n    const inferredName = nameParts[nameParts.length - 1]\n    if (inferredName !== undefined) {\n      Object.defineProperty(fn, \"name\", {\n        configurable: true,\n        value: inferredName,\n      })\n    }\n  }\n\n  const target = fn as unknown as (...args: unknown[]) => unknown\n  return new Proxy(target, {\n    apply(callTarget, thisArg, args) {\n      if (!__bitfabAutoTraceActive()) {\n        return Reflect.apply(callTarget, thisArg, args)\n      }\n      return __bitfabAutoSpan(definition, args, () =>\n        Reflect.apply(callTarget, thisArg, args),\n      )\n    },\n  }) as unknown as T\n}\n\n/**\n * Return whether the current call is inside an automatic trace root.\n *\n * Build transforms use this before allocating function metadata, captured\n * inputs, or an invocation closure. It is internal transform/runtime protocol,\n * not a supported application API.\n *\n * @experimental The automatic tracing protocol may change.\n */\nexport function __bitfabAutoTraceActive(): boolean {\n  if (autoTraceState.activeRoots === 0) {\n    return false\n  }\n  return currentAutoTraceScope() !== undefined\n}\n\nfunction currentAutoTraceScope(): AutoTraceScope | undefined {\n  initializeAutoTraceStorage()\n  return autoTraceState.storage?.getStore() ?? autoTraceState.browserScope\n}\n\nexport function __setBitfabAutoTraceCapturePolicy(\n  client: object,\n  traceFunctionKey: string,\n  functionIds: Iterable<string>,\n): void {\n  const policies = autoTraceState.capturePolicies.get(client) ?? new Map()\n  policies.set(traceFunctionKey, new Set(functionIds))\n  autoTraceState.capturePolicies.set(client, policies)\n}\n\nexport function getAutoTraceCapturePolicy(\n  client: object,\n  traceFunctionKey: string,\n): ReadonlySet<string> {\n  return (\n    autoTraceState.capturePolicies.get(client)?.get(traceFunctionKey) ??\n    new Set()\n  )\n}\n"],"mappings":";;;;;AA4DA,IAAM,kBAAkB;AACxB,IAAM,iBACJ,gBAAgB,4BAA4B;AAAA,EAC1C,SAAS;AAAA,EACT,cAAc;AAAA,EACd,iBAAiB,oBAAI,QAAQ;AAAA,EAC7B,aAAa;AACf;AACF,gBAAgB,2BAA2B;AAE3C,SAAS,6BAAmC;AAC1C,iBAAe,YAAf,eAAe,UAAY,wBAAwC;AACrE;AAEO,SAAS,wBACd,SACA,IACA,QAAQ,GACL;AACH,6BAA2B;AAC3B,QAAM,QAAQ,EAAE,SAAS,MAAM;AAC/B,MAAI,eAAe,SAAS;AAC1B,WAAO,eAAe,QAAQ,IAAI,OAAO,EAAE;AAAA,EAC7C;AAEA,QAAM,WAAW,eAAe;AAChC,iBAAe,eAAe;AAC9B,MAAI;AACF,WAAO,GAAG;AAAA,EACZ,UAAE;AACA,mBAAe,eAAe;AAAA,EAChC;AACF;AAEO,SAAS,kCACd,mBACA,IACG;AACH,QAAM,QAAQ,sBAAsB;AACpC,MAAI,CAAC,OAAO;AACV,WAAO,GAAG;AAAA,EACZ;AAEA,QAAM,kBAAkB,EAAE,GAAG,OAAO,kBAAkB;AACtD,MAAI;AACJ,MAAI,eAAe,SAAS;AAC1B,aAAS,eAAe,QAAQ,IAAI,iBAAiB,EAAE;AAAA,EACzD,OAAO;AACL,UAAM,WAAW,eAAe;AAChC,mBAAe,eAAe;AAC9B,QAAI;AACF,eAAS,GAAG;AAAA,IACd,UAAE;AACA,qBAAe,eAAe;AAAA,IAChC;AAAA,EACF;AAEA,MAAI,0BAA0B,MAAM,GAAG;AACrC,WAAO,gCAAgC,mBAAmB,MAAM;AAAA,EAClE;AACA,SAAO;AACT;AAEO,SAAS,4BACd,SACA,IACG;AACH,iBAAe,eAAe;AAC9B,MAAI;AACJ,MAAI;AACF,aAAS,wBAAwB,SAAS,EAAE;AAAA,EAC9C,SAAS,OAAO;AACd,mBAAe,eAAe;AAC9B,UAAM;AAAA,EACR;AAEA,MAAI,0BAA0B,MAAM,GAAG;AACrC,mBAAe,eAAe;AAC9B,WAAO,4BAA4B,SAAS,MAAM;AAAA,EACpD;AAEA,MAAI,kBAAkB,SAAS;AAC7B,WAAO,OAAO,QAAQ,MAAM;AAC1B,qBAAe,eAAe;AAAA,IAChC,CAAC;AAAA,EACH;AAEA,iBAAe,eAAe;AAC9B,SAAO;AACT;AAEA,SAAS,0BACP,OACkC;AAClC,MAAI,UAAU,QAAQ,OAAO,UAAU,UAAU;AAC/C,WAAO;AAAA,EACT;AACA,QAAM,YAAY;AAClB,SACE,OAAO,UAAU,SAAS,cAC1B,OAAO,UAAU,WAAW,cAC5B,OAAO,UAAU,UAAU,cAC3B,OAAO,UAAU,OAAO,aAAa,MAAM;AAE/C;AAEA,SAAS,4BACP,SACA,QACyB;AACzB,QAAM,OAAO,CACX,QACA,UAEA,4BAA4B,SAAS,MAAM,OAAO,MAAM,EAAE,KAAK,CAAC;AAClE,QAAM,UAAmC;AAAA,IACvC,MAAM,CAAC,UAAU,KAAK,QAAQ,KAAK;AAAA,IACnC,QAAQ,CAAC,UAAU,KAAK,UAAU,KAAK;AAAA,IACvC,OAAO,CAAC,UAAU,KAAK,SAAS,KAAK;AAAA,IACrC,CAAC,OAAO,aAAa,GAAG,MAAM;AAAA,EAChC;AACA,SAAO;AACT;AAEA,SAAS,gCACP,mBACA,QACyB;AACzB,QAAM,OAAO,CACX,QACA,UAEA;AAAA,IAAkC;AAAA,IAAmB,MACnD,OAAO,MAAM,EAAE,KAAK;AAAA,EACtB;AACF,QAAM,UAAmC;AAAA,IACvC,MAAM,CAAC,UAAU,KAAK,QAAQ,KAAK;AAAA,IACnC,QAAQ,CAAC,UAAU,KAAK,UAAU,KAAK;AAAA,IACvC,OAAO,CAAC,UAAU,KAAK,SAAS,KAAK;AAAA,IACrC,CAAC,OAAO,aAAa,GAAG,MAAM;AAAA,EAChC;AACA,SAAO;AACT;AAEO,SAAS,iBACd,YACA,QACA,IACG;AACH,QAAM,QAAQ,sBAAsB;AACpC,MAAI,CAAC,OAAO;AACV,WAAO,GAAG;AAAA,EACZ;AACA,QAAM,YAAY,WAAW,KAAK,MAAM,GAAG;AAC3C,QAAM,aAAa,UAAU,UAAU,SAAS,CAAC;AACjD,QAAM,oBACJ,eAAe,MAAM,mBAAmB,eACpC,MAAM,oBACN;AACN,SAAO,MAAM,QAAQ;AAAA,IACnB;AAAA,IACA;AAAA,IACA;AAAA,IACA,MAAM;AAAA,IACN;AAAA,EACF;AACF;AAYO,SAAS,iBACd,YACA,IACG;AACH,MAAI,GAAG,SAAS,IAAI;AAClB,UAAM,YAAY,WAAW,KAAK,MAAM,GAAG;AAC3C,UAAM,eAAe,UAAU,UAAU,SAAS,CAAC;AACnD,QAAI,iBAAiB,QAAW;AAC9B,aAAO,eAAe,IAAI,QAAQ;AAAA,QAChC,cAAc;AAAA,QACd,OAAO;AAAA,MACT,CAAC;AAAA,IACH;AAAA,EACF;AAEA,QAAM,SAAS;AACf,SAAO,IAAI,MAAM,QAAQ;AAAA,IACvB,MAAM,YAAY,SAAS,MAAM;AAC/B,UAAI,CAAC,wBAAwB,GAAG;AAC9B,eAAO,QAAQ,MAAM,YAAY,SAAS,IAAI;AAAA,MAChD;AACA,aAAO;AAAA,QAAiB;AAAA,QAAY;AAAA,QAAM,MACxC,QAAQ,MAAM,YAAY,SAAS,IAAI;AAAA,MACzC;AAAA,IACF;AAAA,EACF,CAAC;AACH;AAWO,SAAS,0BAAmC;AACjD,MAAI,eAAe,gBAAgB,GAAG;AACpC,WAAO;AAAA,EACT;AACA,SAAO,sBAAsB,MAAM;AACrC;AAEA,SAAS,wBAAoD;AAC3D,6BAA2B;AAC3B,SAAO,eAAe,SAAS,SAAS,KAAK,eAAe;AAC9D;AAEO,SAAS,kCACd,QACA,kBACA,aACM;AACN,QAAM,WAAW,eAAe,gBAAgB,IAAI,MAAM,KAAK,oBAAI,IAAI;AACvE,WAAS,IAAI,kBAAkB,IAAI,IAAI,WAAW,CAAC;AACnD,iBAAe,gBAAgB,IAAI,QAAQ,QAAQ;AACrD;AAEO,SAAS,0BACd,QACA,kBACqB;AACrB,SACE,eAAe,gBAAgB,IAAI,MAAM,GAAG,IAAI,gBAAgB,KAChE,oBAAI,IAAI;AAEZ;","names":[]}

package/dist/{chunk-CWXCVY75.js → chunk-NKX3EY35.js}

@@ -1,9 +1,11 @@
 import {
+  __bitfabAutoTraceActive,
   __setBitfabAutoTraceCapturePolicy,
   getAutoTraceCapturePolicy,
   runWithAutoTraceContext,
+  runWithAutoTraceNodeConfiguration,
   runWithAutoTraceRootContext
-} from "./chunk-GFBQ2AMO.js";
+} from "./chunk-J47KPS77.js";
 import {
   BitfabError,
   DEFAULT_SERVICE_URL,
@@ -16,7 +18,7 @@ import {
   toJsonSafe,
   toJsonSafeReport,
   warnOnce
-} from "./chunk-SU7EAKOV.js";
+} from "./chunk-5GWBJGYZ.js";
 import {
   __privateAdd,
   __privateGet,
@@ -2447,6 +2449,102 @@ var Bitfab = class {
     const name = fn.name !== "" ? fn.name : traceFunctionKey;
     return this.createAutoTraceRoot(traceFunctionKey, name, options, fn);
   }
+  /**
+   * Configure a transformed class method when it is discovered beneath a
+   * {@link Bitfab.trace} root.
+   *
+   * The decorator creates no span or trace by itself. Beneath an active trace,
+   * it can rename or retype the discovered call, capture its contents, mark it
+   * for recorded-output replay, finalize its output, or omit it while leaving
+   * captured descendants attached to the nearest captured parent.
+   *
+   * @param options - Trace-owned call configuration.
+   * @experimental Automatic child-call instrumentation is experimental.
+   */
+  node(options = {}) {
+    const configuration = this.resolveNodeConfiguration(options);
+    const decorator = (...args) => {
+      if (args.length === 3) {
+        const descriptor = args[2];
+        if (!descriptor || typeof descriptor.value !== "function") {
+          throw new BitfabError("@bitfab.node can only decorate methods");
+        }
+        if (!this.explicitlyEnabled) {
+          return;
+        }
+        descriptor.value = this.createAutoTraceNode(
+          configuration,
+          descriptor.value,
+          String(args[1])
+        );
+        return;
+      }
+      const method = args[0];
+      const context = args[1];
+      if (typeof method !== "function" || context?.kind !== "method") {
+        throw new BitfabError("@bitfab.node can only decorate methods");
+      }
+      if (!this.explicitlyEnabled) {
+        return method;
+      }
+      return this.createAutoTraceNode(
+        configuration,
+        method,
+        String(context.name)
+      );
+    };
+    return decorator;
+  }
+  withNode(optionsOrFn, maybeFn, internalFunctionName) {
+    const options = typeof optionsOrFn === "function" ? {} : optionsOrFn;
+    const fn = typeof optionsOrFn === "function" ? optionsOrFn : maybeFn;
+    if (!fn) {
+      throw new BitfabError("bitfab.withNode requires a function");
+    }
+    const configuration = this.resolveNodeConfiguration(options);
+    if (!this.explicitlyEnabled) {
+      return fn;
+    }
+    const functionName = internalFunctionName ?? fn.name;
+    if (functionName === "") {
+      throw new BitfabError(
+        "bitfab.withNode requires a named function so the subtree transform can bind its configuration to the correct call."
+      );
+    }
+    return this.createAutoTraceNode(configuration, fn, functionName);
+  }
+  resolveNodeConfiguration(options) {
+    const capture = options.capture ?? true;
+    if (!capture && options.mockOnReplay === true) {
+      throw new BitfabError(
+        "bitfab.node({ capture: false }) cannot use mockOnReplay: true because an uncaptured node has no recorded output."
+      );
+    }
+    return {
+      capture,
+      type: options.type ?? "custom",
+      ...options.name !== void 0 && { name: options.name },
+      ...options.testRunId !== void 0 && {
+        testRunId: options.testRunId
+      },
+      ...options.mockOnReplay !== void 0 && {
+        mockOnReplay: options.mockOnReplay
+      },
+      ...options.finalize !== void 0 && { finalize: options.finalize }
+    };
+  }
+  createAutoTraceNode(configuration, fn, functionName) {
+    const nodeConfiguration = { ...configuration, functionName };
+    return function(...args) {
+      if (!__bitfabAutoTraceActive()) {
+        return fn.apply(this, args);
+      }
+      return runWithAutoTraceNodeConfiguration(
+        nodeConfiguration,
+        () => fn.apply(this, args)
+      );
+    };
+  }
   createAutoTraceRoot(traceFunctionKey, name, options, fn) {
     const self = this;
     const maxDepth = autoTraceLimit(
@@ -2485,29 +2583,51 @@ var Bitfab = class {
           );
         };
         const autoTraceContext = {
-          invoke(definition, inputs, invokeFn, depth) {
+          invoke(definition, inputs, invokeFn, depth, nodeConfiguration) {
             const nameParts = definition.name.split(".");
             const simpleName = nameParts[nameParts.length - 1];
-            if (excluded.has(definition.name) || simpleName !== void 0 && excluded.has(simpleName) || definition.wrapper === true && !includeWrappers) {
-              return invokeFn();
+            const invokeWithoutNode = () => nodeConfiguration === void 0 ? invokeFn() : runWithAutoTraceContext(autoTraceContext, invokeFn, depth);
+            if (excluded.has(definition.name) || simpleName !== void 0 && excluded.has(simpleName) || nodeConfiguration === void 0 && definition.wrapper === true && !includeWrappers) {
+              return invokeWithoutNode();
+            }
+            if (nodeConfiguration?.capture === false) {
+              return runWithAutoTraceContext(autoTraceContext, invokeFn, depth);
             }
             if (depth >= maxDepth || spansUsed >= maxSpans) {
               warnTruncated();
-              return invokeFn();
+              return invokeWithoutNode();
             }
             spansUsed += 1;
             const childOptions = {
-              name: definition.name,
-              type: "function",
+              name: nodeConfiguration?.name ?? definition.name,
+              type: nodeConfiguration?.type ?? "function",
               captureWhen: "nested",
               functionId: definition.id,
-              captureContent: capturePolicy.has(definition.id),
-              autoTraceDefinition: definition
+              captureContent: nodeConfiguration !== void 0 || capturePolicy.has(definition.id),
+              autoTraceDefinition: definition,
+              ...nodeConfiguration?.testRunId !== void 0 && {
+                testRunId: nodeConfiguration.testRunId
+              },
+              ...nodeConfiguration?.mockOnReplay !== void 0 && {
+                mockOnReplay: nodeConfiguration.mockOnReplay
+              },
+              ...nodeConfiguration?.finalize !== void 0 && {
+                finalize: nodeConfiguration.finalize
+              }
             };
+            const invokeWithAutoTraceContext = () => runWithAutoTraceContext(autoTraceContext, invokeFn, depth + 1);
+            if (definition.async === true) {
+              const tracedAsyncChild = self.withSpan(
+                traceFunctionKey,
+                childOptions,
+                async (..._inputs) => await invokeWithAutoTraceContext()
+              );
+              return tracedAsyncChild(...inputs);
+            }
             const tracedChild = self.withSpan(
               traceFunctionKey,
               childOptions,
-              (..._inputs) => runWithAutoTraceContext(autoTraceContext, invokeFn, depth + 1)
+              (..._inputs) => invokeWithAutoTraceContext()
             );
             return tracedChild(...inputs);
           }
@@ -3077,18 +3197,17 @@ var Bitfab = class {
         newStack = [...currentStack, newContext];
         const inputs = args;
         const startedAt = (/* @__PURE__ */ new Date()).toISOString();
+        const replayCtxAtStart = getReplayContext();
+        const testRunId = replayCtxAtStart?.testRunId ?? options.testRunId;
         if (isRootSpan && !activeTraceStates.has(traceId)) {
-          const replayCtxAtRoot = getReplayContext();
           const dbSnapshotRef = buildSnapshotRef(self.dbSnapshot, startedAt);
           activeTraceStates.set(traceId, {
             traceId,
             startedAt,
             contexts: [],
-            ...replayCtxAtRoot?.testRunId && {
-              testRunId: replayCtxAtRoot.testRunId
-            },
-            ...replayCtxAtRoot?.inputSourceTraceId && {
-              inputSourceTraceId: replayCtxAtRoot.inputSourceTraceId
+            ...testRunId !== void 0 && { testRunId },
+            ...replayCtxAtStart?.inputSourceTraceId && {
+              inputSourceTraceId: replayCtxAtStart.inputSourceTraceId
             },
             dbSnapshotRef
           });
@@ -3121,9 +3240,7 @@ var Bitfab = class {
                 contexts: newContext.contexts,
                 prompt: newContext.prompt,
                 endedAt,
-                ...replayCtx?.testRunId && {
-                  testRunId: replayCtx.testRunId
-                },
+                ...testRunId !== void 0 && { testRunId },
                 ...replayCtx?.inputSourceSpanId && {
                   inputSourceSpanId: replayCtx.inputSourceSpanId
                 }
@@ -3608,7 +3725,7 @@ var Bitfab = class {
         `Function is wrapped with trace function key '${wrappedKey}' but replay was called with '${traceFunctionKey}'. Pass matching keys, or pass the unwrapped function to replay it under the explicit key.`
       );
     }
-    const { replay: doReplay } = await import("./replay-A5UVFNZ3.js");
+    const { replay: doReplay } = await import("./replay-O2WFIRM4.js");
     return doReplay(
       this.httpClient,
       this.serviceUrl,
@@ -3832,6 +3949,11 @@ var finalizers = {
   readableStream
 };
 
+// src/replayRegistry.ts
+function defineReplayRegistry(registry) {
+  return registry;
+}
+
 export {
   BitfabClaudeAgentHandler,
   SUPPORTED_PROVIDERS,
@@ -3844,6 +3966,7 @@ export {
   getCurrentTrace,
   Bitfab,
   BitfabFunction,
-  finalizers
+  finalizers,
+  defineReplayRegistry
 };
-//# sourceMappingURL=chunk-CWXCVY75.js.map
+//# sourceMappingURL=chunk-NKX3EY35.js.map