bitfab 0.38.4 → 0.38.6

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

@@ -155,11 +155,18 @@ function currentAutoTraceScope() {
 }
 function __setBitfabAutoTraceCapturePolicy(client, traceFunctionKey, functionIds) {
   const policies = autoTraceState.capturePolicies.get(client) ?? /* @__PURE__ */ new Map();
+  if (functionIds === void 0) {
+    policies.delete(traceFunctionKey);
+    if (policies.size === 0) {
+      autoTraceState.capturePolicies.delete(client);
+    }
+    return;
+  }
   policies.set(traceFunctionKey, new Set(functionIds));
   autoTraceState.capturePolicies.set(client, policies);
 }
 function getAutoTraceCapturePolicy(client, traceFunctionKey) {
-  return autoTraceState.capturePolicies.get(client)?.get(traceFunctionKey) ?? /* @__PURE__ */ new Set();
+  return autoTraceState.capturePolicies.get(client)?.get(traceFunctionKey);
 }
 
 export {
@@ -172,4 +179,4 @@ export {
   __setBitfabAutoTraceCapturePolicy,
   getAutoTraceCapturePolicy
 };
-//# sourceMappingURL=chunk-J47KPS77.js.map
+//# sourceMappingURL=chunk-ZUD7OFYB.js.map

package/dist/{chunk-J47KPS77.js.map → chunk-ZUD7OFYB.js.map}

@@ -1 +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":[]}
+{"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> | undefined,\n): void {\n  const policies = autoTraceState.capturePolicies.get(client) ?? new Map()\n  if (functionIds === undefined) {\n    policies.delete(traceFunctionKey)\n    if (policies.size === 0) {\n      autoTraceState.capturePolicies.delete(client)\n    }\n    return\n  }\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> | undefined {\n  return autoTraceState.capturePolicies.get(client)?.get(traceFunctionKey)\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,MAAI,gBAAgB,QAAW;AAC7B,aAAS,OAAO,gBAAgB;AAChC,QAAI,SAAS,SAAS,GAAG;AACvB,qBAAe,gBAAgB,OAAO,MAAM;AAAA,IAC9C;AACA;AAAA,EACF;AACA,WAAS,IAAI,kBAAkB,IAAI,IAAI,WAAW,CAAC;AACnD,iBAAe,gBAAgB,IAAI,QAAQ,QAAQ;AACrD;AAEO,SAAS,0BACd,QACA,kBACiC;AACjC,SAAO,eAAe,gBAAgB,IAAI,MAAM,GAAG,IAAI,gBAAgB;AACzE;","names":[]}

package/dist/index.cjs

@@ -51,7 +51,7 @@ var __version__, __packageName__;
 var init_version_generated = __esm({
   "src/version.generated.ts"() {
     "use strict";
-    __version__ = "0.38.4";
+    __version__ = "0.38.6";
     __packageName__ = "bitfab";
   }
 });
@@ -2196,11 +2196,16 @@ function normalizeMockOverrides(mockOverride) {
   if (mockOverride === void 0) {
     return [];
   }
-  return Array.isArray(mockOverride) ? mockOverride : [mockOverride];
+  const overrides = Array.isArray(mockOverride) ? mockOverride : [mockOverride];
+  return overrides.map(
+    (override) => typeof override === "function" ? { match: () => true, value: override } : override
+  );
 }
+var NO_MOCK_OVERRIDE;
 var init_mockOverride = __esm({
   "src/mockOverride.ts"() {
     "use strict";
+    NO_MOCK_OVERRIDE = /* @__PURE__ */ Symbol("bitfab.noMockOverride");
   }
 });
 
@@ -3142,6 +3147,7 @@ __export(index_exports, {
   DEFAULT_SERVICE_URL: () => DEFAULT_SERVICE_URL,
   DbBranchReplayError: () => DbBranchReplayError,
   HttpClient: () => HttpClient,
+  NO_MOCK_OVERRIDE: () => NO_MOCK_OVERRIDE,
   ReplayError: () => ReplayError,
   SUPPORTED_PROVIDERS: () => SUPPORTED_PROVIDERS,
   __version__: () => __version__,
@@ -3883,11 +3889,18 @@ function currentAutoTraceScope() {
 }
 function __setBitfabAutoTraceCapturePolicy(client, traceFunctionKey, functionIds) {
   const policies = autoTraceState.capturePolicies.get(client) ?? /* @__PURE__ */ new Map();
+  if (functionIds === void 0) {
+    policies.delete(traceFunctionKey);
+    if (policies.size === 0) {
+      autoTraceState.capturePolicies.delete(client);
+    }
+    return;
+  }
   policies.set(traceFunctionKey, new Set(functionIds));
   autoTraceState.capturePolicies.set(client, policies);
 }
 function getAutoTraceCapturePolicy(client, traceFunctionKey) {
-  return autoTraceState.capturePolicies.get(client)?.get(traceFunctionKey) ?? /* @__PURE__ */ new Set();
+  return autoTraceState.capturePolicies.get(client)?.get(traceFunctionKey);
 }
 
 // src/optionalPeer.ts
@@ -5676,8 +5689,9 @@ var Bitfab = class {
    * Decorate a class method as an automatically expanded trace root.
    *
    * Build instrumentation turns repository functions called beneath this
-   * method into nested spans. Every generated span preserves structure; only
-   * function IDs selected by the capture policy include inputs and output.
+   * method into nested spans that capture inputs, outputs, and errors by
+   * default. A confirmed capture policy can narrow rich capture to selected
+   * function IDs.
    * Without a compatible build transform, this still records the decorated
    * method as a normal rich root span but cannot discover child calls.
    *
@@ -5887,7 +5901,7 @@ var Bitfab = class {
               type: nodeConfiguration?.type ?? "function",
               captureWhen: "nested",
               functionId: definition.id,
-              captureContent: nodeConfiguration !== void 0 || capturePolicy.has(definition.id),
+              captureContent: nodeConfiguration !== void 0 || capturePolicy === void 0 || capturePolicy.has(definition.id),
               autoTraceDefinition: definition,
               ...nodeConfiguration?.testRunId !== void 0 && {
                 testRunId: nodeConfiguration.testRunId
@@ -5952,7 +5966,11 @@ var Bitfab = class {
       const functionIds = Array.isArray(policy.functionIds) ? policy.functionIds.filter(
         (id) => typeof id === "string" && id.startsWith(`${AUTO_TRACE_PROTOCOL}:`)
       ).slice(0, DEFAULT_AUTO_TRACE_MAX_SPANS) : [];
-      __setBitfabAutoTraceCapturePolicy(this, traceFunctionKey, functionIds);
+      __setBitfabAutoTraceCapturePolicy(
+        this,
+        traceFunctionKey,
+        policy.revision === null ? void 0 : functionIds
+      );
       state.refreshAfter = Date.now() + AUTO_TRACE_POLICY_REFRESH_MS;
     }).catch(() => {
       state.refreshAfter = Date.now() + AUTO_TRACE_POLICY_RETRY_MS;
@@ -6564,6 +6582,49 @@ var Bitfab = class {
           } catch {
           }
         };
+        const recordSpan = (result) => {
+          if (options.finalize) {
+            void self.httpClient.trackDeferred(
+              Promise.resolve().then(() => options.finalize(result)).then((output) => sendSpan({ result: output })).catch(
+                (error) => sendSpan({
+                  result: void 0,
+                  error: error instanceof Error ? `finalize failed: ${error.message}` : `finalize failed: ${String(error)}`
+                })
+              )
+            );
+          } else {
+            void sendSpan({ result });
+          }
+        };
+        executeWithContext = () => {
+          let result;
+          try {
+            result = fn.apply(this, args);
+          } catch (error) {
+            void sendSpan({
+              result: void 0,
+              error: error instanceof Error ? error.message : String(error)
+            });
+            throw error;
+          }
+          if (result instanceof Promise) {
+            return result.then((resolvedResult) => {
+              recordSpan(resolvedResult);
+              return resolvedResult;
+            }).catch((error) => {
+              void sendSpan({
+                result: void 0,
+                error: error instanceof Error ? error.message : String(error)
+              });
+              throw error;
+            });
+          }
+          if (isAsyncGenerator(result)) {
+            return wrapAsyncGenerator(result, newStack, sendSpan);
+          }
+          recordSpan(result);
+          return result;
+        };
         const replayCtxForMock = getReplayContext();
         if (replayCtxForMock?.mockTree && !isRootSpan) {
           const counters = replayCtxForMock.callCounters;
@@ -6622,6 +6683,7 @@ var Bitfab = class {
             }
             return output;
           };
+          const shouldMockWithBaseStrategy = replayCtxForMock.mockStrategy === "all" || replayCtxForMock.mockStrategy === "marked" && options.mockOnReplay === true;
           if (replayCtxForMock.mockOverrides?.length) {
             const nodeMeta = {
               traceFunctionKey,
@@ -6629,28 +6691,75 @@ var Bitfab = class {
               type: options.type ?? "custom",
               originalSpanId: mockSpan?.sourceSpanId
             };
-            const override = replayCtxForMock.mockOverrides.find(
-              (o) => o.match(nodeMeta)
-            );
-            if (override) {
-              const injected = resolveMockValue(override.value, {
-                node: nodeMeta,
-                inputs: args,
-                getOriginalOutput: () => Promise.resolve(resolveRecordedOutput())
-              });
-              if (injected instanceof Promise) {
-                return emitMockAsync(injected, "override");
+            const overrideCtx = {
+              node: nodeMeta,
+              inputs: args,
+              getOriginalOutput: () => Promise.resolve(resolveRecordedOutput())
+            };
+            const resolveOverrideFrom = (startIndex) => {
+              for (let index = startIndex; index < replayCtxForMock.mockOverrides.length; index += 1) {
+                const override = replayCtxForMock.mockOverrides[index];
+                if (!override?.match(nodeMeta)) {
+                  continue;
+                }
+                const injected = resolveMockValue(override.value, overrideCtx);
+                if (injected instanceof Promise) {
+                  return injected.then(
+                    (output) => output === NO_MOCK_OVERRIDE ? resolveOverrideFrom(index + 1) : { matched: true, output }
+                  );
+                }
+                if (injected !== NO_MOCK_OVERRIDE) {
+                  return { matched: true, output: injected };
+                }
               }
-              return emitMock(injected, "override");
+              return { matched: false };
+            };
+            const resolution = resolveOverrideFrom(0);
+            if (resolution instanceof Promise) {
+              if (!fnReturnsPromise) {
+                throw new BitfabError(
+                  `Cannot resolve an asynchronous mock override for synchronous span "${traceFunctionKey}". Make the wrapped function async or return NO_MOCK_OVERRIDE synchronously.`
+                );
+              }
+              return runWithSpanStack(newStack, async () => {
+                const resolved = await resolution;
+                if (resolved.matched) {
+                  void sendSpan({
+                    result: resolved.output,
+                    mocked: true,
+                    mockTarget: "output",
+                    mockSource: "override"
+                  });
+                  return resolved.output;
+                }
+                if (shouldMockWithBaseStrategy && !mockSpan) {
+                  throw new BitfabError(
+                    `Replay selected span "${traceFunctionKey}:${baseSpanParams.spanName}" for mocking, but recorded occurrence ${callIndex + 1} is unavailable. The real span was not executed.`
+                  );
+                }
+                if (shouldMockWithBaseStrategy) {
+                  const output = await resolveRecordedOutput();
+                  void sendSpan({
+                    result: output,
+                    mocked: true,
+                    mockTarget: "output",
+                    mockSource: "recorded"
+                  });
+                  return output;
+                }
+                return executeWithContext();
+              });
+            }
+            if (resolution.matched) {
+              return emitMock(resolution.output, "override");
             }
           }
-          const shouldMock = replayCtxForMock.mockStrategy === "all" || replayCtxForMock.mockStrategy === "marked" && options.mockOnReplay === true;
-          if (shouldMock && !mockSpan) {
+          if (shouldMockWithBaseStrategy && !mockSpan) {
             throw new BitfabError(
               `Replay selected span "${traceFunctionKey}:${baseSpanParams.spanName}" for mocking, but recorded occurrence ${callIndex + 1} is unavailable. The real span was not executed.`
             );
           }
-          if (shouldMock) {
+          if (shouldMockWithBaseStrategy) {
             const recorded = resolveRecordedOutput();
             if (recorded instanceof Promise) {
               return emitMockAsync(recorded, "recorded");
@@ -6658,49 +6767,6 @@ var Bitfab = class {
             return emitMock(recorded, "recorded");
           }
         }
-        const recordSpan = (result) => {
-          if (options.finalize) {
-            void self.httpClient.trackDeferred(
-              Promise.resolve().then(() => options.finalize(result)).then((output) => sendSpan({ result: output })).catch(
-                (error) => sendSpan({
-                  result: void 0,
-                  error: error instanceof Error ? `finalize failed: ${error.message}` : `finalize failed: ${String(error)}`
-                })
-              )
-            );
-          } else {
-            void sendSpan({ result });
-          }
-        };
-        executeWithContext = () => {
-          let result;
-          try {
-            result = fn.apply(this, args);
-          } catch (error) {
-            void sendSpan({
-              result: void 0,
-              error: error instanceof Error ? error.message : String(error)
-            });
-            throw error;
-          }
-          if (result instanceof Promise) {
-            return result.then((resolvedResult) => {
-              recordSpan(resolvedResult);
-              return resolvedResult;
-            }).catch((error) => {
-              void sendSpan({
-                result: void 0,
-                error: error instanceof Error ? error.message : String(error)
-              });
-              throw error;
-            });
-          }
-          if (isAsyncGenerator(result)) {
-            return wrapAsyncGenerator(result, newStack, sendSpan);
-          }
-          recordSpan(result);
-          return result;
-        };
       } catch (setupError) {
         if (registeredTraceId) {
           activeTraceStates.delete(registeredTraceId);
@@ -6987,8 +7053,40 @@ var Bitfab = class {
       ...params.mockSource && { mockSource: params.mockSource }
     });
   }
-  registerMockOverride(overrideOrMatch, value) {
-    const override = typeof overrideOrMatch === "function" ? { match: overrideOrMatch, value } : overrideOrMatch;
+  registerMockOverride(overrideOrResolverOrMatch, ...values) {
+    let override;
+    if (typeof overrideOrResolverOrMatch === "string") {
+      const keyedOverride = values[0];
+      if (values.length !== 1 || keyedOverride === void 0) {
+        throw new BitfabError(
+          "registerMockOverride(traceFunctionKey, override) requires a resolver function or { match, value } override."
+        );
+      }
+      if (typeof keyedOverride === "function") {
+        override = {
+          match: (node) => node.traceFunctionKey === overrideOrResolverOrMatch,
+          value: keyedOverride
+        };
+      } else if (typeof keyedOverride === "object" && keyedOverride !== null && "match" in keyedOverride && "value" in keyedOverride) {
+        override = {
+          match: (node) => node.traceFunctionKey === overrideOrResolverOrMatch && keyedOverride.match(node),
+          value: keyedOverride.value
+        };
+      } else {
+        throw new BitfabError(
+          "registerMockOverride(traceFunctionKey, override) requires a resolver function or { match, value } override."
+        );
+      }
+    } else if (typeof overrideOrResolverOrMatch !== "function") {
+      override = overrideOrResolverOrMatch;
+    } else if (values.length === 0) {
+      override = { match: () => true, value: overrideOrResolverOrMatch };
+    } else {
+      override = {
+        match: overrideOrResolverOrMatch,
+        value: values[0]
+      };
+    }
     this.mockOverrides.push(override);
   }
   /** Remove all overrides registered via {@link registerMockOverride}. */
@@ -7238,6 +7336,7 @@ var finalizers = {
 
 // src/index.ts
 init_http();
+init_mockOverride();
 init_replay();
 
 // src/replayRegistry.ts
@@ -7261,6 +7360,7 @@ function defineReplayRegistry(registry) {
   DEFAULT_SERVICE_URL,
   DbBranchReplayError,
   HttpClient,
+  NO_MOCK_OVERRIDE,
   ReplayError,
   SUPPORTED_PROVIDERS,
   __version__,