bitfab 0.53.4 → 0.54.0

package/dist/{chunk-53RMN23L.js → chunk-JAXMD6AP.js}

@@ -19,7 +19,7 @@ import {
   serializeValue,
   toJsonSafe,
   toJsonSafeReport
-} from "./chunk-JKVR5436.js";
+} from "./chunk-O2OJIGL3.js";
 import {
   BitfabError,
   DEFAULT_SERVICE_URL,
@@ -35,7 +35,7 @@ import {
   recordCallerTraceMetadata,
   retireTraceMetadata,
   warnOnce
-} from "./chunk-RZC7MYVZ.js";
+} from "./chunk-SEHWHFUY.js";
 import {
   __privateAdd,
   __privateGet,
@@ -46,6 +46,36 @@ import {
   isCaptureSuppressed
 } from "./chunk-VCRFFCLY.js";
 
+// src/assertionCategories.ts
+var CATEGORIES_PATH = "/api/sdk/assertionCategories";
+var AssertionCategoriesClient = class {
+  constructor(httpClient) {
+    this.httpClient = httpClient;
+  }
+  /** Create a category, or update it by ID. Omitted descriptions are preserved. */
+  async save(params) {
+    const response = await this.httpClient.request(CATEGORIES_PATH, params);
+    return response.category;
+  }
+  /** Read one category in the API key's organization. */
+  async get(id) {
+    const response = await this.httpClient.get(
+      `${CATEGORIES_PATH}/${encodeURIComponent(id)}`
+    );
+    return response.category;
+  }
+  /** List the organization's categories ordered by title. */
+  async list() {
+    const response = await this.httpClient.get(CATEGORIES_PATH);
+    return response.categories;
+  }
+  /** Delete a category and clear its assignments, preserving assertions and verdicts. */
+  async delete(id) {
+    const response = await this.httpClient.request(`${CATEGORIES_PATH}/${encodeURIComponent(id)}`, {}, { method: "DELETE" });
+    return response.category;
+  }
+};
+
 // src/assertions.ts
 var ASSERTIONS_PATH = "/api/sdk/traces/assertions";
 function traceAssertionsPath(traceId, suffix = "") {
@@ -1290,14 +1320,24 @@ var DatasetsClient = class {
   }
   /**
    * List datasets, scoped to one trace function when `traceFunctionKey` is
-   * given and organization-wide otherwise.
+   * given and organization-wide otherwise. Fetches all pages automatically.
    */
   async list(params = {}) {
-    const query = params.traceFunctionKey === void 0 ? "" : `?traceFunctionKey=${encodeURIComponent(params.traceFunctionKey)}`;
-    const response = await this.httpClient.get(
-      `/api/sdk/datasets${query}`
-    );
-    return response.datasets;
+    const datasets = [];
+    let cursor;
+    do {
+      const query = new URLSearchParams({ limit: "100" });
+      if (params.traceFunctionKey !== void 0) {
+        query.set("traceFunctionKey", params.traceFunctionKey);
+      }
+      if (cursor !== void 0) {
+        query.set("cursor", cursor);
+      }
+      const response = await this.httpClient.get(`/api/sdk/datasets?${query}`);
+      datasets.push(...response.datasets);
+      cursor = response.nextCursor ?? void 0;
+    } while (cursor !== void 0);
+    return datasets;
   }
   /** Fetch one dataset by id. Rejects with a 404 `BitfabError` when it is not in this organization. */
   async get(datasetId) {
@@ -1306,11 +1346,18 @@ var DatasetsClient = class {
     );
     return response.dataset;
   }
-  /** The ids of every trace in the dataset, the same membership a replay with `datasetId` selects. */
+  /** Fetches all trace-ID pages, the same membership a replay with `datasetId` selects. */
   async listTraces(datasetId) {
-    return this.httpClient.get(
-      datasetPath(datasetId, "/traces")
-    );
+    const traceIds = [];
+    const query = new URLSearchParams({ limit: "100" });
+    for (; ; ) {
+      const page = await this.httpClient.get(`${datasetPath(datasetId, "/traces")}?${query}`);
+      traceIds.push(...page.traceIds);
+      if (page.nextCursor === void 0 || page.nextCursor === null) {
+        return { datasetId: page.datasetId, traceIds };
+      }
+      query.set("cursor", page.nextCursor);
+    }
   }
   /**
    * Add traces to the dataset (1 to 100 ids per call). Traces outside the
@@ -3357,6 +3404,7 @@ var Bitfab = class {
     this.httpClient.releaseHeldExternalSpans = (timeoutMs) => this.simulationPlan.release(timeoutMs);
     this.httpClient.stopSimulationPlan = () => this.simulationPlan.stop();
     this.datasets = new DatasetsClient(this.httpClient);
+    this.assertionCategories = new AssertionCategoriesClient(this.httpClient);
     this.traces = new TracesClient(this.httpClient);
     this.labels = new LabelsClient(this.httpClient);
     this.graders = new GradersClient(this.httpClient);
@@ -5059,7 +5107,7 @@ var Bitfab = class {
       await runWithSeedContext({ traceId }, async () => target(...args));
     } finally {
       try {
-        const { flushTraces: flushTraces2 } = await import("./http-7WGJNBGI.js");
+        const { flushTraces: flushTraces2 } = await import("./http-RXH2N5BB.js");
         await flushTraces2(3e4);
       } finally {
         unrecorded = activeTraceStates.delete(traceId);
@@ -5118,7 +5166,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-LRCSZ7IX.js");
+    const { replay: doReplay } = await import("./replay-XQITDIRJ.js");
     return doReplay(
       this.httpClient,
       this.serviceUrl,
@@ -5402,7 +5450,7 @@ async function seedFromRegistry(registry, pipeline, cases, options = {}) {
       sessionId: seedCase.sessionId
     })
   );
-  const { flushTraces: flushTraces2 } = await import("./http-7WGJNBGI.js");
+  const { flushTraces: flushTraces2 } = await import("./http-RXH2N5BB.js");
   await flushTraces2(3e4);
   return { pipeline, traceFunctionKey, traceIds };
 }
@@ -5418,6 +5466,7 @@ function resolveTraceFunctionKey(registration) {
 }
 
 export {
+  AssertionCategoriesClient,
   TracesClient,
   BitfabClaudeAgentHandler,
   DatasetsClient,
@@ -5439,4 +5488,4 @@ export {
   reseedFromRegistry,
   seedFromRegistry
 };
-//# sourceMappingURL=chunk-53RMN23L.js.map
+//# sourceMappingURL=chunk-JAXMD6AP.js.map