@bitfab/sdk 0.28.7 → 0.28.10

package/dist/index.d.cts

@@ -1015,12 +1015,14 @@ interface CurrentTrace {
      */
     addContext(context: Record<string, unknown>): void;
     /**
-     * Flag this trace to be dropped. When the trace completes, the server scrubs
-     * its payloads (trace, external trace, and all spans) and marks it `dropped`
-     * instead of `completed`, retaining only a skeleton audit record. Use this to
-     * discard runs you never want stored (e.g. health checks, or a run you know
-     * contains sensitive data). Takes effect at trace completion; a trace that is
-     * flagged but never completes is not scrubbed.
+     * Flag this trace to be dropped. Once flagged, spans that complete afterward
+     * are not uploaded at all, and when the trace completes the server scrubs any
+     * payloads that already raced out (trace, external trace, and sibling spans),
+     * marking it `dropped` instead of `completed` and retaining only a skeleton
+     * audit record. Use this to discard runs you never want stored (e.g. health
+     * checks, or a run you know contains sensitive data). Takes effect
+     * immediately for later spans; the server-side scrub takes effect at trace
+     * completion, so a trace that is flagged but never completes is not scrubbed.
      */
     drop(): void;
 }
@@ -1612,7 +1614,7 @@ declare class BitfabFunction {
 /**
  * SDK version from package.json (injected at build time)
  */
-declare const __version__ = "0.28.7";
+declare const __version__ = "0.28.10";
 
 /**
  * Constants for the Bitfab SDK.

package/dist/index.d.ts

@@ -1015,12 +1015,14 @@ interface CurrentTrace {
      */
     addContext(context: Record<string, unknown>): void;
     /**
-     * Flag this trace to be dropped. When the trace completes, the server scrubs
-     * its payloads (trace, external trace, and all spans) and marks it `dropped`
-     * instead of `completed`, retaining only a skeleton audit record. Use this to
-     * discard runs you never want stored (e.g. health checks, or a run you know
-     * contains sensitive data). Takes effect at trace completion; a trace that is
-     * flagged but never completes is not scrubbed.
+     * Flag this trace to be dropped. Once flagged, spans that complete afterward
+     * are not uploaded at all, and when the trace completes the server scrubs any
+     * payloads that already raced out (trace, external trace, and sibling spans),
+     * marking it `dropped` instead of `completed` and retaining only a skeleton
+     * audit record. Use this to discard runs you never want stored (e.g. health
+     * checks, or a run you know contains sensitive data). Takes effect
+     * immediately for later spans; the server-side scrub takes effect at trace
+     * completion, so a trace that is flagged but never completes is not scrubbed.
      */
     drop(): void;
 }
@@ -1612,7 +1614,7 @@ declare class BitfabFunction {
 /**
  * SDK version from package.json (injected at build time)
  */
-declare const __version__ = "0.28.7";
+declare const __version__ = "0.28.10";
 
 /**
  * Constants for the Bitfab SDK.

package/dist/index.js

@@ -14,7 +14,7 @@ import {
   flushTraces,
   getCurrentSpan,
   getCurrentTrace
-} from "./chunk-TIMBGA3A.js";
+} from "./chunk-SK4TNXRC.js";
 import {
   BITFAB_PROGRESS_PREFIX,
   BitfabError,

package/dist/node.cjs

@@ -697,7 +697,7 @@ registerAsyncLocalStorageClass(
 );
 
 // src/version.generated.ts
-var __version__ = "0.28.7";
+var __version__ = "0.28.10";
 
 // src/constants.ts
 var DEFAULT_SERVICE_URL = "https://bitfab.ai";
@@ -3143,8 +3143,11 @@ init_warnOnce();
 var activeTraceStates = /* @__PURE__ */ new Map();
 var pendingSpanPromises = /* @__PURE__ */ new Map();
 var asyncLocalStorage = null;
+var initializeAsyncContext = () => {
+  asyncLocalStorage ?? (asyncLocalStorage = createAsyncLocalStorage());
+};
 var asyncLocalStorageReady = asyncStorageReady.then(() => {
-  asyncLocalStorage = createAsyncLocalStorage();
+  initializeAsyncContext();
 });
 var browserSpanStack = [];
 function getSpanStack() {
@@ -3913,6 +3916,7 @@ var Bitfab = class {
       if (!self.isTracingEnabled()) {
         return fn.apply(this, args);
       }
+      initializeAsyncContext();
       if (!asyncLocalStorage && !isAsyncStorageInitDone()) {
         return asyncLocalStorageReady.then(
           () => wrappedFn.apply(this, args)
@@ -3976,13 +3980,16 @@ var Bitfab = class {
           }
           try {
             const endedAt = (/* @__PURE__ */ new Date()).toISOString();
-            const spanPromise = self.sendWrapperSpan({
+            const traceDropped = activeTraceStates.get(traceId)?.dropped === true;
+            const spanPromise = traceDropped ? Promise.resolve() : self.sendWrapperSpan({
               ...baseSpanParams,
               ...params,
               contexts: newContext.contexts,
               prompt: newContext.prompt,
               endedAt,
-              ...replayCtx?.testRunId && { testRunId: replayCtx.testRunId },
+              ...replayCtx?.testRunId && {
+                testRunId: replayCtx.testRunId
+              },
               ...replayCtx?.inputSourceSpanId && {
                 inputSourceSpanId: replayCtx.inputSourceSpanId
               }
@@ -4070,7 +4077,7 @@ var Bitfab = class {
                   meta: mockSpan.outputMeta
                 });
               }
-              void sendSpan({ result: output });
+              void sendSpan({ result: output, mocked: true });
               if (fnReturnsPromise) {
                 return Promise.resolve(output);
               }
@@ -4322,7 +4329,8 @@ var Bitfab = class {
       sourceTraceId: params.traceId,
       traceFunctionKey: params.traceFunctionKey,
       rawSpan: externalSpan,
-      ...params.testRunId && { testRunId: params.testRunId }
+      ...params.testRunId && { testRunId: params.testRunId },
+      ...params.mocked && { mocked: true }
     });
   }
   /**