bitfab 0.57.8 → 0.58.1

package/dist/{chunk-CPVOJHEH.js → chunk-4GTG3VPY.js}

@@ -6,7 +6,7 @@ import {
   replayContextReady,
   runWithReplayContext,
   warnOnce
-} from "./chunk-HKRCT4OE.js";
+} from "./chunk-EQLQVY2O.js";
 import {
   runWithCaptureSuppressed
 } from "./chunk-VCRFFCLY.js";
@@ -1263,4 +1263,4 @@ export {
   sleepForReplayPersistence,
   replay
 };
-//# sourceMappingURL=chunk-CPVOJHEH.js.map
+//# sourceMappingURL=chunk-4GTG3VPY.js.map

package/dist/{chunk-2E7Q52KB.js → chunk-756R63DD.js}

@@ -17,10 +17,11 @@ import {
   serializeValue,
   toJsonSafe,
   toJsonSafeReport
-} from "./chunk-CPVOJHEH.js";
+} from "./chunk-4GTG3VPY.js";
 import {
   BitfabError,
   CONTENT_OFF_KEY,
+  DECLARED_NODE_FIELD,
   DEFAULT_SERVICE_URL,
   HttpClient,
   MixedTracingError,
@@ -33,11 +34,9 @@ import {
   makeSpanOrigin,
   readEnv,
   recordCallerTraceMetadata,
-  recordedByFramework,
   retireTraceMetadata,
-  traceIdOf,
   warnOnce
-} from "./chunk-HKRCT4OE.js";
+} from "./chunk-EQLQVY2O.js";
 import {
   __privateAdd,
   __privateGet,
@@ -296,11 +295,6 @@ var BitfabClaudeAgentHandler = class {
     }
     this.runToSpan.set(spanId, spanInfo);
     this.httpClient.traceCompletion.start(spanInfo.traceId, spanInfo.spanId);
-    this.httpClient.reserveFrameworkSpan(
-      spanInfo.traceId,
-      spanInfo.spanId,
-      spanInfo.parentId
-    );
     return spanInfo;
   }
   completeSpan(spanId, output, error, extraContexts) {
@@ -580,11 +574,6 @@ var BitfabClaudeAgentHandler = class {
       this.conversationHistory.push(...this.pendingMessages);
       this.pendingMessages = [];
       this.currentLlmSpanId = randomUuid();
-      this.httpClient.reserveFrameworkSpan(
-        this.ensureTrace(),
-        this.currentLlmSpanId,
-        this.getParentId()
-      );
       this.currentLlmMessageId = messageId ?? null;
       this.currentLlmContent = [];
       this.currentLlmModel = inner.model ?? null;
@@ -1825,11 +1814,6 @@ var BitfabLangGraphCallbackHandler = class {
     this.runToSpan.set(runId, spanInfo);
     if (spanInfo.hidden !== true) {
       this.httpClient.traceCompletion.start(spanInfo.traceId, spanInfo.spanId);
-      this.httpClient.reserveFrameworkSpan(
-        spanInfo.traceId,
-        spanInfo.spanId,
-        spanInfo.parentId
-      );
     }
     if (isRootInvocation) {
       this.sendTraceStart(spanInfo);
@@ -3349,11 +3333,6 @@ var BitfabOpenAITracingProcessor = class {
       const mapping = this.activeSpanMappings[span.traceId];
       const traceId = mapping?.traceId ?? span.traceId;
       this.httpClient.traceCompletion.start(traceId, span.spanId);
-      this.httpClient.reserveFrameworkSpan(
-        traceId,
-        span.spanId,
-        span.parentId ?? mapping?.spanId
-      );
     }
   }
   /**
@@ -3741,56 +3720,9 @@ var BitfabVercelAiHandler = class {
 // src/client.ts
 var TRUNCATION_LIMIT_ORDER = [
   "max_depth",
-  "max_spans",
   "max_captured_subtree_spans"
 ];
 var activeTraceStates = /* @__PURE__ */ new Map();
-var REFUSED_FRAMEWORK_SPAN_TRACES_RETAINED = 1024;
-var refusedFrameworkSpans = /* @__PURE__ */ new Map();
-function frameworkSpanRefused(traceId, spanId) {
-  return refusedFrameworkSpans.get(traceId)?.has(spanId) === true;
-}
-function refuseFrameworkSpan(traceId, spanId) {
-  const refused = refusedFrameworkSpans.get(traceId);
-  if (refused !== void 0) {
-    refused.add(spanId);
-    return;
-  }
-  refusedFrameworkSpans.set(traceId, /* @__PURE__ */ new Set([spanId]));
-  while (refusedFrameworkSpans.size > REFUSED_FRAMEWORK_SPAN_TRACES_RETAINED) {
-    const oldest = refusedFrameworkSpans.keys().next();
-    if (oldest.done) {
-      return;
-    }
-    refusedFrameworkSpans.delete(oldest.value);
-  }
-}
-function reserveFrameworkSpan(traceId, spanId, parentSpanId) {
-  const spanBudget = activeTraceStates.get(traceId)?.spanBudget;
-  const parentRefused = parentSpanId !== null && frameworkSpanRefused(traceId, parentSpanId);
-  if (!parentRefused && (spanBudget === void 0 || spanBudget.spansUsed < spanBudget.maxSpans)) {
-    if (spanBudget !== void 0) {
-      spanBudget.spansUsed += 1;
-    }
-    return;
-  }
-  refuseFrameworkSpan(traceId, spanId);
-  spanBudget?.truncate("max_spans");
-}
-function isRefusedFrameworkSpan(payload) {
-  if (!recordedByFramework(payload)) {
-    return false;
-  }
-  const rawSpan = payload.rawSpan;
-  const spanId = rawSpan?.id;
-  if (typeof spanId !== "string") {
-    return false;
-  }
-  const traceIds = [payload.sourceTraceId, traceIdOf(payload)];
-  return traceIds.some(
-    (traceId) => typeof traceId === "string" && frameworkSpanRefused(traceId, spanId)
-  );
-}
 function truncationMetadata(truncation) {
   if (truncation === void 0 || truncation.limits.size === 0) {
     return void 0;
@@ -4180,7 +4112,6 @@ function readBooleanEnv(name) {
   return void 0;
 }
 var DEFAULT_AUTO_TRACE_MAX_DEPTH = 30;
-var DEFAULT_AUTO_TRACE_MAX_SPANS = 1e4;
 var DEFAULT_AUTO_TRACE_MAX_CAPTURED_SUBTREE_SPANS = 500;
 function spanOwnerName(receiver) {
   if (typeof receiver === "function") {
@@ -4248,7 +4179,6 @@ var Bitfab = class {
       config.simulationPlan ?? true
     );
     this.httpClient.externalSpanTransform = (payload, submit, drop) => this.applySimulationPlan(payload, submit, drop);
-    this.httpClient.frameworkSpanStart = reserveFrameworkSpan;
     this.httpClient.externalTraceTransform = (payload, submit) => this.simulationPlan.sendTrace(payload, submit);
     this.httpClient.releaseHeldExternalSpans = (timeoutMs) => this.simulationPlan.release(timeoutMs);
     this.httpClient.stopSimulationPlan = () => this.simulationPlan.stop();
@@ -4411,21 +4341,9 @@ var Bitfab = class {
       options.maxDepth,
       DEFAULT_AUTO_TRACE_MAX_DEPTH
     );
-    if (options.maxSpans !== void 0 && options.maxSpans < 1) {
-      throw new BitfabError(
-        `maxSpans must be at least 1 because the root span counts toward it (got ${options.maxSpans})`
-      );
-    }
-    const maxSpans = autoTraceLimit(
-      options.maxSpans,
-      DEFAULT_AUTO_TRACE_MAX_SPANS
-    );
-    const maxCapturedSubtreeSpans = Math.min(
-      autoTraceLimit(
-        options.maxCapturedSubtreeSpans,
-        DEFAULT_AUTO_TRACE_MAX_CAPTURED_SUBTREE_SPANS
-      ),
-      maxSpans
+    const maxCapturedSubtreeSpans = autoTraceLimit(
+      options.maxCapturedSubtreeSpans,
+      DEFAULT_AUTO_TRACE_MAX_CAPTURED_SUBTREE_SPANS
     );
     const rootIsAsync = fn.constructor.name === "AsyncFunction";
     const excluded = new Set(options.exclude ?? []);
@@ -4456,19 +4374,11 @@ var Bitfab = class {
           truncation.droppedSpans += 1;
           warnOnce(
             `auto-trace-truncated:${traceFunctionKey}`,
-            `"${traceFunctionKey}" hit an automatic subtree capture limit (maxDepth=${maxDepth}, maxCapturedSubtreeSpans=${maxCapturedSubtreeSpans}, maxSpans=${maxSpans}); its trace is incomplete. Raise the limits, declare the calls you need with node(), or narrow the subtree with exclude.`
+            `"${traceFunctionKey}" hit an automatic subtree capture limit (maxDepth=${maxDepth}, maxCapturedSubtreeSpans=${maxCapturedSubtreeSpans}); its trace is incomplete. Raise the limits, declare the calls you need with node(), or narrow the subtree with exclude.`
           );
         };
-        const spanBudget = {
-          spansUsed: 1,
-          maxSpans,
-          truncate
-        };
         if (rootTraceState !== void 0) {
           rootTraceState.truncation = truncation;
-          if (rootTraceState.spanBudget === void 0) {
-            rootTraceState.spanBudget = spanBudget;
-          }
         }
         const spanStackOf = (entry) => entry.spanStack;
         const openChild = (entry, childOptions, inputs, isAsync, invokeFn) => {
@@ -4504,16 +4414,16 @@ var Bitfab = class {
           return innermost === void 0 ? traceFunctionKey : rootTraceFunctionKeyOfTrace(innermost.traceId, traceFunctionKey);
         };
         const recordsSpan = (entry, spanName, declared) => {
-          const countsTowardCapturedSubtree = !declared && !self.simulationPlan.contentOffNow(
+          const contentOff = !declared && self.simulationPlan.contentOffNow(
             rootTraceFunctionKeyOfEntry(entry),
             spanName
           );
-          const refusedBy = entry.depth >= maxDepth ? "max_depth" : spanBudget.spansUsed >= maxSpans ? "max_spans" : countsTowardCapturedSubtree && capturedSubtreeCount >= maxCapturedSubtreeSpans ? "max_captured_subtree_spans" : void 0;
+          const countsTowardCapturedSubtree = !declared && !contentOff;
+          const refusedBy = entry.depth >= maxDepth ? "max_depth" : countsTowardCapturedSubtree && capturedSubtreeCount >= maxCapturedSubtreeSpans ? "max_captured_subtree_spans" : void 0;
           if (refusedBy !== void 0) {
             truncate(refusedBy);
             return false;
           }
-          spanBudget.spansUsed += 1;
           if (countsTowardCapturedSubtree) {
             capturedSubtreeCount += 1;
           }
@@ -4532,7 +4442,8 @@ var Bitfab = class {
               return clearAutoTraceNodeConfiguration(invokeFn);
             }
             const spanName = nodeConfiguration?.name ?? definition.name;
-            if (!recordsSpan(entry, spanName, nodeConfiguration !== void 0)) {
+            const declaredNode = nodeConfiguration !== void 0;
+            if (!recordsSpan(entry, spanName, declaredNode)) {
               return invokeWithoutNode();
             }
             const mockOnReplay = nodeConfiguration?.mockOnReplay ?? options.mockOnReplayDefault;
@@ -4545,6 +4456,7 @@ var Bitfab = class {
               functionId: definition.id,
               captureContent: true,
               autoTraceDefinition: definition,
+              ...declaredNode && { declaredNode },
               ...nodeConfiguration?.testRunId !== void 0 && {
                 testRunId: nodeConfiguration.testRunId
               },
@@ -4574,6 +4486,7 @@ var Bitfab = class {
               surface: "opt-out",
               instrumentation: "trace",
               captureContent: true,
+              declaredNode: true,
               ...link !== void 0 && { nestedTrace: link }
             };
             return openChild(entry, childOptions, inputs, false, invokeFn);
@@ -4645,10 +4558,6 @@ var Bitfab = class {
     return autoTraceRoot;
   }
   applySimulationPlan(payload, submit, drop) {
-    if (isRefusedFrameworkSpan(payload)) {
-      drop();
-      return;
-    }
     this.simulationPlan.send(
       payload,
       rootTraceFunctionKeyOf(payload),
@@ -5323,9 +5232,6 @@ var Bitfab = class {
           registeredTraceId = traceId;
         }
         self.httpClient.traceCompletion.start(traceId, spanId);
-        if (isFrameworkInstrumentation(options.instrumentation)) {
-          reserveFrameworkSpan(traceId, spanId, parentSpanId);
-        }
         startedSpan = { traceId, spanId };
         const functionName = fn.name !== "" ? fn.name : void 0;
         const rootTraceFunctionKey = rootTraceFunctionKeyOfTrace(
@@ -5347,6 +5253,7 @@ var Bitfab = class {
           functionId: options.functionId,
           captureContent: options.captureContent ?? true,
           autoTraceDefinition: options.autoTraceDefinition,
+          declaredNode: options.declaredNode === true,
           nestedTrace: options.nestedTrace,
           enclosingTrace: options.enclosingTrace
         };
@@ -5872,7 +5779,7 @@ var Bitfab = class {
    * Queued on the client's span transport; delivery is the transport's job.
    */
   sendWrapperSpan(params) {
-    const contentOff = this.contentOffBySimulationPlan(params);
+    const contentOff = params.declaredNode !== true && this.contentOffBySimulationPlan(params);
     const buildsContent = params.captureContent && !contentOff;
     const serializedInputs = buildsContent ? serializeValue(params.inputs) : void 0;
     const serializedResult = buildsContent ? serializeValue(params.result) : void 0;
@@ -5946,6 +5853,7 @@ var Bitfab = class {
       ...params.rootTraceFunctionKey !== void 0 && {
         [ROOT_TRACE_FUNCTION_KEY_FIELD]: params.rootTraceFunctionKey
       },
+      ...params.declaredNode === true && { [DECLARED_NODE_FIELD]: true },
       rawSpan: externalSpan,
       ...params.testRunId && { testRunId: params.testRunId },
       ...params.mocked && { mocked: true },
@@ -6093,7 +6001,7 @@ var Bitfab = class {
       await runWithSeedContext({ traceId }, async () => target(...args));
     } finally {
       try {
-        const { flushTraces: flushTraces2 } = await import("./http-2EGLRQUQ.js");
+        const { flushTraces: flushTraces2 } = await import("./http-7BKMLT23.js");
         await flushTraces2(3e4);
       } finally {
         unrecorded = activeTraceStates.delete(traceId);
@@ -6152,7 +6060,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-BWUZ2NDQ.js");
+    const { replay: doReplay } = await import("./replay-FMSQGE2H.js");
     return doReplay(
       this.httpClient,
       this.serviceUrl,
@@ -6472,7 +6380,7 @@ async function seedFromRegistry(registry, pipeline, cases, options = {}) {
       sessionId: seedCase.sessionId
     })
   );
-  const { flushTraces: flushTraces2 } = await import("./http-2EGLRQUQ.js");
+  const { flushTraces: flushTraces2 } = await import("./http-7BKMLT23.js");
   await flushTraces2(3e4);
   return { pipeline, traceFunctionKey, traceIds };
 }
@@ -6511,4 +6419,4 @@ export {
   reseedFromRegistry,
   seedFromRegistry
 };
-//# sourceMappingURL=chunk-2E7Q52KB.js.map
+//# sourceMappingURL=chunk-756R63DD.js.map