@bitfab/sdk 0.50.0 → 0.52.0

package/dist/{http-W2YPLY7H.js → http-4A2JEPBD.js}

@@ -6,7 +6,7 @@ import {
   flushTraces,
   parseRetryAfterMs,
   serializePayloadBody
-} from "./chunk-W5P5JAH7.js";
+} from "./chunk-KXBXRR4I.js";
 export {
   BitfabError,
   HttpClient,
@@ -16,4 +16,4 @@ export {
   parseRetryAfterMs,
   serializePayloadBody
 };
-//# sourceMappingURL=http-W2YPLY7H.js.map
+//# sourceMappingURL=http-4A2JEPBD.js.map

package/dist/{http-PFP2EQVE.js → http-W6SXQA3Q.js}

@@ -6,7 +6,7 @@ import {
   flushTraces,
   parseRetryAfterMs,
   serializePayloadBody
-} from "./chunk-XZDYNFI5.js";
+} from "./chunk-4EOEBN3M.js";
 import "./chunk-H6LZRFMN.js";
 export {
   BitfabError,
@@ -17,4 +17,4 @@ export {
   parseRetryAfterMs,
   serializePayloadBody
 };
-//# sourceMappingURL=http-PFP2EQVE.js.map
+//# sourceMappingURL=http-W6SXQA3Q.js.map

package/dist/index.cjs

@@ -42,7 +42,7 @@ var __version__, __packageName__;
 var init_version_generated = __esm({
   "src/version.generated.ts"() {
     "use strict";
-    __version__ = "0.50.0";
+    __version__ = "0.52.0";
     __packageName__ = "@bitfab/sdk";
   }
 });
@@ -1847,7 +1847,7 @@ var init_http = __esm({
        * Start a replay session by fetching historical traces.
        * Blocking call - creates a test run and returns lightweight item references.
        */
-      async startReplay(traceFunctionKey, limit, traceIds, name, codeChangeDescription, codeChangeFiles, includeDbBranchLease, experimentGroupId, datasetId, graderIds, dbBranchSettings, attempts, includeOriginalMetadata, onlyWithAssertions) {
+      async startReplay(traceFunctionKey, limit, traceIds, name, codeChangeDescription, codeChangeFiles, includeDbBranchLease, experimentGroupId, datasetIds, graderIds, dbBranchSettings, attempts, includeOriginalMetadata, onlyWithAssertions) {
         const payload = { traceFunctionKey };
         if (limit !== void 0) {
           payload.limit = limit;
@@ -1871,8 +1871,12 @@ var init_http = __esm({
         if (experimentGroupId !== void 0) {
           payload.experimentGroupId = experimentGroupId;
         }
-        if (datasetId !== void 0) {
-          payload.datasetId = datasetId;
+        if (datasetIds !== void 0) {
+          if (datasetIds.length === 1) {
+            payload.datasetId = datasetIds[0];
+          } else {
+            payload.datasetIds = datasetIds;
+          }
         }
         if (graderIds !== void 0) {
           payload.graderIds = graderIds;
@@ -2474,6 +2478,23 @@ function resolveAttempts(attempts) {
   }
   return attempts;
 }
+function resolveDatasetIds(options) {
+  if (options?.datasetId !== void 0 && options?.datasetIds !== void 0) {
+    throw new BitfabError(
+      "datasetId and datasetIds are the same selector: pass one dataset through datasetId, or several through datasetIds."
+    );
+  }
+  if (options?.datasetId !== void 0) {
+    return [options.datasetId];
+  }
+  if (options?.datasetIds === void 0) {
+    return void 0;
+  }
+  if (options.datasetIds.length === 0) {
+    throw new BitfabError("datasetIds must contain at least one dataset ID.");
+  }
+  return [...new Set(options.datasetIds)];
+}
 function dbBranchEnabled(dbBranch) {
   return dbBranch !== void 0 && dbBranch !== false;
 }
@@ -2906,9 +2927,10 @@ async function replay(httpClient, serviceUrl, traceFunctionKey, fn, options, reg
       );
     }
   }
-  if (options?.traceIds !== void 0 && options?.datasetId !== void 0) {
+  const datasetIds = resolveDatasetIds(options);
+  if (options?.traceIds !== void 0 && datasetIds !== void 0) {
     throw new BitfabError(
-      "traceIds and datasetId select different replay sources and cannot be used together."
+      "traceIds and datasetIds select different replay sources and cannot be used together."
     );
   }
   if (options?.limit !== void 0 && options?.traceIds !== void 0) {
@@ -2950,7 +2972,7 @@ async function replay(httpClient, serviceUrl, traceFunctionKey, fn, options, reg
     // before it could report its resolved inputs.
     dbBranchEnabled(options?.dbBranch) && options?.dryRun !== true,
     options?.experimentGroupId,
-    options?.datasetId,
+    datasetIds,
     options?.graderIds,
     resolveDbBranchSettings(options?.dbBranch),
     attempts,
@@ -3294,7 +3316,8 @@ __export(index_exports, {
 module.exports = __toCommonJS(index_exports);
 
 // src/assertions.ts
-function expectationsPath(traceId, suffix = "") {
+var ASSERTIONS_PATH = "/api/sdk/traces/assertions";
+function traceAssertionsPath(traceId, suffix = "") {
   return `/api/sdk/traces/${encodeURIComponent(traceId)}/assertions${suffix}`;
 }
 var TracesClient = class {
@@ -3302,18 +3325,29 @@ var TracesClient = class {
     this.httpClient = httpClient;
   }
   async getAssertions(traceId) {
-    return this.httpClient.get(expectationsPath(traceId));
+    return this.httpClient.get(
+      traceAssertionsPath(traceId)
+    );
   }
   async saveAssertions(params) {
-    const response = await this.httpClient.request(expectationsPath(params.traceId), {
-      assertions: params.assertions,
+    return this.saveAssertionsAll({
+      updates: [{ traceId: params.traceId, assertions: params.assertions }],
+      source: params.source
+    });
+  }
+  async saveAssertionsAll(params) {
+    if (params.updates.length === 0) {
+      return [];
+    }
+    const response = await this.httpClient.request(ASSERTIONS_PATH, {
+      updates: params.updates,
       source: params.source ?? "agent"
     });
     return response.assertions;
   }
   async archiveAssertions(params) {
     const response = await this.httpClient.request(
-      expectationsPath(params.traceId, "/archive"),
+      traceAssertionsPath(params.traceId, "/archive"),
       { assertionIds: params.assertionIds }
     );
     return response.archived;