bitfab 0.39.0 → 0.41.0

package/dist/node.cjs

@@ -97,7 +97,7 @@ var __version__, __packageName__;
 var init_version_generated = __esm({
   "src/version.generated.ts"() {
     "use strict";
-    __version__ = "0.39.0";
+    __version__ = "0.41.0";
     __packageName__ = "bitfab";
   }
 });
@@ -1748,6 +1748,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();
@@ -1761,7 +1765,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();
@@ -3213,6 +3220,7 @@ __export(node_exports, {
   BitfabOpenAITracingProcessor: () => BitfabOpenAITracingProcessor,
   BitfabVercelAiHandler: () => BitfabVercelAiHandler,
   DEFAULT_SERVICE_URL: () => DEFAULT_SERVICE_URL,
+  DatasetsClient: () => DatasetsClient,
   DbBranchReplayError: () => DbBranchReplayError,
   HttpClient: () => HttpClient,
   NO_MOCK_OVERRIDE: () => NO_MOCK_OVERRIDE,
@@ -3538,8 +3546,7 @@ var BitfabClaudeAgentHandler = class {
     const externalTrace = {
       id: traceId,
       started_at: this.traceStartedAt ?? nowIso(),
-      ended_at: endedAt ?? nowIso(),
-      workflow_name: this.traceFunctionKey
+      ended_at: endedAt ?? nowIso()
     };
     if (metadata) {
       externalTrace.metadata = metadata;
@@ -4299,6 +4306,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"];
@@ -4708,8 +4840,7 @@ var BitfabLangGraphCallbackHandler = class {
       externalTrace: {
         id: rootSpan.traceId,
         started_at: rootSpan.startedAt,
-        ended_at: rootSpan.endedAt ?? nowIso2(),
-        workflow_name: this.traceFunctionKey
+        ended_at: rootSpan.endedAt ?? nowIso2()
       },
       completed
     };
@@ -4727,8 +4858,7 @@ var BitfabLangGraphCallbackHandler = class {
       traceFunctionKey: this.traceFunctionKey,
       externalTrace: {
         id: rootSpan.traceId,
-        started_at: rootSpan.startedAt,
-        workflow_name: this.traceFunctionKey
+        started_at: rootSpan.startedAt
       },
       completed: false
     };
@@ -5886,6 +6016,8 @@ var noOpSpan = {
 var noOpTrace = {
   setSessionId() {
   },
+  setName() {
+  },
   setMetadata() {
   },
   addContext() {
@@ -5960,6 +6092,15 @@ function getCurrentTrace() {
       } catch {
       }
     },
+    setName(name) {
+      if (typeof name !== "string" || name.length === 0) {
+        return;
+      }
+      try {
+        getOrCreateTraceState().name = name;
+      } catch {
+      }
+    },
     setMetadata(metadata) {
       try {
         if (typeof metadata !== "object" || metadata === null) {
@@ -6034,6 +6175,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.
@@ -6938,6 +7080,7 @@ var Bitfab = class {
                 startedAt: traceState?.startedAt ?? startedAt,
                 endedAt,
                 sessionId: traceState?.sessionId,
+                name: traceState?.name,
                 metadata: traceState?.metadata,
                 contexts: traceState?.contexts ?? [],
                 testRunId: traceState?.testRunId,
@@ -7256,6 +7399,15 @@ var Bitfab = class {
           return Promise.resolve();
         }
         return this.httpClient.patchTrace(traceId, { setSessionId: sessionId });
+      },
+      setName: (name) => {
+        if (!this.isTracingEnabled()) {
+          return Promise.resolve();
+        }
+        if (typeof name !== "string" || name.length === 0) {
+          return Promise.resolve();
+        }
+        return this.httpClient.patchTrace(traceId, { setName: name });
       }
     };
   }
@@ -7316,9 +7468,11 @@ var Bitfab = class {
     const rawTrace = {
       id: params.traceId,
       started_at: params.startedAt,
-      ended_at: params.endedAt,
-      workflow_name: params.traceFunctionKey
+      ended_at: params.endedAt
     };
+    if (params.name) {
+      rawTrace.name = params.name;
+    }
     if (params.metadata && Object.keys(params.metadata).length > 0) {
       rawTrace.metadata = params.metadata;
     }
@@ -7511,6 +7665,7 @@ var Bitfab = class {
       contexts: [],
       ingestionType: "seeded",
       ...options.sessionId !== void 0 && { sessionId: options.sessionId },
+      ...options.name !== void 0 && { name: options.name },
       ...options.metadata !== void 0 && { metadata: options.metadata }
     });
     try {
@@ -7533,6 +7688,7 @@ var Bitfab = class {
         startedAt,
         endedAt: startedAt,
         sessionId: options.sessionId,
+        name: options.name,
         metadata: options.metadata,
         contexts: [],
         ingestionType: "seeded"
@@ -7851,6 +8007,7 @@ assertAsyncStorageRegistered();
   BitfabOpenAITracingProcessor,
   BitfabVercelAiHandler,
   DEFAULT_SERVICE_URL,
+  DatasetsClient,
   DbBranchReplayError,
   HttpClient,
   NO_MOCK_OVERRIDE,