bitfab 0.39.0 → 0.40.0

package/dist/{http-ALSBGWPS.js → http-G5CODRDH.js}

@@ -6,7 +6,7 @@ import {
   flushTraces,
   parseRetryAfterMs,
   serializePayloadBody
-} from "./chunk-3JIPWENI.js";
+} from "./chunk-YZGGAUG6.js";
 export {
   BitfabError,
   HttpClient,
@@ -16,4 +16,4 @@ export {
   parseRetryAfterMs,
   serializePayloadBody
 };
-//# sourceMappingURL=http-ALSBGWPS.js.map
+//# sourceMappingURL=http-G5CODRDH.js.map

package/dist/{http-RMF7T3LG.js → http-MRZNL7RE.js}

@@ -6,7 +6,7 @@ import {
   flushTraces,
   parseRetryAfterMs,
   serializePayloadBody
-} from "./chunk-OXE6M7CZ.js";
+} from "./chunk-DYURJJSV.js";
 import "./chunk-H6LZRFMN.js";
 export {
   BitfabError,
@@ -17,4 +17,4 @@ export {
   parseRetryAfterMs,
   serializePayloadBody
 };
-//# sourceMappingURL=http-RMF7T3LG.js.map
+//# sourceMappingURL=http-MRZNL7RE.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.39.0";
+    __version__ = "0.40.0";
     __packageName__ = "bitfab";
   }
 });
