@bitfab/sdk 0.39.0 → 0.41.0

package/dist/{chunk-FW7PZ3EP.js → chunk-RWZYZRYI.js}

@@ -5,7 +5,7 @@ import {
   replayContextReady,
   runWithReplayContext,
   warnOnce
-} from "./chunk-2JQSYJJR.js";
+} from "./chunk-IFYX3KW6.js";
 
 // src/codeChange.ts
 var MAX_FILES = 60;
@@ -1147,4 +1147,4 @@ export {
   sleepForReplayPersistence,
   replay
 };
-//# sourceMappingURL=chunk-FW7PZ3EP.js.map
+//# sourceMappingURL=chunk-RWZYZRYI.js.map

package/dist/{chunk-SHJ3ABUG.js → chunk-U3AKIFW3.js}

@@ -14,14 +14,14 @@ import {
   serializeValue,
   toJsonSafe,
   toJsonSafeReport
-} from "./chunk-FW7PZ3EP.js";
+} from "./chunk-RWZYZRYI.js";
 import {
   BitfabError,
   DEFAULT_SERVICE_URL,
   HttpClient,
   getReplayContext,
   warnOnce
-} from "./chunk-2JQSYJJR.js";
+} from "./chunk-IFYX3KW6.js";
 import {
   __privateAdd,
   __privateGet,
@@ -321,8 +321,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;
@@ -949,6 +948,131 @@ async function runFunctionWithBaml(bamlSource, inputs, providers, envVars) {
   };
 }
 
+// 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
 var SUPPORTED_PROVIDERS = ["neon"];
 function validateDbSnapshotConfig(config) {
@@ -1350,8 +1474,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
     };
@@ -1369,8 +1492,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
     };
@@ -2512,6 +2634,8 @@ var noOpSpan = {
 var noOpTrace = {
   setSessionId() {
   },
+  setName() {
+  },
   setMetadata() {
   },
   addContext() {
@@ -2586,6 +2710,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) {
@@ -2660,6 +2793,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.
@@ -3564,6 +3698,7 @@ var Bitfab = class {
                 startedAt: traceState?.startedAt ?? startedAt,
                 endedAt,
                 sessionId: traceState?.sessionId,
+                name: traceState?.name,
                 metadata: traceState?.metadata,
                 contexts: traceState?.contexts ?? [],
                 testRunId: traceState?.testRunId,
@@ -3882,6 +4017,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 });
       }
     };
   }
@@ -3942,9 +4086,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;
     }
@@ -4137,6 +4283,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 {
@@ -4159,6 +4306,7 @@ var Bitfab = class {
         startedAt,
         endedAt: startedAt,
         sessionId: options.sessionId,
+        name: options.name,
         metadata: options.metadata,
         contexts: [],
         ingestionType: "seeded"
@@ -4182,7 +4330,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-MZYX4IFV.js");
+    const { replay: doReplay } = await import("./replay-HMMWDSLB.js");
     return doReplay(
       this.httpClient,
       this.serviceUrl,
@@ -4435,7 +4583,7 @@ async function seedFromRegistry(registry, pipeline, cases) {
       sessionId: seedCase.sessionId
     })
   );
-  const { flushTraces: flushTraces2 } = await import("./http-PRCLI43J.js");
+  const { flushTraces: flushTraces2 } = await import("./http-UCJCOKW7.js");
   await flushTraces2(3e4);
   return { pipeline, traceFunctionKey, traceIds };
 }
@@ -4452,6 +4600,7 @@ function resolveTraceFunctionKey(registration) {
 
 export {
   BitfabClaudeAgentHandler,
+  DatasetsClient,
   SUPPORTED_PROVIDERS,
   BitfabLangGraphCallbackHandler,
   BitfabLangGraphIntegration,
@@ -4467,4 +4616,4 @@ export {
   defineReplayRegistry,
   seedFromRegistry
 };
-//# sourceMappingURL=chunk-SHJ3ABUG.js.map
+//# sourceMappingURL=chunk-U3AKIFW3.js.map