@bitfab/sdk 0.38.1 → 0.38.3

package/dist/index.cjs

@@ -42,7 +42,7 @@ var __version__, __packageName__;
 var init_version_generated = __esm({
   "src/version.generated.ts"() {
     "use strict";
-    __version__ = "0.38.1";
+    __version__ = "0.38.3";
     __packageName__ = "@bitfab/sdk";
   }
 });
@@ -812,7 +812,7 @@ var init_otel = __esm({
     EXPORT_CONCURRENCY_ENV = "BITFAB_OTEL_EXPORT_CONCURRENCY";
     MAX_QUEUE_SIZE = 8192;
     DIRECT_MAX_EXPORT_BATCH_SIZE = 512;
-    DIRECT_MAX_REQUEST_BATCH_SIZE = 8;
+    DIRECT_MAX_REQUEST_BATCH_SIZE = 128;
     DEFAULT_EXPORT_CONCURRENCY = 32;
     MAX_EXPORT_CONCURRENCY = 64;
     SCHEDULE_DELAY_MILLIS = 5e3;
@@ -1704,6 +1704,12 @@ var init_http = __esm({
       async lookupFunction(name) {
         return this.request("/api/sdk/functions/lookup", { name });
       }
+      async getAutoTracePolicy(traceFunctionKey, protocol) {
+        return this.request("/api/sdk/auto-trace/policy", {
+          traceFunctionKey,
+          protocol
+        });
+      }
       async getTraceSpan(traceId, lookup) {
         const searchParams = new URLSearchParams();
         if (lookup.id !== void 0) {
@@ -2798,6 +2804,11 @@ async function replay(httpClient, serviceUrl, traceFunctionKey, fn, options, reg
       );
     }
   }
+  if (options?.traceIds !== void 0 && options?.datasetId !== void 0) {
+    throw new BitfabError(
+      "traceIds and datasetId select different replay sources and cannot be used together."
+    );
+  }
   if (options?.limit !== void 0 && options?.traceIds !== void 0) {
     try {
       console.warn(
@@ -3125,6 +3136,7 @@ __export(index_exports, {
   ReplayError: () => ReplayError,
   SUPPORTED_PROVIDERS: () => SUPPORTED_PROVIDERS,
   __version__: () => __version__,
+  defineReplayRegistry: () => defineReplayRegistry,
   finalizers: () => finalizers,
   flushTraces: () => flushTraces,
   getCurrentReplayBranch: () => getCurrentReplayBranch,
@@ -3749,6 +3761,80 @@ var BitfabClaudeAgentHandler = class {
 // src/client.ts
 init_asyncStorage();
 
+// src/autoTrace.ts
+init_asyncStorage();
+var autoTraceGlobal = globalThis;
+var autoTraceState = autoTraceGlobal.__bitfabAutoTraceStateV3 ?? {
+  storage: null,
+  browserScope: void 0,
+  capturePolicies: /* @__PURE__ */ new WeakMap(),
+  activeRoots: 0
+};
+autoTraceGlobal.__bitfabAutoTraceStateV3 = autoTraceState;
+function initializeAutoTraceStorage() {
+  autoTraceState.storage ?? (autoTraceState.storage = createAsyncLocalStorage());
+}
+function runWithAutoTraceContext(context, fn, depth = 0) {
+  initializeAutoTraceStorage();
+  const scope = { context, depth };
+  if (autoTraceState.storage) {
+    return autoTraceState.storage.run(scope, fn);
+  }
+  const previous = autoTraceState.browserScope;
+  autoTraceState.browserScope = scope;
+  try {
+    return fn();
+  } finally {
+    autoTraceState.browserScope = previous;
+  }
+}
+function runWithAutoTraceRootContext(context, fn) {
+  autoTraceState.activeRoots += 1;
+  let result;
+  try {
+    result = runWithAutoTraceContext(context, fn);
+  } catch (error) {
+    autoTraceState.activeRoots -= 1;
+    throw error;
+  }
+  if (isAutoTraceAsyncGenerator(result)) {
+    autoTraceState.activeRoots -= 1;
+    return wrapAutoTraceAsyncGenerator(context, result);
+  }
+  if (result instanceof Promise) {
+    return result.finally(() => {
+      autoTraceState.activeRoots -= 1;
+    });
+  }
+  autoTraceState.activeRoots -= 1;
+  return result;
+}
+function isAutoTraceAsyncGenerator(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 wrapAutoTraceAsyncGenerator(context, source) {
+  const step = (method, value) => runWithAutoTraceRootContext(context, () => 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 __setBitfabAutoTraceCapturePolicy(client, traceFunctionKey, functionIds) {
+  const policies = autoTraceState.capturePolicies.get(client) ?? /* @__PURE__ */ new Map();
+  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();
+}
+
 // src/optionalPeer.ts
 function importOptionalPeer(specifierParts) {
   const specifier = specifierParts.join("/");
@@ -5490,6 +5576,14 @@ function readEnv2(name) {
   }
   return void 0;
 }
+var DEFAULT_AUTO_TRACE_MAX_DEPTH = 30;
+var DEFAULT_AUTO_TRACE_MAX_SPANS = 500;
+var AUTO_TRACE_PROTOCOL = "ts-auto-v1";
+var AUTO_TRACE_POLICY_REFRESH_MS = 6e4;
+var AUTO_TRACE_POLICY_RETRY_MS = 1e4;
+function autoTraceLimit(value, fallback) {
+  return value !== void 0 && Number.isFinite(value) && value >= 0 ? Math.floor(value) : fallback;
+}
 var Bitfab = class {
   /**
    * Initialize the Bitfab client.
@@ -5499,6 +5593,7 @@ var Bitfab = class {
   constructor(config) {
     /** Gate the empty-key warning to fire at most once. */
     this.apiKeyWarned = false;
+    this.autoTracePolicyRefreshes = /* @__PURE__ */ new Map();
     /**
      * Mock overrides registered via {@link Bitfab.registerMockOverride}, applied
      * to every `replay` on this client (after any per-call `mockOverride`). In
@@ -5522,6 +5617,178 @@ var Bitfab = class {
       timeout: this.timeout
     });
   }
+  /**
+   * 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.
+   * Without a compatible build transform, this still records the decorated
+   * method as a normal rich root span but cannot discover child calls.
+   *
+   * @param traceFunctionKey - Groups traces and their capture policy.
+   * @param options - Root presentation, subtree bounds, and exclusions.
+   * @experimental Automatic child-call instrumentation is experimental.
+   */
+  trace(traceFunctionKey, options = {}) {
+    const decorator = (...args) => {
+      if (args.length === 3) {
+        const propertyKey = args[1];
+        const descriptor = args[2];
+        if (!descriptor || typeof descriptor.value !== "function") {
+          throw new BitfabError("@bitfab.trace can only decorate methods");
+        }
+        if (!this.explicitlyEnabled) {
+          return;
+        }
+        descriptor.value = this.createAutoTraceRoot(
+          traceFunctionKey,
+          String(propertyKey),
+          options,
+          descriptor.value
+        );
+        return;
+      }
+      const method = args[0];
+      const context = args[1];
+      if (typeof method !== "function" || context?.kind !== "method" || context.name === void 0) {
+        throw new BitfabError("@bitfab.trace can only decorate methods");
+      }
+      if (!this.explicitlyEnabled) {
+        return method;
+      }
+      return this.createAutoTraceRoot(
+        traceFunctionKey,
+        String(context.name),
+        options,
+        method
+      );
+    };
+    return decorator;
+  }
+  withTrace(traceFunctionKey, optionsOrFn, maybeFn) {
+    const options = typeof optionsOrFn === "function" ? {} : optionsOrFn;
+    const fn = typeof optionsOrFn === "function" ? optionsOrFn : maybeFn;
+    if (!fn) {
+      throw new BitfabError("bitfab.withTrace requires a function");
+    }
+    if (!this.explicitlyEnabled) {
+      return fn;
+    }
+    const name = fn.name !== "" ? fn.name : traceFunctionKey;
+    return this.createAutoTraceRoot(traceFunctionKey, name, options, fn);
+  }
+  createAutoTraceRoot(traceFunctionKey, name, options, fn) {
+    const self = this;
+    const maxDepth = autoTraceLimit(
+      options.maxDepth,
+      DEFAULT_AUTO_TRACE_MAX_DEPTH
+    );
+    const maxSpans = autoTraceLimit(
+      options.maxSpans,
+      DEFAULT_AUTO_TRACE_MAX_SPANS
+    );
+    const excluded = new Set(options.exclude ?? []);
+    const includeWrappers = options.includeWrappers ?? false;
+    const tracedRoot = this.withSpan(
+      traceFunctionKey,
+      { name: options.name ?? name, type: options.type ?? "custom" },
+      function(...args) {
+        const capturePolicy = getAutoTraceCapturePolicy(self, traceFunctionKey);
+        self.refreshAutoTraceCapturePolicy(traceFunctionKey);
+        let spansUsed = 0;
+        let truncated = false;
+        const warnTruncated = () => {
+          if (!truncated) {
+            truncated = true;
+            getCurrentTrace().setMetadata({
+              bitfabAutoTrace: {
+                protocol: AUTO_TRACE_PROTOCOL,
+                truncated: true,
+                maxDepth,
+                maxSpans
+              }
+            });
+          }
+          warnOnce(
+            `auto-trace-truncated:${traceFunctionKey}`,
+            `"${traceFunctionKey}" hit an automatic subtree capture limit (maxDepth=${maxDepth}, maxSpans=${maxSpans}); its trace is incomplete. Raise the limits or narrow the subtree with exclude.`
+          );
+        };
+        const autoTraceContext = {
+          invoke(definition, inputs, invokeFn, depth) {
+            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();
+            }
+            if (depth >= maxDepth || spansUsed >= maxSpans) {
+              warnTruncated();
+              return invokeFn();
+            }
+            spansUsed += 1;
+            const childOptions = {
+              name: definition.name,
+              type: "function",
+              captureWhen: "nested",
+              functionId: definition.id,
+              captureContent: capturePolicy.has(definition.id),
+              autoTraceDefinition: definition
+            };
+            const tracedChild = self.withSpan(
+              traceFunctionKey,
+              childOptions,
+              (..._inputs) => runWithAutoTraceContext(autoTraceContext, invokeFn, depth + 1)
+            );
+            return tracedChild(...inputs);
+          }
+        };
+        return runWithAutoTraceRootContext(
+          autoTraceContext,
+          () => fn.apply(this, args)
+        );
+      }
+    );
+    const autoTraceRoot = function(...args) {
+      if (!self.isTracingEnabled()) {
+        return fn.apply(this, args);
+      }
+      return tracedRoot.apply(this, args);
+    };
+    Object.defineProperty(autoTraceRoot, "_bitfabTraceFunctionKey", {
+      value: traceFunctionKey
+    });
+    return autoTraceRoot;
+  }
+  refreshAutoTraceCapturePolicy(traceFunctionKey) {
+    const now = Date.now();
+    const state = this.autoTracePolicyRefreshes.get(traceFunctionKey) ?? {
+      refreshAfter: 0
+    };
+    if (state.inFlight || now < state.refreshAfter) {
+      return;
+    }
+    const request = this.httpClient.getAutoTracePolicy(
+      traceFunctionKey,
+      AUTO_TRACE_PROTOCOL
+    ).then((policy) => {
+      if (policy.protocol !== AUTO_TRACE_PROTOCOL) {
+        state.refreshAfter = Date.now() + AUTO_TRACE_POLICY_RETRY_MS;
+        return;
+      }
+      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);
+      state.refreshAfter = Date.now() + AUTO_TRACE_POLICY_REFRESH_MS;
+    }).catch(() => {
+      state.refreshAfter = Date.now() + AUTO_TRACE_POLICY_RETRY_MS;
+    }).finally(() => {
+      state.inFlight = void 0;
+    });
+    state.inFlight = request;
+    this.autoTracePolicyRefreshes.set(traceFunctionKey, state);
+  }
   /**
    * Flush and permanently close this client's tracing resources: its pending
    * requests and the single span-transport worker shared by its decorators and
@@ -6068,7 +6335,10 @@ var Bitfab = class {
           parentSpanId,
           inputs,
           startedAt,
-          spanType: options.type ?? "custom"
+          spanType: options.type ?? "custom",
+          functionId: options.functionId,
+          captureContent: options.captureContent ?? true,
+          autoTraceDefinition: options.autoTraceDefinition
         };
         const sendSpan = async (params) => {
           const replayCtx = getReplayContext();
@@ -6233,7 +6503,16 @@ var Bitfab = class {
           }
         };
         executeWithContext = () => {
-          const result = fn.apply(this, args);
+          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);
@@ -6474,8 +6753,8 @@ var Bitfab = class {
    * Queued on the client's span transport; delivery is the transport's job.
    */
   sendWrapperSpan(params) {
-    const serializedInputs = serializeValue(params.inputs);
-    const serializedResult = serializeValue(params.result);
+    const serializedInputs = params.captureContent ? serializeValue(params.inputs) : void 0;
+    const serializedResult = params.captureContent ? serializeValue(params.result) : void 0;
     const externalSpan = {
       id: params.spanId,
       trace_id: params.traceId,
@@ -6484,26 +6763,38 @@ var Bitfab = class {
       span_data: {
         name: params.spanName,
         type: params.spanType,
-        input: serializedInputs.json,
-        output: serializedResult.json,
-        // Include superjson meta for type preservation
-        ...serializedInputs.meta !== void 0 && {
-          input_meta: serializedInputs.meta
+        ...params.functionId !== void 0 && {
+          function_id: params.functionId,
+          content_captured: params.captureContent
+        },
+        ...params.autoTraceDefinition !== void 0 && {
+          function_file: params.autoTraceDefinition.file,
+          function_line: params.autoTraceDefinition.line,
+          function_column: params.autoTraceDefinition.column
         },
-        ...serializedResult.meta !== void 0 && {
-          output_meta: serializedResult.meta
+        ...serializedInputs !== void 0 && {
+          input: serializedInputs.json,
+          ...serializedInputs.meta !== void 0 && {
+            input_meta: serializedInputs.meta
+          }
+        },
+        ...serializedResult !== void 0 && {
+          output: serializedResult.json,
+          ...serializedResult.meta !== void 0 && {
+            output_meta: serializedResult.meta
+          }
         },
         ...params.functionName !== void 0 && {
           function_name: params.functionName
         },
-        ...params.error !== void 0 && {
+        ...params.captureContent && params.error !== void 0 && {
           error: params.error,
           error_source: "code"
         },
-        ...params.contexts && params.contexts.length > 0 && {
+        ...params.captureContent && params.contexts && params.contexts.length > 0 && {
           contexts: params.contexts
         },
-        ...params.prompt !== void 0 && { prompt: params.prompt }
+        ...params.captureContent && params.prompt !== void 0 && { prompt: params.prompt }
       }
     };
     if (params.parentSpanId) {
@@ -6778,6 +7069,13 @@ var finalizers = {
 // src/index.ts
 init_http();
 init_replay();
+
+// src/replayRegistry.ts
+init_errors();
+init_replay();
+function defineReplayRegistry(registry) {
+  return registry;
+}
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   BITFAB_PROGRESS_PREFIX,
@@ -6796,6 +7094,7 @@ init_replay();
   ReplayError,
   SUPPORTED_PROVIDERS,
   __version__,
+  defineReplayRegistry,
   finalizers,
   flushTraces,
   getCurrentReplayBranch,