@@ -1741,6 +1741,10 @@ var init_http = __esm({
         const response = await this.get(endpoint);
         return response.span;
       }
+      /**
+       * GET a JSON endpoint on the service with the client's API key. Throws a
+       * `BitfabError` carrying the status text for any non-2xx response.
+       */
       async get(endpoint) {
         const url = `${this.serviceUrl}${endpoint}`;
         const controller = new AbortController();
@@ -1754,7 +1758,10 @@ var init_http = __esm({
           if (!response.ok) {
             const errorText = await response.text();
             throw new BitfabError(
-              `HTTP ${response.status}: ${errorText.slice(0, 500)}`
+              `HTTP ${response.status}: ${errorText.slice(0, 500)}`,
+              void 0,
+              response.status,
+              parseRetryAfterMs(readHeader(response, "retry-after"))
             );
           }
           return await response.json();
@@ -3206,6 +3213,7 @@ __export(index_exports, {
   BitfabOpenAITracingProcessor: () => BitfabOpenAITracingProcessor,
   BitfabVercelAiHandler: () => BitfabVercelAiHandler,
   DEFAULT_SERVICE_URL: () => DEFAULT_SERVICE_URL,
+  DatasetsClient: () => DatasetsClient,
   DbBranchReplayError: () => DbBranchReplayError,
   HttpClient: () => HttpClient,
   NO_MOCK_OVERRIDE: () => NO_MOCK_OVERRIDE,
@@ -4285,6 +4293,131 @@ async function runFunctionWithBaml(bamlSource, inputs, providers, envVars) {
 // src/client.ts
 init_constants();
 
+// src/datasets.ts
+var DEFAULT_RERUN_TIMEOUT_MS = 9e4;
+var DEFAULT_RERUN_POLL_INTERVAL_MS = 1e3;
+var TERMINAL_RERUN_STATUSES = /* @__PURE__ */ new Set([
+  "completed",
+  "errored"
+]);
+function sleep(ms) {
+  return new Promise((resolve) => setTimeout(resolve, ms));
+}
+function datasetPath(datasetId, suffix = "") {
+  return `/api/sdk/datasets/${encodeURIComponent(datasetId)}${suffix}`;
+}
+var DatasetsClient = class {
+  constructor(httpClient) {
+    this.httpClient = httpClient;
+  }
+  /**
+   * Create a dataset, or update the one already named this way under the same
+   * trace function. `created` reports which happened. An omitted description
+   * leaves an existing one untouched.
+   */
+  async save(params) {
+    return this.httpClient.request("/api/sdk/datasets", {
+      traceFunctionKey: params.traceFunctionKey,
+      name: params.name,
+      ...params.description === void 0 ? {} : { description: params.description }
+    });
+  }
+  /**
+   * List datasets, scoped to one trace function when `traceFunctionKey` is
+   * given and organization-wide otherwise.
+   */
+  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;
+  }
+  /** Fetch one dataset by id. Rejects with a 404 `BitfabError` when it is not in this organization. */
+  async get(datasetId) {
+    const response = await this.httpClient.get(
+      datasetPath(datasetId)
+    );
+    return response.dataset;
+  }
+  /** The ids of every trace in the dataset, the same membership a replay with `datasetId` selects. */
+  async listTraces(datasetId) {
+    return this.httpClient.get(
+      datasetPath(datasetId, "/traces")
+    );
+  }
+  /**
+   * Add traces to the dataset (1 to 100 ids per call). Traces outside the
+   * organization or under another trace function are reported in
+   * `skippedTraceIds` rather than failing the call.
+   */
+  async addTraces(datasetId, traceIds) {
+    return this.httpClient.request(
+      datasetPath(datasetId, "/traces"),
+      { traceIds }
+    );
+  }
+  /** Remove traces from the dataset. The traces themselves are never deleted. */
+  async removeTraces(datasetId, traceIds) {
+    return this.httpClient.request(
+      datasetPath(datasetId, "/removeTraces"),
+      { traceIds }
+    );
+  }
+  /**
+   * Assign graders to the dataset (1 to 100 ids per call). Graders outside the
+   * organization or under another trace function are reported in
+   * `skippedGraderIds` rather than failing the call.
+   */
+  async addGraders(datasetId, graderIds) {
+    return this.httpClient.request(
+      datasetPath(datasetId, "/graders"),
+      { graderIds }
+    );
+  }
+  /** Unassign graders from the dataset. */
+  async removeGraders(datasetId, graderIds) {
+    return this.httpClient.request(
+      datasetPath(datasetId, "/removeGraders"),
+      { graderIds }
+    );
+  }
+  /**
+   * Re-run graders over every trace in the dataset. Defaults to every assigned
+   * grader; an unassigned id is rejected. Waits for the run to finish (up to
+   * `timeoutMs`, default 90s) unless `wait` is `false`, and returns the last
+   * run state seen either way. A request matching an in-flight run joins it.
+   */
+  async rerunGraders(datasetId, options = {}) {
+    const started = await this.httpClient.request(
+      datasetPath(datasetId, "/rerunGraders"),
+      options.graderIds === void 0 ? {} : { graderIds: options.graderIds }
+    );
+    if (options.wait === false) {
+      return started;
+    }
+    const deadline = Date.now() + (options.timeoutMs ?? DEFAULT_RERUN_TIMEOUT_MS);
+    const interval = options.pollIntervalMs ?? DEFAULT_RERUN_POLL_INTERVAL_MS;
+    let run = started.run;
+    while (!TERMINAL_RERUN_STATUSES.has(run.status) && Date.now() < deadline) {
+      await sleep(interval);
+      run = await this.getGraderRerun(datasetId, run.id) ?? run;
+    }
+    return { run, joinedExisting: started.joinedExisting };
+  }
+  /**
+   * The dataset's active grader re-run, or the run named by `runId`. Returns
+   * `null` when nothing is active or the run does not belong to this dataset.
+   */
+  async getGraderRerun(datasetId, runId) {
+    const query = runId === void 0 ? "" : `?runId=${encodeURIComponent(runId)}`;
+    const response = await this.httpClient.get(
+      datasetPath(datasetId, `/rerunGraders${query}`)
+    );
+    return response.run;
+  }
+};
+
 // src/dbSnapshot.ts
 init_errors();
 var SUPPORTED_PROVIDERS = ["neon"];
@@ -6020,6 +6153,7 @@ var Bitfab = class {
       serviceUrl: this.serviceUrl,
       timeout: this.timeout
     });
+    this.datasets = new DatasetsClient(this.httpClient);
   }
   /**
    * Decorate a class method as an automatically expanded trace root.
@@ -7833,6 +7967,7 @@ function resolveTraceFunctionKey(registration) {
   BitfabOpenAITracingProcessor,
   BitfabVercelAiHandler,
   DEFAULT_SERVICE_URL,
+  DatasetsClient,
   DbBranchReplayError,
   HttpClient,
   NO_MOCK_OVERRIDE,