bitfab 0.54.12 → 0.54.14

package/dist/index.cjs

@@ -51,7 +51,7 @@ var __version__, __packageName__;
 var init_version_generated = __esm({
   "src/version.generated.ts"() {
     "use strict";
-    __version__ = "0.54.12";
+    __version__ = "0.54.14";
     __packageName__ = "bitfab";
   }
 });
@@ -1265,6 +1265,7 @@ var init_simulationPlan = __esm({
         }
         this.warnNeverLoaded();
         this.warnDropped();
+        return this.held.length === 0;
       }
       waitForFirstLoad(timeoutMs) {
         const firstLoad = this.firstLoad();
@@ -2087,9 +2088,14 @@ __export(http_exports, {
 });
 async function releaseHeldExternalSpans(timeoutMs) {
   const deadline = Date.now() + Math.max(timeoutMs, 0);
+  let released = true;
   for (const client of [...liveHttpClients]) {
-    await client.releaseHeldExternalSpans?.(Math.max(0, deadline - Date.now()));
+    const result = await client.releaseHeldExternalSpans?.(
+      Math.max(0, deadline - Date.now())
+    );
+    released = result !== false && released;
   }
+  return released;
 }
 function stopSimulationPlans() {
   for (const client of [...liveHttpClients]) {
@@ -2106,14 +2112,15 @@ function awaitOnExit(promise) {
 }
 async function flushTraces(timeoutMs = 5e3) {
   const deadline = Date.now() + Math.max(timeoutMs, 0);
-  await releaseHeldExternalSpans(planWaitSliceMs(timeoutMs));
+  const released = await releaseHeldExternalSpans(planWaitSliceMs(timeoutMs));
   const requestsFlushed = await awaitPendingRequests(
     Math.max(0, deadline - Date.now())
   );
+  const releasedAfterDeferred = await releaseHeldExternalSpans(0);
   const transportsFlushed = await flushTraceTransports(
     Math.max(0, deadline - Date.now())
   );
-  return requestsFlushed && transportsFlushed;
+  return released && releasedAfterDeferred && requestsFlushed && transportsFlushed;
 }
 async function awaitPendingRequests(timeoutMs = DEFAULT_LIFECYCLE_TIMEOUT_MS2) {
   await replayContextReady.catch(() => {
@@ -2523,13 +2530,17 @@ var init_http = __esm({
        */
       async settleDeferredWork(timeoutMs = DEFAULT_LIFECYCLE_TIMEOUT_MS2) {
         const deadline = Date.now() + Math.max(timeoutMs, 0);
-        await this.releaseHeldExternalSpans?.(planWaitSliceMs(timeoutMs));
+        const released = await this.releaseHeldExternalSpans?.(
+          planWaitSliceMs(timeoutMs)
+        );
         await replayContextReady.catch(() => {
         });
-        return waitForPromises(
+        const settled2 = await waitForPromises(
           Array.from(this.deferredWork),
           Math.max(0, deadline - Date.now())
         );
+        const releasedAfterDeferred = await this.releaseHeldExternalSpans?.(0);
+        return released !== false && releasedAfterDeferred !== false && settled2;
       }
       /**
        * Wait for spans queued by this client to be delivered, within one deadline.
@@ -2554,7 +2565,9 @@ var init_http = __esm({
         }
         const deadline = Date.now() + Math.max(timeoutMs, 0);
         this.closing = (async () => {
-          await this.releaseHeldExternalSpans?.(planWaitSliceMs(timeoutMs));
+          const released = await this.releaseHeldExternalSpans?.(
+            planWaitSliceMs(timeoutMs)
+          );
           this.stopSimulationPlan?.();
           this.externalSpanTransform = void 0;
           this.externalTraceTransform = void 0;
@@ -2568,7 +2581,7 @@ var init_http = __esm({
           const transport = this.traceTransport;
           this.traceTransport = void 0;
           const shutdownOk = await transport?.shutdown(Math.max(0, deadline - Date.now())) ?? true;
-          return settled2 && shutdownOk;
+          return released !== false && settled2 && shutdownOk;
         })();
         return this.closing;
       }
@@ -2666,12 +2679,6 @@ 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 getSimulationPlan() {
         return this.get(
           "/api/sdk/sim-plan",
@@ -4008,6 +4015,12 @@ async function replay(httpClient, serviceUrl, traceFunctionKey, fn, options, reg
     }
   }
   const attempts = resolveAttempts(options?.attempts);
+  const maxConcurrency = options?.maxConcurrency ?? 10;
+  if (!Number.isInteger(maxConcurrency) || maxConcurrency < 1) {
+    throw new BitfabError(
+      `maxConcurrency must be a positive integer (got ${maxConcurrency}).`
+    );
+  }
   await replayContextReady;
   let codeChangeDescription = options?.codeChangeDescription;
   let codeChangeFiles = options?.codeChangeFiles;
@@ -4054,7 +4067,6 @@ async function replay(httpClient, serviceUrl, traceFunctionKey, fn, options, reg
     }
   }
   const mockStrategy = options?.mock ?? "marked";
-  const maxConcurrency = options?.maxConcurrency ?? 10;
   const fullTestRunUrl = `${serviceUrl}${testRunUrl}`;
   const resolvedOverrides = [
     ...normalizeMockOverrides(options?.mockOverride),
@@ -4151,6 +4163,7 @@ async function replay(httpClient, serviceUrl, traceFunctionKey, fn, options, reg
             originalDurationMs: item.originalDurationMs,
             originalTokens: item.originalTokens,
             originalModel: item.originalModel,
+            ingestionType: item.ingestionType,
             tokens: item.tokens,
             model: item.model,
             dbSnapshotRef: item.dbSnapshotRef,
@@ -5195,21 +5208,6 @@ function currentAutoTraceScope() {
   initializeAutoTraceStorage();
   return autoTraceState.storage?.getStore() ?? autoTraceState.browserScope;
 }
-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);
-}
 
 // src/optionalPeer.ts
 function importOptionalPeer(specifierParts) {
@@ -5999,6 +5997,14 @@ var LabelsClient = class {
     );
     return response.labels;
   }
+  /** Withhold a verdict for a trace or replay attempt whose check did not run. */
+  async skip(target, testRunId) {
+    return this.save({ ...target, skip: true }, testRunId);
+  }
+  /** Clear a previous verdict for a trace, replay attempt, or assertion. */
+  async archive(target, testRunId) {
+    return this.save({ ...target, archive: true }, testRunId);
+  }
   async saveHuman(update) {
     const [outcome] = await this.saveHumanAll([update]);
     return outcome;
@@ -6917,6 +6923,15 @@ var BitfabLangGraphIntegration = class {
 // src/client.ts
 init_mockOverride();
 
+// src/streamFinalizationError.ts
+var StreamFinalizationError = class extends Error {
+  constructor(message, output) {
+    super(message);
+    this.output = output;
+    this.name = "StreamFinalizationError";
+  }
+};
+
 // src/openaiAgentSdk.ts
 var BitfabOpenAIAgentHandler = class {
   constructor(config) {
@@ -6956,7 +6971,11 @@ var BitfabOpenAIAgentHandler = class {
       if (isStreaming && res?.completed) {
         try {
           await res.completed;
-        } catch {
+        } catch (error) {
+          throw new StreamFinalizationError(
+            error instanceof Error ? error.message : String(error),
+            res.finalOutput
+          );
         }
       }
       return res?.finalOutput;
@@ -7058,7 +7077,6 @@ function openAiAgentsParentSpanId(bitfabTraceId, enclosingSpanIds) {
 }
 
 // src/client.ts
-init_policyRefresh();
 init_randomUuid();
 init_replay();
 
@@ -7715,17 +7733,6 @@ function runWithSeedContext(ctx, fn) {
 init_serialize();
 init_simulationPlan();
 init_spanOrigin();
-
-// src/streamFinalizationError.ts
-var StreamFinalizationError = class extends Error {
-  constructor(message, output) {
-    super(message);
-    this.output = output;
-    this.name = "StreamFinalizationError";
-  }
-};
-
-// src/client.ts
 init_traceMetadata();
 
 // src/traces.ts
@@ -8642,8 +8649,6 @@ function readBooleanEnv(name) {
 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 spanOwnerName(receiver) {
   if (typeof receiver === "function") {
     return receiver.name !== "" ? receiver.name : void 0;
@@ -8676,7 +8681,6 @@ 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
@@ -8725,12 +8729,12 @@ var Bitfab = class {
    *
    * Build instrumentation turns repository functions called beneath this
    * method into nested spans that capture inputs, outputs, and errors by
-   * default. A confirmed capture policy can narrow rich capture to selected
-   * function IDs.
+   * default. Source exclusions and sim-plan capture-off decisions are explicit
+   * opt-outs from that default.
    * 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 traceFunctionKey - Groups traces and their capture controls.
    * @param options - Root presentation, subtree bounds, and exclusions.
    * @experimental Automatic child-call instrumentation is experimental.
    */
@@ -8888,11 +8892,6 @@ var Bitfab = class {
       traceFunctionKey,
       spanOptions,
       function(...args) {
-        const capturePolicy = getAutoTraceCapturePolicy(
-          self,
-          traceFunctionKey
-        );
-        self.refreshAutoTraceCapturePolicy(traceFunctionKey);
         self.simulationPlan.refresh();
         let spansUsed = 0;
         let truncated = false;
@@ -8972,7 +8971,7 @@ var Bitfab = class {
               surface: "opt-out",
               instrumentation: "trace",
               functionId: definition.id,
-              captureContent: nodeConfiguration !== void 0 || capturePolicy === void 0 || capturePolicy.has(definition.id),
+              captureContent: true,
               autoTraceDefinition: definition,
               ...nodeConfiguration?.testRunId !== void 0 && {
                 testRunId: nodeConfiguration.testRunId
@@ -9071,34 +9070,6 @@ var Bitfab = class {
   applySimulationPlan(payload, submit) {
     this.simulationPlan.send(payload, rootTraceFunctionKeyOf(payload), submit);
   }
-  refreshAutoTraceCapturePolicy(traceFunctionKey) {
-    const state2 = this.autoTracePolicyRefreshes.get(traceFunctionKey) ?? new PolicyRefresh();
-    this.autoTracePolicyRefreshes.set(traceFunctionKey, state2);
-    state2.run(
-      () => this.httpClient.getAutoTracePolicy(
-        traceFunctionKey,
-        AUTO_TRACE_PROTOCOL
-      ),
-      (policy) => {
-        if (policy.protocol !== AUTO_TRACE_PROTOCOL) {
-          state2.hold(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,
-          policy.revision === null ? void 0 : functionIds
-        );
-        state2.hold(AUTO_TRACE_POLICY_REFRESH_MS);
-      },
-      () => {
-        state2.hold(AUTO_TRACE_POLICY_RETRY_MS);
-      }
-    );
-  }
   /**
    * Flush and permanently close this client's tracing resources: its pending
    * requests and the single span-transport worker shared by its decorators and