@bitfab/sdk 0.19.0 → 0.20.0

package/dist/{chunk-FA6DBCAT.js → chunk-IUZIGC6T.js}

@@ -10,7 +10,7 @@ import {
 } from "./chunk-EQI6ZJC3.js";
 
 // src/version.generated.ts
-var __version__ = "0.19.0";
+var __version__ = "0.20.0";
 
 // src/constants.ts
 var DEFAULT_SERVICE_URL = "https://bitfab.ai";
@@ -287,7 +287,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;
@@ -307,6 +307,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
@@ -429,23 +432,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;
 }
@@ -1234,7 +1244,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) {
@@ -1243,10 +1253,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;
     }
@@ -1261,25 +1271,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;
     }
@@ -1287,7 +1297,7 @@ function normalizeTokenUsage(raw) {
       inputTokens,
       outputTokens,
       totalTokens,
-      cachedInputTokens: asTokenCount(inputDetails.cache_read)
+      cachedInputTokens: asTokenCount2(inputDetails.cache_read)
     });
   }
   return null;
@@ -2984,7 +2994,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-QAWGVRCZ.js");
+    const { replay: doReplay } = await import("./replay-V6RPJYXZ.js");
     return doReplay(
       this.httpClient,
       this.serviceUrl,
@@ -3063,4 +3073,4 @@ export {
   Bitfab,
   BitfabFunction
 };
-//# sourceMappingURL=chunk-FA6DBCAT.js.map
+//# sourceMappingURL=chunk-IUZIGC6T.js.map