@bitfab/sdk 0.19.0 → 0.20.0

package/dist/index.d.cts

@@ -396,6 +396,13 @@ interface ReplayOptions {
     environment?: ReplayEnvironment;
     /** Group ID to associate this replay with an experiment group for live streaming in Studio. */
     experimentGroupId?: string;
+    /**
+     * Dataset this replay runs against. When set, the resulting experiment is
+     * durably attributed to the dataset (stored on the test run), so it appears
+     * under the dataset's experiments even if the trace lineage can't be
+     * reconstructed. Validated server-side against the org.
+     */
+    datasetId?: string;
     /**
      * Reshape recorded inputs before they are spread into `fn`.
      *
@@ -1063,7 +1070,7 @@ declare class BitfabFunction {
 /**
  * SDK version from package.json (injected at build time)
  */
-declare const __version__ = "0.19.0";
+declare const __version__ = "0.20.0";
 
 /**
  * Constants for the Bitfab SDK.

package/dist/index.d.ts

@@ -396,6 +396,13 @@ interface ReplayOptions {
     environment?: ReplayEnvironment;
     /** Group ID to associate this replay with an experiment group for live streaming in Studio. */
     experimentGroupId?: string;
+    /**
+     * Dataset this replay runs against. When set, the resulting experiment is
+     * durably attributed to the dataset (stored on the test run), so it appears
+     * under the dataset's experiments even if the trace lineage can't be
+     * reconstructed. Validated server-side against the org.
+     */
+    datasetId?: string;
     /**
      * Reshape recorded inputs before they are spread into `fn`.
      *
@@ -1063,7 +1070,7 @@ declare class BitfabFunction {
 /**
  * SDK version from package.json (injected at build time)
  */
-declare const __version__ = "0.19.0";
+declare const __version__ = "0.20.0";
 
 /**
  * Constants for the Bitfab SDK.

package/dist/index.js

@@ -11,7 +11,7 @@ import {
   flushTraces,
   getCurrentSpan,
   getCurrentTrace
-} from "./chunk-FA6DBCAT.js";
+} from "./chunk-IUZIGC6T.js";
 import {
   BitfabError
 } from "./chunk-EQI6ZJC3.js";

package/dist/node.cjs

@@ -401,7 +401,8 @@ async function replay(httpClient, serviceUrl, traceFunctionKey, fn, options) {
     options?.codeChangeFiles,
     options?.environment !== void 0,
     // includeDbBranchLease
-    options?.experimentGroupId
+    options?.experimentGroupId,
+    options?.datasetId
   );
   const mockStrategy = options?.mock ?? "none";
   const maxConcurrency = options?.maxConcurrency ?? 10;
@@ -502,7 +503,7 @@ registerAsyncLocalStorageClass(
 );
 
 // src/version.generated.ts
-var __version__ = "0.19.0";
+var __version__ = "0.20.0";
 
 // src/constants.ts
 var DEFAULT_SERVICE_URL = "https://bitfab.ai";
@@ -780,7 +781,7 @@ var HttpClient = class {
    * Start a replay session by fetching historical traces.
    * Blocking call — creates a test run and returns lightweight item references.
    */
-  async startReplay(traceFunctionKey, limit, traceIds, codeChangeDescription, codeChangeFiles, includeDbBranchLease, experimentGroupId) {
+  async startReplay(traceFunctionKey, limit, traceIds, codeChangeDescription, codeChangeFiles, includeDbBranchLease, experimentGroupId, datasetId) {
     const payload = { traceFunctionKey };
     if (limit !== void 0) {
       payload.limit = limit;
@@ -800,6 +801,9 @@ var HttpClient = class {
     if (experimentGroupId !== void 0) {
       payload.experimentGroupId = experimentGroupId;
     }
+    if (datasetId !== void 0) {
+      payload.datasetId = datasetId;
+    }
     const timeout = includeDbBranchLease ? 18e4 : 3e4;
     return this.request("/api/sdk/replay/start", payload, {
       timeout
@@ -923,23 +927,30 @@ function extractContentBlocks(content) {
   }
   return content.map((block) => safeSerialize(block));
 }
+function asTokenCount(val) {
+  return typeof val === "number" && Number.isFinite(val) ? val : null;
+}
 function extractUsage(message) {
   const usageInfo = {};
   const usage = message.usage;
   if (!usage) {
     return usageInfo;
   }
-  const mapping = {
-    input_tokens: "inputTokens",
-    output_tokens: "outputTokens",
-    cache_read_input_tokens: "cacheReadTokens",
-    cache_creation_input_tokens: "cacheCreationTokens"
-  };
-  for (const [srcKey, dstKey] of Object.entries(mapping)) {
-    const val = usage[srcKey];
-    if (val !== void 0 && val !== null) {
-      usageInfo[dstKey] = val;
-    }
+  const baseInput = asTokenCount(usage.input_tokens);
+  const cacheRead = asTokenCount(usage.cache_read_input_tokens);
+  const cacheCreation = asTokenCount(usage.cache_creation_input_tokens);
+  if (baseInput !== null || cacheRead !== null || cacheCreation !== null) {
+    usageInfo.inputTokens = (baseInput ?? 0) + (cacheRead ?? 0) + (cacheCreation ?? 0);
+  }
+  const output = asTokenCount(usage.output_tokens);
+  if (output !== null) {
+    usageInfo.outputTokens = output;
+  }
+  if (cacheRead !== null) {
+    usageInfo.cacheReadTokens = cacheRead;
+  }
+  if (cacheCreation !== null) {
+    usageInfo.cacheCreationTokens = cacheCreation;
   }
   return usageInfo;
 }
@@ -1733,7 +1744,7 @@ function extractModelName(serialized, metadata) {
   }
   return void 0;
 }
-function asTokenCount(value) {
+function asTokenCount2(value) {
   return typeof value === "number" && Number.isFinite(value) ? value : null;
 }
 function normalizeTokenUsage(raw) {
@@ -1742,10 +1753,10 @@ function normalizeTokenUsage(raw) {
   }
   const u = raw;
   if ("cache_read_input_tokens" in u || "cache_creation_input_tokens" in u) {
-    const cacheRead = asTokenCount(u.cache_read_input_tokens);
-    const cacheCreation = asTokenCount(u.cache_creation_input_tokens);
-    const baseInput = asTokenCount(u.input_tokens);
-    const outputTokens = asTokenCount(u.output_tokens);
+    const cacheRead = asTokenCount2(u.cache_read_input_tokens);
+    const cacheCreation = asTokenCount2(u.cache_creation_input_tokens);
+    const baseInput = asTokenCount2(u.input_tokens);
+    const outputTokens = asTokenCount2(u.output_tokens);
     if (cacheRead === null && cacheCreation === null && baseInput === null && outputTokens === null) {
       return null;
     }
@@ -1760,25 +1771,25 @@ function normalizeTokenUsage(raw) {
   if ("prompt_tokens" in u || "completion_tokens" in u || "promptTokens" in u || "completionTokens" in u) {
     const promptDetails = u.prompt_tokens_details ?? {};
     return withAnyTokenCount({
-      inputTokens: asTokenCount(u.prompt_tokens) ?? asTokenCount(u.promptTokens),
-      outputTokens: asTokenCount(u.completion_tokens) ?? asTokenCount(u.completionTokens),
-      totalTokens: asTokenCount(u.total_tokens) ?? asTokenCount(u.totalTokens),
-      cachedInputTokens: asTokenCount(promptDetails.cached_tokens)
+      inputTokens: asTokenCount2(u.prompt_tokens) ?? asTokenCount2(u.promptTokens),
+      outputTokens: asTokenCount2(u.completion_tokens) ?? asTokenCount2(u.completionTokens),
+      totalTokens: asTokenCount2(u.total_tokens) ?? asTokenCount2(u.totalTokens),
+      cachedInputTokens: asTokenCount2(promptDetails.cached_tokens)
     });
   }
   if ("prompt_token_count" in u || "candidates_token_count" in u) {
     return withAnyTokenCount({
-      inputTokens: asTokenCount(u.prompt_token_count),
-      outputTokens: asTokenCount(u.candidates_token_count),
-      totalTokens: asTokenCount(u.total_token_count),
-      cachedInputTokens: asTokenCount(u.cached_content_token_count)
+      inputTokens: asTokenCount2(u.prompt_token_count),
+      outputTokens: asTokenCount2(u.candidates_token_count),
+      totalTokens: asTokenCount2(u.total_token_count),
+      cachedInputTokens: asTokenCount2(u.cached_content_token_count)
     });
   }
   if ("input_tokens" in u || "output_tokens" in u) {
     const inputDetails = u.input_token_details ?? {};
-    const inputTokens = asTokenCount(u.input_tokens);
-    const outputTokens = asTokenCount(u.output_tokens);
-    let totalTokens = asTokenCount(u.total_tokens);
+    const inputTokens = asTokenCount2(u.input_tokens);
+    const outputTokens = asTokenCount2(u.output_tokens);
+    let totalTokens = asTokenCount2(u.total_tokens);
     if (totalTokens === null && inputTokens !== null && outputTokens !== null) {
       totalTokens = inputTokens + outputTokens;
     }
@@ -1786,7 +1797,7 @@ function normalizeTokenUsage(raw) {
       inputTokens,
       outputTokens,
       totalTokens,
-      cachedInputTokens: asTokenCount(inputDetails.cache_read)
+      cachedInputTokens: asTokenCount2(inputDetails.cache_read)
     });
   }
   return null;