@valerypopoff/rivet2-evaluations 2.4.0 → 2.6.0

package/dist/cjs/bundle.cjs

@@ -28,11 +28,12 @@ var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__ge
 var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
 
 // src/index.ts
-var index_exports = {};
-__export(index_exports, {
+var src_exports = {};
+__export(src_exports, {
   EvaluationGraphExecutionError: () => EvaluationGraphExecutionError,
   InMemoryEvaluationRunStore: () => InMemoryEvaluationRunStore,
   LEGACY_EVALUATOR_INPUT_IDS: () => LEGACY_EVALUATOR_INPUT_IDS,
+  applyEvaluationLibraryMutation: () => applyEvaluationLibraryMutation,
   areEvaluationDataTypesCompatible: () => areEvaluationDataTypesCompatible,
   assertEvaluationDatasetSnapshot: () => assertEvaluationDatasetSnapshot,
   assertEvaluationDatasetValuesMatchDeclaredTypes: () => assertEvaluationDatasetValuesMatchDeclaredTypes,
@@ -51,12 +52,16 @@ __export(index_exports, {
   deserializeEvaluationProjectData: () => deserializeEvaluationProjectData,
   deserializeEvaluationProjectDataFromString: () => deserializeEvaluationProjectDataFromString,
   deserializeEvaluationSuiteBundleJson: () => deserializeEvaluationSuiteBundleJson,
+  diffEvaluationLibraryMutation: () => diffEvaluationLibraryMutation,
   evaluateAssertion: () => evaluateAssertion,
   evaluateThresholdResults: () => evaluateThresholdResults,
+  evaluationLibraryValueEquals: () => evaluationLibraryValueEquals,
   finalizeEvaluationRecordingRetention: () => finalizeEvaluationRecordingRetention,
   finalizeEvaluationRun: () => finalizeEvaluationRun,
   fingerprint: () => fingerprint,
   fingerprintEvaluationDataset: () => fingerprintEvaluationDataset,
+  getEvaluationLibraryResource: () => getEvaluationLibraryResource,
+  getEvaluationSuiteBundle: () => getEvaluationSuiteBundle,
   getEvaluationSuiteMode: () => getEvaluationSuiteMode,
   getEvaluationTopLevelOutputId: () => getEvaluationTopLevelOutputId,
   hasAuthoritativeEvaluationCriteria: () => hasAuthoritativeEvaluationCriteria,
@@ -88,7 +93,7 @@ __export(index_exports, {
   validateEvaluationDataset: () => validateEvaluationDataset,
   validateEvaluationDatasetForTransfer: () => validateEvaluationDatasetForTransfer
 });
-module.exports = __toCommonJS(index_exports);
+module.exports = __toCommonJS(src_exports);
 
 // src/types.ts
 var EvaluationGraphExecutionError = class extends Error {
@@ -3360,11 +3365,111 @@ function deserializeEvaluationSuiteBundleJson(source, scope) {
     }
   };
 }
+
+// src/librarySync.ts
+function evaluationLibraryValueEquals(left, right) {
+  if (left === void 0 || right === void 0) return left === right;
+  return canonicalStringify(left) === canonicalStringify(right);
+}
+function getEvaluationSuiteBundle(library, id) {
+  const suite = library.data.suites.find((candidate) => candidate.id === id);
+  if (!suite) return void 0;
+  return {
+    suite,
+    baselines: library.data.baselines.filter((baseline) => baseline.suiteId === id)
+  };
+}
+function getEvaluationLibraryResource(library, kind, id) {
+  if (kind === "suite") return { kind, id, value: getEvaluationSuiteBundle(library, id) };
+  return { kind, id, value: library.datasets.find((dataset) => dataset.id === id) };
+}
+function resourceIds(library, kind) {
+  return new Set(kind === "suite" ? library.data.suites.map((suite) => suite.id) : library.datasets.map((dataset) => dataset.id));
+}
+function diffEvaluationLibraryMutation(previous, next, versions) {
+  const changes = [];
+  for (const kind of ["dataset", "suite"]) {
+    const ids = /* @__PURE__ */ new Set([...resourceIds(previous, kind), ...resourceIds(next, kind)]);
+    for (const id of [...ids].sort()) {
+      if (kind === "suite") {
+        const before = getEvaluationSuiteBundle(previous, id);
+        const after = getEvaluationSuiteBundle(next, id);
+        if (evaluationLibraryValueEquals(before, after)) continue;
+        const currentVersion = versions.suites[id];
+        if (before !== void 0 && currentVersion === void 0) {
+          throw new Error(`The suite "${id}" has no synchronization version.`);
+        }
+        const expectedVersion = before === void 0 ? null : currentVersion;
+        if (after === void 0) {
+          changes.push({ kind: "delete-suite", id, expectedVersion });
+        } else {
+          changes.push({
+            kind: "put-suite",
+            id,
+            expectedVersion,
+            suite: after.suite,
+            baselines: after.baselines
+          });
+        }
+      } else {
+        const before = previous.datasets.find((dataset) => dataset.id === id);
+        const after = next.datasets.find((dataset) => dataset.id === id);
+        if (evaluationLibraryValueEquals(before, after)) continue;
+        const currentVersion = versions.datasets[id];
+        if (before !== void 0 && currentVersion === void 0) {
+          throw new Error(`The dataset "${id}" has no synchronization version.`);
+        }
+        const expectedVersion = before === void 0 ? null : currentVersion;
+        if (after === void 0) {
+          changes.push({ kind: "delete-dataset", id, expectedVersion });
+        } else {
+          changes.push({ kind: "put-dataset", id, expectedVersion, dataset: after });
+        }
+      }
+    }
+  }
+  return changes.length === 0 ? void 0 : { changes };
+}
+function replaceById(items, value) {
+  const index = items.findIndex((item) => item.id === value.id);
+  if (index === -1) return [...items, value];
+  return items.map((item) => item.id === value.id ? value : item);
+}
+function applyEvaluationLibraryMutation(library, mutation) {
+  let suites = library.data.suites;
+  let baselines = library.data.baselines;
+  let datasets = library.datasets;
+  for (const change of mutation.changes) {
+    switch (change.kind) {
+      case "put-suite":
+        suites = replaceById(suites, change.suite);
+        baselines = [...baselines.filter((baseline) => baseline.suiteId !== change.id), ...change.baselines];
+        break;
+      case "delete-suite":
+        suites = suites.filter((suite) => suite.id !== change.id);
+        baselines = baselines.filter((baseline) => baseline.suiteId !== change.id);
+        break;
+      case "put-dataset":
+        datasets = replaceById(datasets, change.dataset);
+        break;
+      case "delete-dataset":
+        datasets = datasets.filter((dataset) => dataset.id !== change.id);
+        break;
+    }
+  }
+  return normalizeEvaluationLibrary({
+    version: 1,
+    data: { version: 1, suites, baselines },
+    datasets,
+    migratedLegacyProjectIds: library.migratedLegacyProjectIds
+  });
+}
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   EvaluationGraphExecutionError,
   InMemoryEvaluationRunStore,
   LEGACY_EVALUATOR_INPUT_IDS,
+  applyEvaluationLibraryMutation,
   areEvaluationDataTypesCompatible,
   assertEvaluationDatasetSnapshot,
   assertEvaluationDatasetValuesMatchDeclaredTypes,
@@ -3383,12 +3488,16 @@ function deserializeEvaluationSuiteBundleJson(source, scope) {
   deserializeEvaluationProjectData,
   deserializeEvaluationProjectDataFromString,
   deserializeEvaluationSuiteBundleJson,
+  diffEvaluationLibraryMutation,
   evaluateAssertion,
   evaluateThresholdResults,
+  evaluationLibraryValueEquals,
   finalizeEvaluationRecordingRetention,
   finalizeEvaluationRun,
   fingerprint,
   fingerprintEvaluationDataset,
+  getEvaluationLibraryResource,
+  getEvaluationSuiteBundle,
   getEvaluationSuiteMode,
   getEvaluationTopLevelOutputId,
   hasAuthoritativeEvaluationCriteria,