@m4trix/evals 0.21.0 → 0.21.1

package/dist/cli-simple.cjs

@@ -8,12 +8,16 @@ var path = require('path');
 var jitiModule = require('jiti');
 var promises = require('fs/promises');
 var url = require('url');
-var jsonDiff = require('json-diff');
+var diff = require('diff');
+var stringify = require('fast-json-stable-stringify');
+var os = require('os');
 var React2 = require('react');
 var ink = require('ink');
 var jsxRuntime = require('react/jsx-runtime');
 
 var _documentCurrentScript = typeof document !== 'undefined' ? document.currentScript : null;
+function _interopDefault (e) { return e && e.__esModule ? e : { default: e }; }
+
 function _interopNamespace(e) {
   if (e && e.__esModule) return e;
   var n = Object.create(null);
@@ -33,6 +37,7 @@ function _interopNamespace(e) {
 }
 
 var jitiModule__namespace = /*#__PURE__*/_interopNamespace(jitiModule);
+var stringify__default = /*#__PURE__*/_interopDefault(stringify);
 var React2__namespace = /*#__PURE__*/_interopNamespace(React2);
 
 // src/runner/config.ts
@@ -284,10 +289,102 @@ async function collectTestCasesFromFiles(config) {
   );
   return found.flat();
 }
+function preprocessForDiff(value, options) {
+  if (options?.sort && Array.isArray(value)) {
+    return [...value].sort((a, b) => {
+      const aStr = stringify__default.default(preprocessForDiff(a, options));
+      const bStr = stringify__default.default(preprocessForDiff(b, options));
+      return aStr.localeCompare(bStr);
+    }).map((item) => preprocessForDiff(item, options));
+  }
+  if (value !== null && typeof value === "object" && !Array.isArray(value) && options?.excludeKeys) {
+    const keys = Array.isArray(options.excludeKeys) ? options.excludeKeys : options.excludeKeys.split(",").map((k) => k.trim());
+    const filtered = {};
+    for (const [k, v] of Object.entries(value)) {
+      if (!keys.includes(k)) {
+        filtered[k] = preprocessForDiff(v, options);
+      }
+    }
+    return filtered;
+  }
+  if (value !== null && typeof value === "object" && !Array.isArray(value)) {
+    const result = {};
+    for (const [k, v] of Object.entries(value)) {
+      result[k] = preprocessForDiff(v, options);
+    }
+    return result;
+  }
+  if (typeof value === "number" && options?.precision !== void 0) {
+    return Number(value.toFixed(options.precision));
+  }
+  return value;
+}
+function toPrettyJson(value) {
+  const str = stringify__default.default(value);
+  try {
+    const parsed = JSON.parse(str);
+    return JSON.stringify(parsed, null, 2);
+  } catch {
+    return str;
+  }
+}
+function formatDiffParts(parts) {
+  const lines = [];
+  for (const part of parts) {
+    const prefix = part.added ? "+ " : part.removed ? "- " : "";
+    const partLines = part.value.split("\n");
+    for (let i = 0; i < partLines.length; i++) {
+      const line = partLines[i];
+      if (i === partLines.length - 1 && line === "")
+        continue;
+      lines.push(prefix + line);
+    }
+  }
+  return lines.join("\n");
+}
 function createDiffString(expected, actual, diffOptions) {
-  const opts = { ...diffOptions, color: false };
-  const result = jsonDiff.diffString(expected, actual, opts);
-  return typeof result === "string" ? result : "";
+  const expectedProcessed = preprocessForDiff(expected, diffOptions);
+  const actualProcessed = preprocessForDiff(actual, diffOptions);
+  if (diffOptions?.keysOnly) {
+    const expectedKeys = JSON.stringify(
+      extractKeys(expectedProcessed),
+      null,
+      2
+    );
+    const actualKeys = JSON.stringify(
+      extractKeys(actualProcessed),
+      null,
+      2
+    );
+    const parts2 = diff.diffLines(expectedKeys, actualKeys);
+    return formatDiffParts(parts2);
+  }
+  const expectedStr = toPrettyJson(expectedProcessed);
+  const actualStr = toPrettyJson(actualProcessed);
+  if (expectedStr === actualStr) {
+    return "";
+  }
+  const parts = diff.diffLines(expectedStr, actualStr);
+  if (diffOptions?.outputNewOnly) {
+    const filtered = parts.filter(
+      (p) => p.added === true
+    );
+    return formatDiffParts(filtered);
+  }
+  return formatDiffParts(parts);
+}
+function extractKeys(value) {
+  if (value === null || typeof value !== "object") {
+    return "\xB7";
+  }
+  if (Array.isArray(value)) {
+    return value.map(extractKeys);
+  }
+  const result = {};
+  for (const [k, v] of Object.entries(value)) {
+    result[k] = extractKeys(v);
+  }
+  return result;
 }
 function formatLogMessage(msg) {
   if (typeof msg === "string")
@@ -1327,8 +1424,9 @@ var EffectRunner = class {
     );
   }
 };
