@m4trix/evals 0.17.0 → 0.18.0

package/dist/cli-simple.js

@@ -333,20 +333,70 @@ function getMetricById(id) {
 
 // src/evals/score.ts
 var registry2 = /* @__PURE__ */ new Map();
+function formatScoreData(def, data, options) {
+  return options?.isAggregated ? def.formatAggregate(data) : def.formatValue(data);
+}
+var ScoreAggregate = {
+  /** Average numeric fields. Use for scores like { value, delta }. */
+  averageFields(fields) {
+    return (values) => {
+      const count = values.length || 1;
+      const result = {};
+      for (const field of fields) {
+        result[field] = values.reduce((s, v) => s + (v[field] ?? 0), 0) / count;
+      }
+      return result;
+    };
+  },
+  /** Average `value` with sample std dev. Use for percent-style scores. */
+  averageWithVariance(values) {
+    if (values.length === 0) {
+      return { value: 0, stdDev: void 0, count: 0 };
+    }
+    const sum = values.reduce((s, v) => s + v.value, 0);
+    const sumSq = values.reduce((s, v) => s + v.value * v.value, 0);
+    const mean = sum / values.length;
+    let stdDev;
+    if (values.length >= 2) {
+      const variance = (sumSq - values.length * mean * mean) / (values.length - 1);
+      stdDev = variance > 0 ? Math.sqrt(variance) : 0;
+    }
+    return { ...values[0], value: mean, stdDev, count: values.length };
+  },
+  /** All runs must pass. Use for binary scores. */
+  all(values) {
+    const total = values.length;
+    const passedCount = values.filter((v) => v.passed).length;
+    return {
+      ...values[0],
+      passed: total > 0 && values.every((v) => v.passed),
+      passedCount,
+      totalCount: total
+    };
+  },
+  /** Take last value (no aggregation). Use when aggregation is not meaningful. */
+  last(values) {
+    return values[values.length - 1] ?? {};
+  }
+};
 var Score = {
+  aggregate: ScoreAggregate,
   of(config) {
     const def = {
       id: config.id,
       name: config.name,
       displayStrategy: config.displayStrategy,
-      aggregate: config.aggregate,
-      format: config.format,
+      formatValue: config.formatValue,
+      formatAggregate: config.formatAggregate,
+      aggregateValues: config.aggregateValues,
       make: (data, options) => {
         const passed = options?.definePassed !== void 0 ? options.definePassed(data) : void 0;
         return {
           id: config.id,
           data,
-          ...passed !== void 0 && { passed }
+          ...passed !== void 0 && { passed },
+          def
+          // Attach def so rendering/aggregation works without registry lookup
         };
       }
     };
@@ -359,29 +409,6 @@ function getScoreById(id) {
 }
 
 // src/evals/aggregators.ts
-function aggregateAverageWithVariance(values) {
-  if (values.length === 0) {
-    return { value: 0, count: 0 };
-  }
-  const sum = values.reduce((s, v) => s + v.value, 0);
-  const sumSq = values.reduce((s, v) => s + v.value * v.value, 0);
-  const mean = sum / values.length;
-  let stdDev;
-  if (values.length >= 2) {
-    const variance = (sumSq - values.length * mean * mean) / (values.length - 1);
-    stdDev = variance > 0 ? Math.sqrt(variance) : 0;
-  }
-  return { value: mean, stdDev, count: values.length };
-}
-function aggregateAll(values) {
-  const total = values.length;
-  const passedCount = values.filter((v) => v.passed).length;
-  return {
-    passed: total > 0 && values.every((v) => v.passed),
-    passedCount,
-    totalCount: total
-  };
-}
 function aggregateTokenCountSum(values) {
   const initial = {
     input: 0,
@@ -434,40 +461,45 @@ Score.of({
   id: "percent",
   name: "Score",
   displayStrategy: "bar",
-  format: (data, options) => {
-    if (options?.isAggregated) {
-      return data.stdDev != null ? `Avg: ${data.value.toFixed(2)} \xB1 ${data.stdDev.toFixed(2)}` : `Avg: ${data.value.toFixed(2)}`;
-    }
-    return data.value.toFixed(2);
-  },
-  aggregate: aggregateAverageWithVariance
+  formatValue: (data) => data.value.toFixed(2),
+  formatAggregate: (data) => data.stdDev != null ? `Avg: ${data.value.toFixed(2)} \xB1 ${data.stdDev.toFixed(2)}` : `Avg: ${data.value.toFixed(2)}`,
+  aggregateValues: Score.aggregate.averageWithVariance
+});
+Score.of({
+  id: "delta",
+  name: "Delta",
+  displayStrategy: "number",
+  formatValue: (data) => `${data.value.toFixed(2)} (${data.delta >= 0 ? "+" : ""}${data.delta.toFixed(2)} vs baseline)`,
+  formatAggregate: (data) => `Avg: ${data.value.toFixed(2)} (Delta: ${data.delta >= 0 ? "+" : ""}${data.delta.toFixed(2)})`,
+  aggregateValues: Score.aggregate.averageFields(["value", "delta"])
 });
 Score.of({
   id: "binary",
   name: "Result",
   displayStrategy: "passFail",
-  format: (data, options) => {
-    if (options?.isAggregated) {
-      const base = data.passed ? "All: PASSED" : "Some: FAILED";
-      if (data.passedCount != null && data.totalCount != null && data.totalCount > 1) {
-        return `${base} (${data.passedCount}/${data.totalCount})`;
-      }
-      return base;
+  formatValue: (data) => data.passed ? "PASSED" : "NOT PASSED",
+  formatAggregate: (data) => {
+    const base = data.passed ? "All: PASSED" : "Some: FAILED";
+    if (data.passedCount != null && data.totalCount != null && data.totalCount > 1) {
+      return `${base} (${data.passedCount}/${data.totalCount})`;
     }
-    return data.passed ? "PASSED" : "NOT PASSED";
+    return base;
   },
-  aggregate: aggregateAll
+  aggregateValues: Score.aggregate.all
 });
 
 // src/runner/score-utils.ts
+function getScoreDef(item) {
+  return item.def ?? getScoreById(item.id);
+}
 function aggregateScoreItems(items) {
   if (items.length === 0)
     return void 0;
-  const def = getScoreById(items[0].id);
-  if (!def?.aggregate)
+  const def = getScoreDef(items[0]);
+  if (!def?.aggregateValues)
     return items[items.length - 1];
-  const aggregated = def.aggregate(items.map((i) => i.data));
-  return { ...items[0], data: aggregated };
+  const aggregated = def.aggregateValues(items.map((i) => i.data));
+  return { ...items[0], data: aggregated, def };
 }
 function aggregateMetricItems(items) {
   if (items.length === 0)
@@ -480,7 +512,7 @@ function aggregateMetricItems(items) {
 }
 function toNumericScoreFromScores(scores) {
   for (const item of scores) {
-    const def = getScoreById(item.id);
+    const def = getScoreDef(item);
     if (def && def.displayStrategy === "bar" && typeof item.data === "object" && item.data !== null && "value" in item.data) {
       const value = item.data.value;
       if (typeof value === "number" && Number.isFinite(value)) {
@@ -868,7 +900,7 @@ var createPersistenceWorker = (queue) => Effect.forever(
       () => appendJsonLine(message.artifactPath, {
         runId: message.runId,
         ts: Date.now(),
-        ...message.payload
+        ...typeof message.payload === "object" && message.payload !== null && !Array.isArray(message.payload) ? message.payload : {}
       })
     );
   })
@@ -1454,7 +1486,7 @@ function aggregateEvaluatorScores(events, nameById) {
       if (agg)
         aggregatedScores.push(agg);
     }
-    const aggregatedMetrics = Array.from(metricIdToItems.entries()).map(([id, items]) => aggregateMetricItems(items)).filter((m) => m !== void 0);
+    const aggregatedMetrics = Array.from(metricIdToItems.entries()).map(([, items]) => aggregateMetricItems(items)).filter((m) => m !== void 0);
     const passed = events.every((ev) => {
       const es = ev.evaluatorScores.find((x) => x.evaluatorId === evaluatorId);
       return es?.passed ?? false;
@@ -1474,13 +1506,13 @@ function aggregateEvaluatorScores(events, nameById) {
   }
   return result;
 }
-function formatScorePart(item, scoreToColor2, options) {
-  const def = getScoreById(item.id);
+function formatScorePart(item, _scoreToColor, options) {
+  const def = item.def ?? getScoreById(item.id);
   if (!def) {
     const numeric = toNumericScore(item.data);
     return numeric !== void 0 ? `${numeric.toFixed(2)}` : "n/a";
   }
-  const formatted = def.format(item.data, options);
+  const formatted = formatScoreData(def, item.data, options);
   if (def.displayStrategy === "bar") {
     const numeric = typeof item.data === "object" && item.data !== null && "value" in item.data ? item.data.value : toNumericScore(item.data);
     if (typeof numeric === "number" && Number.isFinite(numeric)) {
@@ -1538,8 +1570,6 @@ function RunView({
     const done = new Promise((resolve5) => {
       const unsubscribe = runner.subscribeRunEvents((event) => {
         if (event.type === "TestCaseProgress") {
-          const numericScores = event.evaluatorScores.map((item) => toNumericScoreFromScores(item.scores)).filter((item) => item !== void 0);
-          numericScores.length > 0 ? numericScores.reduce((sum, v) => sum + v, 0) / numericScores.length : void 0;
           for (const item of event.evaluatorScores) {
             const numeric = toNumericScoreFromScores(item.scores);
             if (numeric !== void 0) {
@@ -1629,16 +1659,17 @@ function RunView({
       onComplete(new Error(`Run failed: ${finalEvent.errorMessage}`));
       return;
     }
+    const completed = finalEvent;
     setSummary({
-      passedTestCases: finalEvent.passedTestCases,
-      failedTestCases: finalEvent.failedTestCases,
-      totalTestCases: finalEvent.totalTestCases,
+      passedTestCases: completed.passedTestCases,
+      failedTestCases: completed.failedTestCases,
+      totalTestCases: completed.totalTestCases,
       overallScoreTotal,
       overallScoreSumSq,
       overallScoreCount,
       aggregates: new Map(aggregates),
       scoreItemsByEvaluatorScore: new Map(scoreItemsByEvaluatorScore),
-      artifactPath: finalEvent.artifactPath
+      artifactPath: completed.artifactPath
     });
     setPhase("completed");
     setTimeout(() => onComplete(), 200);
@@ -1844,11 +1875,9 @@ function RunView({
               const aggregated = aggregateScoreItems(items);
               if (!aggregated)
                 return null;
-              const def = getScoreById(aggregated.id);
+              const def = aggregated.def ?? getScoreById(aggregated.id);
               const label = def ? def.name ?? def.id : aggregated.id;
-              const formatted = def?.format(aggregated.data, {
-                isAggregated: true
-              }) ?? "n/a";
+              const formatted = def ? def.formatAggregate(aggregated.data) : "n/a";
               const numeric = toNumericScore(aggregated.data);
               return /* @__PURE__ */ jsxs(
                 Text,
@@ -2007,9 +2036,9 @@ function getEvaluatorSummaryLines(evaluatorId, evaluatorName, aggregate, scoreIt
     const agg = aggregateScoreItems(items);
     if (!agg)
       continue;
-    const def = getScoreById(agg.id);
+    const def = agg.def ?? getScoreById(agg.id);
     const label = def ? def.name ?? def.id : agg.id;
-    const formatted = def?.format(agg.data, { isAggregated: true }) ?? "n/a";
+    const formatted = def ? def.formatAggregate(agg.data) : "n/a";
     const numeric = toNumericScore(agg.data);
     const colored = numeric !== void 0 ? colorize(formatted, scoreToColor(numeric)) : formatted;
     scoreLines.push(`    ${label}: ${colored}`);
@@ -2027,7 +2056,7 @@ function createBar2(value, max = 100, width = 20) {
   const filled = Math.round(safe / max * width);
   return `${"\u2588".repeat(filled)}${"\u2591".repeat(width - filled)}`;
 }
-function aggregateEvaluatorScoresFromEvents(events, evaluatorNameById) {
+function aggregateEvaluatorScoresFromEvents(events, _evaluatorNameById) {
   if (events.length === 0)
     return [];
   const evaluatorIds = new Set(
@@ -2086,14 +2115,14 @@ function formatEvaluatorScoreLine(name, scores, passed, metrics, options) {
   }
   const scoreLines = [];
   for (const item of scores) {
-    const def = getScoreById(item.id);
+    const def = item.def ?? getScoreById(item.id);
     const scoreLabel = def ? def.name ?? def.id : item.id;
     let formatted;
     if (!def) {
       const numeric = toNumericScore(item.data);
       formatted = numeric !== void 0 ? colorize(numeric.toFixed(2), scoreToColor(numeric)) : "n/a";
     } else {
-      const raw = def.format(item.data, options);
+      const raw = formatScoreData(def, item.data, options);
       switch (def.displayStrategy) {
         case "bar": {
           const numeric = typeof item.data === "object" && item.data !== null && "value" in item.data ? item.data.value : toNumericScore(item.data);
@@ -2245,7 +2274,6 @@ async function runSimpleEvalCommandPlain(runner, datasetName, evaluatorPattern)
           (s, e) => s + e.durationMs,
           0
         );
-        existing.events.every((e) => e.passed);
         const lines = [];
         lines.push(
           `${colorize(`[${event.completedTestCases}/${event.totalTestCases}]`, ansi2.cyan)} ${event.testCaseName} ${colorize(`(${event.rerunIndex}/${event.rerunTotal})`, ansi2.cyan)} ${colorize(`(${durationMs}ms)`, ansi2.dim)}`
@@ -2323,18 +2351,19 @@ async function runSimpleEvalCommandPlain(runner, datasetName, evaluatorPattern)
   if (finalEvent.type === "RunFailed") {
     throw new Error(`Run failed: ${finalEvent.errorMessage}`);
   }
+  const completed = finalEvent;
   console.log("");
   console.log(colorize("=== Run Summary ===", `${ansi2.bold}${ansi2.cyan}`));
   console.log(
     `- passed: ${colorize(
-      `${finalEvent.passedTestCases}/${finalEvent.totalTestCases}`,
+      `${completed.passedTestCases}/${completed.totalTestCases}`,
       ansi2.green
     )}`
   );
   console.log(
     `- failed: ${colorize(
-      `${finalEvent.failedTestCases}/${finalEvent.totalTestCases}`,
-      finalEvent.failedTestCases > 0 ? ansi2.red : ansi2.dim
+      `${completed.failedTestCases}/${completed.totalTestCases}`,
+      completed.failedTestCases > 0 ? ansi2.red : ansi2.dim
     )}`
   );
   if (overallScoreCount > 0) {
@@ -2375,10 +2404,10 @@ async function runSimpleEvalCommandPlain(runner, datasetName, evaluatorPattern)
         );
         continue;
       }
-      const scoreLabel = summary.isAggregated && summary.aggregatedScoreItem ? getScoreById(summary.aggregatedScoreItem.id)?.format(
-        summary.aggregatedScoreItem.data,
-        { isAggregated: true }
-      ) ?? summary.averageScore.toFixed(2) : summary.stdDev !== void 0 && summary.isAggregated ? `${summary.averageScore.toFixed(2)} \xB1 ${summary.stdDev.toFixed(2)}` : summary.averageScore.toFixed(2);
+      const scoreLabel = summary.isAggregated && summary.aggregatedScoreItem ? (() => {
+        const def = summary.aggregatedScoreItem.def ?? getScoreById(summary.aggregatedScoreItem.id);
+        return def ? def.formatAggregate(summary.aggregatedScoreItem.data) : summary.averageScore.toFixed(2);
+      })() : summary.stdDev !== void 0 && summary.isAggregated ? `${summary.averageScore.toFixed(2)} \xB1 ${summary.stdDev.toFixed(2)}` : summary.averageScore.toFixed(2);
       console.log(
         `  ${status} ${summary.name.padEnd(24)} score=${colorize(
           scoreLabel,
@@ -2387,7 +2416,7 @@ async function runSimpleEvalCommandPlain(runner, datasetName, evaluatorPattern)
       );
     }
   }
-  console.log(`- artifact: ${colorize(finalEvent.artifactPath, ansi2.dim)}`);
+  console.log(`- artifact: ${colorize(completed.artifactPath, ansi2.dim)}`);
 }
 async function runSimpleEvalCommandInk(runner, datasetName, evaluatorPattern) {
   return new Promise((resolve5, reject) => {