bitfab 0.48.0 → 0.51.0

package/dist/{http-42E3SI2N.js → http-CS565CIX.js}

@@ -6,7 +6,7 @@ import {
   flushTraces,
   parseRetryAfterMs,
   serializePayloadBody
-} from "./chunk-3VHHLXEJ.js";
+} from "./chunk-4Z5FCOMO.js";
 import "./chunk-H6LZRFMN.js";
 export {
   BitfabError,
@@ -17,4 +17,4 @@ export {
   parseRetryAfterMs,
   serializePayloadBody
 };
-//# sourceMappingURL=http-42E3SI2N.js.map
+//# sourceMappingURL=http-CS565CIX.js.map

package/dist/{http-YN67UB3K.js → http-RQXNGCUQ.js}

@@ -6,7 +6,7 @@ import {
   flushTraces,
   parseRetryAfterMs,
   serializePayloadBody
-} from "./chunk-CPMT6Y43.js";
+} from "./chunk-WBV4YW7X.js";
 export {
   BitfabError,
   HttpClient,
@@ -16,4 +16,4 @@ export {
   parseRetryAfterMs,
   serializePayloadBody
 };
-//# sourceMappingURL=http-YN67UB3K.js.map
+//# sourceMappingURL=http-RQXNGCUQ.js.map

package/dist/index.cjs

@@ -51,7 +51,7 @@ var __version__, __packageName__;
 var init_version_generated = __esm({
   "src/version.generated.ts"() {
     "use strict";
-    __version__ = "0.48.0";
+    __version__ = "0.51.0";
     __packageName__ = "bitfab";
   }
 });
@@ -1856,7 +1856,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;
@@ -1880,8 +1880,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;
@@ -2483,6 +2487,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;
 }
@@ -2915,9 +2936,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) {
@@ -2959,7 +2981,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,
@@ -3281,6 +3303,7 @@ __export(index_exports, {
   DEFAULT_SERVICE_URL: () => DEFAULT_SERVICE_URL,
   DatasetsClient: () => DatasetsClient,
   DbBranchReplayError: () => DbBranchReplayError,
+  GradersClient: () => GradersClient,
   HttpClient: () => HttpClient,
   LabelsClient: () => LabelsClient,
   MixedTracingError: () => MixedTracingError,
@@ -4576,10 +4599,45 @@ function buildSnapshotRef(config, sdkWallClockBeforeFn) {
 
 // src/client.ts
 init_errors();
+
+// src/graders.ts
+init_errors();
+var GRADER_LABELS_PATH = "/api/sdk/graderLabels";
+var GradersClient = class {
+  constructor(httpClient) {
+    this.httpClient = httpClient;
+  }
+  async getLabels(params) {
+    const hasTraceIds = params.traceIds !== void 0 && params.traceIds.length > 0;
+    const hasGraderId = params.graderId !== void 0;
+    if (!hasTraceIds && !hasGraderId) {
+      throw new BitfabError(
+        "Pass traceIds, graderId, or both: there is nothing to read otherwise."
+      );
+    }
+    const query = new URLSearchParams();
+    if (params.traceIds !== void 0 && params.traceIds.length > 0) {
+      query.set("traceIds", params.traceIds.join(","));
+    }
+    if (params.graderId !== void 0) {
+      query.set("graderId", params.graderId);
+    }
+    if (params.limit !== void 0) {
+      query.set("limit", String(params.limit));
+    }
+    const response = await this.httpClient.get(
+      `${GRADER_LABELS_PATH}?${query.toString()}`
+    );
+    return response.labels;
+  }
+};
+
+// src/client.ts
 init_http();
 
 // src/labels.ts
 var LABELS_PATH = "/api/sdk/traces/labels";
+var HUMAN_LABELS_PATH = "/api/sdk/traces/labels/human";
 var LabelsClient = class {
   constructor(httpClient) {
     this.httpClient = httpClient;
@@ -4598,6 +4656,28 @@ var LabelsClient = class {
     );
     return response.labels;
   }
+  async saveHuman(update) {
+    const [outcome] = await this.saveHumanAll([update]);
+    return outcome;
+  }
+  async saveHumanAll(updates) {
+    const response = await this.httpClient.request(HUMAN_LABELS_PATH, { labels: updates });
+    return response.labels;
+  }
+  async get(traceId) {
+    const [labels] = await this.getAll([traceId]);
+    return labels ?? null;
+  }
+  async getAll(traceIds) {
+    if (traceIds.length === 0) {
+      return [];
+    }
+    const query = new URLSearchParams({ traceIds: traceIds.join(",") });
+    const response = await this.httpClient.get(
+      `${LABELS_PATH}?${query.toString()}`
+    );
+    return response.labels;
+  }
 };
 
 // src/langgraph.ts
@@ -6375,6 +6455,7 @@ var Bitfab = class {
     this.datasets = new DatasetsClient(this.httpClient);
     this.traces = new TracesClient(this.httpClient);
     this.labels = new LabelsClient(this.httpClient);
+    this.graders = new GradersClient(this.httpClient);
   }
   /**
    * Decorate a class method as an automatically expanded trace root.
@@ -8275,6 +8356,7 @@ function resolveTraceFunctionKey(registration) {
   DEFAULT_SERVICE_URL,
   DatasetsClient,
   DbBranchReplayError,
+  GradersClient,
   HttpClient,
   LabelsClient,
   MixedTracingError,