@bitfab/sdk 0.65.1 → 0.65.4

package/dist/{chunk-4O5APMS4.js → chunk-ISU6FQKY.js}

@@ -18,7 +18,7 @@ import {
   serializeValue,
   toJsonSafe,
   toJsonSafeReport
-} from "./chunk-S2EYDTUH.js";
+} from "./chunk-SOP5AFMH.js";
 import {
   BitfabError,
   CONTENT_OFF_KEY,
@@ -38,7 +38,7 @@ import {
   readEnv,
   recordCallerTraceMetadata,
   retireTraceMetadata
-} from "./chunk-SXNYSDXU.js";
+} from "./chunk-I233ERUH.js";
 import {
   __privateAdd,
   __privateGet,
@@ -1312,6 +1312,12 @@ function finishedRerun(datasetId, run, joinedExisting) {
   }
   return { run, joinedExisting };
 }
+function normalizeDataset(dataset) {
+  return { ...dataset, golden: dataset.golden ?? false };
+}
+function normalizeDatasetResult(response) {
+  return { ...response, dataset: normalizeDataset(response.dataset) };
+}
 function sleep(ms) {
   return new Promise((resolve) => setTimeout(resolve, ms));
 }
@@ -1328,11 +1334,13 @@ var DatasetsClient = class {
    * leaves an existing one untouched.
    */
   async save(params) {
-    return this.httpClient.request("/api/sdk/datasets", {
+    const response = await this.httpClient.request("/api/sdk/datasets", {
       traceFunctionKey: params.traceFunctionKey,
       name: params.name,
-      ...params.description === void 0 ? {} : { description: params.description }
+      ...params.description === void 0 ? {} : { description: params.description },
+      ...typeof params.golden === "boolean" ? { golden: params.golden } : {}
     });
+    return normalizeDatasetResult(response);
   }
   /**
    * Rename a dataset or change its description in place. The dataset keeps its
@@ -1342,18 +1350,21 @@ var DatasetsClient = class {
    * organization.
    */
   async update(datasetId, params) {
-    if (params.name === void 0 && params.description === void 0) {
-      throw new BitfabError("Provide name and/or description to update.");
+    if (params.name === void 0 && params.description === void 0 && params.golden === void 0) {
+      throw new BitfabError(
+        "Provide name, description, and/or golden to update."
+      );
     }
     const response = await this.httpClient.request(
       datasetPath(datasetId),
       {
         ...params.name === void 0 ? {} : { name: params.name },
-        ...params.description === void 0 ? {} : { description: params.description }
+        ...params.description === void 0 ? {} : { description: params.description },
+        ...typeof params.golden === "boolean" ? { golden: params.golden } : {}
       },
       { method: "PATCH" }
     );
-    return response.dataset;
+    return normalizeDataset(response.dataset);
   }
   /**
    * List datasets, scoped to one trace function when `traceFunctionKey` is
@@ -1371,7 +1382,7 @@ var DatasetsClient = class {
         query.set("cursor", cursor);
       }
       const response = await this.httpClient.get(`/api/sdk/datasets?${query}`);
-      datasets.push(...response.datasets);
+      datasets.push(...response.datasets.map(normalizeDataset));
       cursor = response.nextCursor ?? void 0;
     } while (cursor !== void 0);
     return datasets;
@@ -1381,7 +1392,7 @@ var DatasetsClient = class {
     const response = await this.httpClient.get(
       datasetPath(datasetId)
     );
-    return response.dataset;
+    return normalizeDataset(response.dataset);
   }
   /** Fetches all trace-ID pages, the same membership a replay with `datasetId` selects. */
   async listTraces(datasetId) {
@@ -1402,17 +1413,13 @@ var DatasetsClient = class {
    * `skippedTraceIds` rather than failing the call.
    */
   async addTraces(datasetId, traceIds) {
-    return this.httpClient.request(
-      datasetPath(datasetId, "/traces"),
-      { traceIds }
-    );
+    const response = await this.httpClient.request(datasetPath(datasetId, "/traces"), { traceIds });
+    return normalizeDatasetResult(response);
   }
   /** Remove traces from the dataset. The traces themselves are never deleted. */
   async removeTraces(datasetId, traceIds) {
-    return this.httpClient.request(
-      datasetPath(datasetId, "/removeTraces"),
-      { traceIds }
-    );
+    const response = await this.httpClient.request(datasetPath(datasetId, "/removeTraces"), { traceIds });
+    return normalizeDatasetResult(response);
   }
   /**
    * Assign graders to the dataset (1 to 100 ids per call). Graders outside the
@@ -1420,17 +1427,13 @@ var DatasetsClient = class {
    * `skippedGraderIds` rather than failing the call.
    */
   async addGraders(datasetId, graderIds) {
-    return this.httpClient.request(
-      datasetPath(datasetId, "/graders"),
-      { graderIds }
-    );
+    const response = await this.httpClient.request(datasetPath(datasetId, "/graders"), { graderIds });
+    return normalizeDatasetResult(response);
   }
   /** Unassign graders from the dataset. */
   async removeGraders(datasetId, graderIds) {
-    return this.httpClient.request(
-      datasetPath(datasetId, "/removeGraders"),
-      { graderIds }
-    );
+    const response = await this.httpClient.request(datasetPath(datasetId, "/removeGraders"), { graderIds });
+    return normalizeDatasetResult(response);
   }
   async rerunGraders(datasetId, options = {}) {
     const started = await this.httpClient.request(
@@ -6419,7 +6422,7 @@ var Bitfab = class {
       await runWithSeedContext({ traceId }, async () => target(...args));
     } finally {
       try {
-        const { flushTraces: flushTraces2 } = await import("./http-NWJPWZF5.js");
+        const { flushTraces: flushTraces2 } = await import("./http-LL3JD2JE.js");
         await flushTraces2(3e4);
       } finally {
         unrecorded = activeTraceStates.delete(traceId);
@@ -6478,7 +6481,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-Z5ND6EVB.js");
+    const { replay: doReplay } = await import("./replay-636XZIS2.js");
     return doReplay(
       this.httpClient,
       this.serviceUrl,
@@ -6824,7 +6827,7 @@ async function seedFromRegistry(registry, pipeline, cases, options = {}) {
       sessionId: seedCase.sessionId
     })
   );
-  const { flushTraces: flushTraces2 } = await import("./http-NWJPWZF5.js");
+  const { flushTraces: flushTraces2 } = await import("./http-LL3JD2JE.js");
   await flushTraces2(3e4);
   return { pipeline, traceFunctionKey, traceIds };
 }
@@ -6871,4 +6874,4 @@ export {
   reseedFromRegistry,
   seedFromRegistry
 };
-//# sourceMappingURL=chunk-4O5APMS4.js.map
+//# sourceMappingURL=chunk-ISU6FQKY.js.map