-
-// src/cli-simple/args.ts
+function getDefaultConcurrency() {
+  return Math.max(1, os.cpus().length);
+}
 function parseSimpleCliArgs(argv) {
   const args = {
     help: false,
@@ -1355,6 +1453,14 @@ function parseSimpleCliArgs(argv) {
       index += 1;
       continue;
     }
+    if ((token === "--concurrency" || token === "-c") && argv[index + 1]) {
+      const n = parseInt(argv[index + 1], 10);
+      if (!Number.isNaN(n) && n >= 1) {
+        args.concurrency = n;
+      }
+      index += 1;
+      continue;
+    }
     args.unknownArgs.push(token);
   }
   return args;
@@ -1362,9 +1468,12 @@ function parseSimpleCliArgs(argv) {
 function getSimpleCliUsage() {
   return [
     "Usage:",
-    "  eval-agents-simple run --dataset <datasetName> --evaluator <name-or-pattern>",
+    "  eval-agents-simple run --dataset <datasetName> --evaluator <name-or-pattern> [--concurrency N]",
     "  eval-agents-simple generate --dataset <datasetName>",
     "",
+    "Options:",
+    "  --concurrency, -c N   Max concurrent test cases (default: CPU count). Use 1 for sequential.",
+    "",
     "Pattern examples for --evaluator:",
     "  score-evaluator       exact name (case-insensitive)",
     '  "*score*"             wildcard pattern',
@@ -1653,6 +1762,7 @@ function RunView({
   runner,
   datasetName,
   evaluatorPattern,
+  concurrency,
   onComplete
 }) {
   const [phase, setPhase] = React2.useState(
@@ -1800,7 +1910,8 @@ function RunView({
     });
     const snapshot = await runner.runDatasetWith({
       datasetId: dataset.id,
-      evaluatorIds: evaluators.map((item) => item.id)
+      evaluatorIds: evaluators.map((item) => item.id),
+      concurrency
     });
     setRunInfo({
       runId: snapshot.runId,
@@ -1828,7 +1939,7 @@ function RunView({
     });
     setPhase("completed");
     setTimeout(() => onComplete(), 200);
-  }, [runner, datasetName, evaluatorPattern, onComplete]);
+  }, [runner, datasetName, evaluatorPattern, concurrency, onComplete]);
   React2.useEffect(() => {
     void runEval();
   }, [runEval]);
@@ -1871,22 +1982,30 @@ function RunView({
           label: `Evaluations ${completedEvaluations}/${runInfo?.totalTestCases ?? 0} completed \u2022 ${startedEvaluations}/${runInfo?.totalTestCases ?? 0} started`
         }
       ),
-      runningEvaluations.length > 0 && /* @__PURE__ */ jsxRuntime.jsx(ink.Box, { flexDirection: "column", marginTop: 1, children: runningEvaluations.map((item) => /* @__PURE__ */ jsxRuntime.jsxs(ink.Text, { color: "yellow", children: [
-        "[running ",
-        item.startedTestCases,
-        "/",
-        item.totalTestCases,
-        "] ",
-        item.name,
-        " ",
-        /* @__PURE__ */ jsxRuntime.jsxs(ink.Text, { color: "gray", children: [
-          "(",
-          item.rerunIndex,
-          "/",
-          item.rerunTotal,
-          ")"
-        ] })
-      ] }, `${item.testCaseId}:${item.rerunIndex}`)) })
+      runningEvaluations.length > 0 && /* @__PURE__ */ jsxRuntime.jsx(ink.Box, { flexDirection: "column", marginTop: 1, children: runningEvaluations.map((item) => /* @__PURE__ */ jsxRuntime.jsxs(
+        ink.Text,
+        {
+          color: "yellow",
+          children: [
+            "[running ",
+            item.startedTestCases,
+            "/",
+            item.totalTestCases,
+            "]",
+            " ",
+            item.name,
+            " ",
+            /* @__PURE__ */ jsxRuntime.jsxs(ink.Text, { color: "gray", children: [
+              "(",
+              item.rerunIndex,
+              "/",
+              item.rerunTotal,
+              ")"
+            ] })
+          ]
+        },
+        `${item.testCaseId}:${item.rerunIndex}`
+      )) })
     ] }),
     testCases.length > 0 && /* @__PURE__ */ jsxRuntime.jsx(ink.Box, { flexDirection: "column", marginBottom: 1, children: testCases.map((tc) => /* @__PURE__ */ jsxRuntime.jsxs(ink.Box, { flexDirection: "column", marginBottom: 0, children: [
       /* @__PURE__ */ jsxRuntime.jsxs(ink.Text, { children: [
@@ -1968,7 +2087,7 @@ function RunView({
                 },
                 `${item.evaluatorId}-${s.id}-${idx}`
               );
-            }) : /* @__PURE__ */ jsxRuntime.jsx(ink.Text, { color: "gray", children: "      n/a" }),
+            }) : /* @__PURE__ */ jsxRuntime.jsx(ink.Text, { color: "gray", children: " n/a" }),
             !item.passed && item.logs && item.logs.length > 0 && /* @__PURE__ */ jsxRuntime.jsx(ink.Box, { marginLeft: 2, flexDirection: "column", children: item.logs.map(
               (log, logIdx) => log.type === "diff" ? /* @__PURE__ */ jsxRuntime.jsx(ink.Box, { flexDirection: "column", children: getDiffLines(log).map(
                 ({ type, line }, lineIdx) => /* @__PURE__ */ jsxRuntime.jsx(
@@ -2026,9 +2145,9 @@ function RunView({
         /* @__PURE__ */ jsxRuntime.jsx(ink.Text, { color: "magenta", children: "evaluator averages" }),
         Array.from(evaluatorNameById.entries()).map(([id, name]) => {
           const agg = summary.aggregates.get(id);
-          const scoreKeys = [...summary.scoreItemsByEvaluatorScore?.keys() ?? []].filter(
-            (k) => k.startsWith(`${id}:`)
-          );
+          const scoreKeys = [
+            ...summary.scoreItemsByEvaluatorScore?.keys() ?? []
+          ].filter((k) => k.startsWith(`${id}:`));
           if (scoreKeys.length === 0) {
             return /* @__PURE__ */ jsxRuntime.jsxs(ink.Text, { color: "gray", children: [
               "- ",
@@ -2336,7 +2455,7 @@ function formatEvaluatorScoreLine(name, scores, passed, metrics, options) {
   }
   return lines;
 }
-async function runSimpleEvalCommandPlain(runner, datasetName, evaluatorPattern) {
+async function runSimpleEvalCommandPlain(runner, datasetName, evaluatorPattern, concurrency) {
   const dataset = await runner.resolveDatasetByName(datasetName);
   if (!dataset) {
     const known = await runner.collectDatasets();
@@ -2526,7 +2645,8 @@ async function runSimpleEvalCommandPlain(runner, datasetName, evaluatorPattern)
   });
   const snapshot = await runner.runDatasetWith({
     datasetId: dataset.id,
-    evaluatorIds: evaluators.map((item) => item.id)
+    evaluatorIds: evaluators.map((item) => item.id),
+    concurrency
   });
   totalCount = snapshot.totalTestCases;
   console.log(colorize("=== Eval Run Started ===", `${ansi2.bold}${ansi2.cyan}`));
@@ -2615,13 +2735,14 @@ async function runSimpleEvalCommandPlain(runner, datasetName, evaluatorPattern)
   }
   console.log(`- artifact: ${colorize(completed.artifactPath, ansi2.dim)}`);
 }
-async function runSimpleEvalCommandInk(runner, datasetName, evaluatorPattern) {
+async function runSimpleEvalCommandInk(runner, datasetName, evaluatorPattern, concurrency) {
   return new Promise((resolve5, reject) => {
     const app = ink.render(
       React2__namespace.createElement(RunView, {
         runner,
         datasetName,
         evaluatorPattern,
+        concurrency,
         onComplete: (err) => {
           app.unmount();
           if (err) {
@@ -2668,10 +2789,12 @@ async function main() {
   const runner = createRunner();
   try {
     if (args.command === "run") {
+      const concurrency = args.concurrency ?? getDefaultConcurrency();
       await (useInk ? runSimpleEvalCommandInk : runSimpleEvalCommandPlain)(
         runner,
         args.datasetName,
-        args.evaluatorPattern
+        args.evaluatorPattern,
+        concurrency
       );
       return;
     }