katt 0.0.5 → 0.0.6

package/dist/index.js

@@ -1,56 +1,56 @@
-import { r as E, c as N, a as O, p as I, b as m, l as d, d as w, g as F, e as U, f as C, s as _, h as R, i as D, j as z, k as B, m as W, n as K } from "./runCli-CDRmZ5hw.js";
-import { o as St } from "./runCli-CDRmZ5hw.js";
-import { CopilotClient as H } from "@github/copilot-sdk";
-import { readFile as A } from "node:fs/promises";
-import { resolve as j, dirname as T, isAbsolute as Y, basename as J, join as q } from "node:path";
-import { readFileSync as G, writeFileSync as S, mkdirSync as Q } from "node:fs";
-function vt(t, e) {
-  E(() => {
-    O(), I(t);
-    const n = F(), r = Date.now(), i = () => F() === n, o = () => Date.now() - r;
+import { r as $, c as x, p as E, a as d, l, b as p, g as w, d as N, e as P, f as h, s as A, h as S, i as j, j as _, k as O, m as L, n as I, o as U, q as R } from "./runCli-B3oIBxOl.js";
+import { t as xt } from "./runCli-B3oIBxOl.js";
+import { CopilotClient as z } from "@github/copilot-sdk";
+import { readFile as F } from "node:fs/promises";
+import { resolve as k, dirname as g, isAbsolute as B, basename as W, join as K } from "node:path";
+import { readFileSync as H, writeFileSync as C, mkdirSync as Y } from "node:fs";
+function wt(t, e) {
+  $(() => {
+    N(), E(t);
+    const n = w(), o = Date.now(), r = () => w() === n, s = () => Date.now() - o;
     try {
-      const s = e();
-      if (s && typeof s.then == "function") {
-        m(
-          s.then(() => {
-            d(!0, o());
+      const i = e();
+      if (i && typeof i.then == "function") {
+        d(
+          i.then(() => {
+            l(!0, s());
           }).catch((a) => {
-            throw d(!1, o()), a;
+            throw l(!1, s()), a;
           }).finally(() => {
-            w();
+            p();
           })
         );
         return;
       }
-    } catch (s) {
-      throw d(!1, o()), w(), s;
+    } catch (i) {
+      throw l(!1, s()), p(), i;
     }
-    d(i(), o()), w();
-  }, N());
+    l(r(), s()), p();
+  }, x());
 }
-function Tt(t, e) {
-  E(() => {
-    U(t);
+function Ct(t, e) {
+  $(() => {
+    P(t);
     try {
       const n = e();
       if (n && typeof n.then == "function") {
-        m(
+        d(
           n.finally(() => {
-            C();
+            h();
           })
         );
         return;
       }
     } catch (n) {
-      throw C(), n;
+      throw h(), n;
     }
-    C();
-  }, N());
+    h();
+  }, x());
 }
-function V(t, e) {
+function q(t, e) {
   return typeof t == "object" && t !== null && "code" in t && t.code === e;
 }
-function X(t) {
+function J(t) {
   try {
     const e = JSON.parse(t);
     return typeof e == "object" && e !== null ? e : void 0;
@@ -59,133 +59,155 @@ function X(t) {
     return;
   }
 }
-async function Z() {
-  const t = j(process.cwd(), "katt.json");
+async function G() {
+  const t = k(process.cwd(), "katt.json");
   try {
-    const e = await A(t, "utf8");
-    return X(e);
+    const e = await F(t, "utf8");
+    return J(e);
   } catch (e) {
-    if (V(e, "ENOENT"))
+    if (q(e, "ENOENT"))
       return;
     console.warn(`Failed to read katt.json: ${String(e)}`);
     return;
   }
 }
-function tt(t) {
+function Q(t) {
   const e = t?.copilot;
   if (typeof e != "object" || e === null || Array.isArray(e))
     return;
   const n = {
     ...e
-  }, r = n.model;
-  return (typeof r != "string" || r.length === 0) && delete n.model, Object.keys(n).length > 0 ? n : void 0;
+  }, o = n.model;
+  return (typeof o != "string" || o.length === 0) && delete n.model, Object.keys(n).length > 0 ? n : void 0;
 }
-function et(t) {
+function V(t) {
   if (!(typeof t != "number" || !Number.isFinite(t)) && !(t <= 0))
     return Math.floor(t);
 }
-function nt(t) {
+function X(t) {
   const e = t?.prompt;
   if (!(typeof e != "object" || e === null || Array.isArray(e)))
-    return et(e.timeoutMs);
+    return V(e.timeoutMs);
 }
-async function ot() {
-  const t = await Z();
+async function Z() {
+  const t = await G();
   return {
-    copilot: tt(t),
-    promptTimeoutMs: nt(t)
+    copilot: Q(t),
+    promptTimeoutMs: X(t)
   };
 }
-const rt = 6e5;
-function P(t) {
+const tt = 6e5;
+function M(t) {
   return typeof t == "string" && t.length > 0 ? t : void 0;
 }
-function v(t) {
+function m(t) {
   if (!t)
     return;
   const e = { ...t };
   if (e.model !== void 0) {
-    const n = P(e.model);
+    const n = M(e.model);
     n ? e.model = n : delete e.model;
   }
   return Object.keys(e).length > 0 ? e : void 0;
 }
-function M(t) {
+function b(t) {
   if (!(typeof t != "number" || !Number.isFinite(t)) && !(t <= 0))
     return Math.floor(t);
 }
-function h(t) {
+function f(t) {
   return !Number.isFinite(t) || (t ?? 0) <= 0 ? 0 : Math.floor(t ?? 0);
 }
-function it(t) {
-  return h(t.inputTokens) + h(t.outputTokens) + h(t.cacheReadTokens) + h(t.cacheWriteTokens);
+function et(t) {
+  return f(t.inputTokens) + f(t.outputTokens) + f(t.cacheReadTokens) + f(t.cacheWriteTokens);
 }
-async function $(t, e = {}) {
-  const { timeoutMs: n, ...r } = e, i = await ot(), o = v(i.copilot), s = v(
-    r
-  ), a = v({
-    ...o ?? {},
+function nt(t, e) {
+  const { timeoutMs: n, ...o } = t, r = m(e.copilot), s = m(
+    o
+  ), i = m({
+    ...r ?? {},
     ...s ?? {}
-  }), c = M(i.promptTimeoutMs), L = M(n) ?? c ?? rt, b = P(a?.model), g = new H({ useLoggedInUser: !0 });
-  let p, x, y = 0;
-  try {
-    await g.start(), p = await g.createSession(a), x = p.on("assistant.usage", (f) => {
-      y += it(f.data);
-    });
-    const l = await p.sendAndWait({ prompt: t }, L);
-    if (!l?.data?.content)
-      throw new Error("Copilot did not return a response.");
-    return b && _(b), l.data.content;
-  } finally {
-    const l = [];
-    if (x?.(), y > 0 && R(y), p)
-      try {
-        await p.destroy();
-      } catch (f) {
-        l.push(f);
-      }
+  }), a = b(e.promptTimeoutMs), D = b(n) ?? a ?? tt;
+  return {
+    sessionOptions: i,
+    model: M(i?.model),
+    timeoutMs: D
+  };
+}
+async function ot(t, e) {
+  const n = [];
+  if (e.unsubscribeUsage?.(), e.usedTokens > 0 && j(e.usedTokens), e.session)
     try {
-      const f = await g.stop();
-      l.push(...f);
-    } catch (f) {
-      l.push(f);
+      await e.session.destroy();
+    } catch (o) {
+      n.push(o);
     }
-    l.length > 0 && console.error(
-      `Copilot cleanup encountered ${l.length} error(s).`
+  try {
+    const o = await t.stop();
+    n.push(...o);
+  } catch (o) {
+    n.push(o);
+  }
+  n.length > 0 && console.error(
+    `Copilot cleanup encountered ${n.length} error(s).`
+  );
+}
+async function y(t, e = {}) {
+  const n = await Z(), o = nt(e, n), r = new z({ useLoggedInUser: !0 }), s = {
+    session: void 0,
+    unsubscribeUsage: void 0,
+    usedTokens: 0
+  };
+  try {
+    await r.start(), s.session = await r.createSession(
+      o.sessionOptions
+    ), s.unsubscribeUsage = s.session.on(
+      "assistant.usage",
+      (a) => {
+        s.usedTokens += et(a.data);
+      }
+    );
+    const i = await s.session.sendAndWait(
+      { prompt: t },
+      o.timeoutMs
     );
+    if (!i?.data?.content)
+      throw new Error("Copilot did not return a response.");
+    return o.model && A(o.model), i.data.content;
+  } finally {
+    await ot(r, s);
   }
 }
-async function $t(t, e = {}) {
-  const n = D.getStore(), r = n?.evalFile ? T(n.evalFile) : process.cwd(), i = Y(t) ? t : j(r, t), o = await A(i, "utf8");
-  return $(o, e);
+async function bt(t, e = {}) {
+  const n = S.getStore(), o = n?.evalFile ? g(n.evalFile) : process.cwd(), r = B(t) ? t : k(o, t), s = await F(r, "utf8");
+  return y(s, e);
 }
-function u(t) {
-  z({
-    describePath: W(),
-    itPath: B(),
+function c(t) {
+  _({
+    describePath: L(),
+    itPath: O(),
     message: t
   });
 }
 async function st(t, e) {
-  const n = Date.now(), r = `expected '${t}' to satisfy '${e}'`, i = $(`Evaluate if the expectation is fulfiled in by the input.
+  const n = Date.now(), o = `expected '${t}' to satisfy '${e}'`, r = y(`Evaluate if the expectation is fulfiled in by the input.
         Expectation: "${e}".
         Input:
         ---
         ${t}
         ---
         Important: Answer with "Yes" or "No" only, without any additional text.
-        `).then((o) => {
-    o.includes("Yes") ? d(!0, Date.now() - n, "promptCheck") : o.includes("No") ? u(r) : u(`failed to evaluate expectation '${e}'`);
+        `).then((s) => {
+    s.includes("Yes") ? l(!0, Date.now() - n, "promptCheck") : s.includes("No") ? c(o) : c(`failed to evaluate expectation '${e}'`);
   });
-  return m(i), i;
+  return d(r), r;
 }
-const at = 3;
-function ct(t) {
+const rt = 3;
+function it(t) {
   const e = t.match(/\b([1-5])\b/);
   return !e || !e[1] ? null : Number(e[1]);
 }
-async function ut(t, e, n) {
-  const r = Date.now(), i = n?.threshold ?? at, o = $(
+async function at(t, e, n) {
+  const o = Date.now(), r = n?.threshold ?? rt, s = y(
     `Classify the input by how "${e}" it is on a scale of 1 to 5.
 1 means "not ${e}" and 5 means "very ${e}".
 Return only a single number: 1, 2, 3, 4, or 5.
@@ -195,129 +217,141 @@ Input:
 ${t}
 ---`,
     n?.model ? { model: n.model } : void 0
-  ).then((s) => {
-    const a = ct(s);
+  ).then((i) => {
+    const a = it(i);
     if (a === null) {
-      u(
-        `failed to classify as '${e}'. Evaluator returned '${s}'`
+      c(
+        `failed to classify as '${e}'. Evaluator returned '${i}'`
       );
       return;
     }
-    const c = `expected response to be classified as '${e}' with score >= ${i}, got ${a}`;
-    if (a < i) {
-      u(c);
+    const u = `expected response to be classified as '${e}' with score >= ${r}, got ${a}`;
+    if (a < r) {
+      c(u);
       return;
     }
-    d(
+    l(
       !0,
-      Date.now() - r,
+      Date.now() - o,
       "toBeClassifiedAs"
     );
   });
-  return m(o), o;
+  return d(s), s;
 }
-function lt(t, e) {
+function ct(t, e) {
   const n = `expected '${t}' to include '${e}'`;
-  t.includes(e) || u(n);
+  t.includes(e) || c(n);
+}
+function v(t) {
+  const e = t.trim().replace(/[<>:"/\\|?*\x00-\x1f]/g, "_").replace(/\s+/g, "_");
+  return e.length > 0 ? e : "unnamed";
+}
+function ut() {
+  const t = U().map(
+    (o) => v(o.description)
+  ), e = R().map(
+    (o) => v(o.description)
+  ), n = [...t, ...e];
+  return n.length === 0 ? "root" : n.join("__");
 }
-function ft(t) {
-  const n = J(t).replace(/\.eval\.[^./\\]+$/, "");
-  return q(
-    T(t),
+function lt(t) {
+  const n = W(t).replace(/\.eval\.[^./\\]+$/, ""), o = ut();
+  return K(
+    g(t),
     "__snapshots__",
-    `${n}.snap.md`
+    `${n}__${o}.snap.md`
   );
 }
-function k(t) {
+function T(t) {
   return t.split(/\r?\n/);
 }
-function dt(t, e) {
+function ft(t, e) {
   if (t === e)
     return "  (no diff)";
-  const n = k(t), r = k(e), i = Math.max(n.length, r.length), o = [];
-  for (let s = 0; s < i; s += 1) {
-    const a = n[s], c = r[s];
-    if (a !== c) {
-      if (a === void 0 && c !== void 0) {
-        o.push(`+ ${c}`);
+  const n = T(t), o = T(e), r = Math.max(n.length, o.length), s = [];
+  for (let i = 0; i < r; i += 1) {
+    const a = n[i], u = o[i];
+    if (a !== u) {
+      if (a === void 0 && u !== void 0) {
+        s.push(`+ ${u}`);
         continue;
       }
-      if (a !== void 0 && c === void 0) {
-        o.push(`- ${a}`);
+      if (a !== void 0 && u === void 0) {
+        s.push(`- ${a}`);
         continue;
       }
-      o.push(`- ${a ?? ""}`), o.push(`+ ${c ?? ""}`);
+      s.push(`- ${a ?? ""}`), s.push(`+ ${u ?? ""}`);
     }
   }
-  return o.join(`
+  return s.join(`
 `);
 }
-function pt(t) {
-  const e = D.getStore()?.evalFile;
+function dt(t) {
+  const e = S.getStore()?.evalFile;
   if (!e) {
-    u(
+    c(
       "toMatchSnapshot can only be used while running an eval file."
     );
     return;
   }
-  const n = ft(e);
+  const n = lt(e);
   try {
-    const r = G(n, "utf8");
-    if (r === t)
+    const o = H(n, "utf8");
+    if (o === t)
       return;
-    if (K()) {
-      S(n, t, "utf8");
+    if (I()) {
+      C(n, t, "utf8");
       return;
     }
-    const i = dt(r, t);
-    u(
+    const r = ft(o, t);
+    c(
       [
         `Snapshot mismatch at ${n}`,
         "",
         "Diff:",
-        i,
+        r,
         "",
         "Run katt with --update-snapshots (or -u) to accept this change."
       ].join(`
 `)
     );
-  } catch (r) {
-    if (r.code !== "ENOENT") {
-      u(
-        `Failed to read snapshot at ${n}: ${String(r)}`
+  } catch (o) {
+    if (o.code !== "ENOENT") {
+      c(
+        `Failed to read snapshot at ${n}: ${String(o)}`
       );
       return;
     }
     try {
-      Q(T(n), { recursive: !0 }), S(n, t, "utf8");
-    } catch (o) {
-      u(
-        `Failed to write snapshot at ${n}: ${String(o)}`
+      Y(g(n), { recursive: !0 }), C(n, t, "utf8");
+    } catch (s) {
+      c(
+        `Failed to write snapshot at ${n}: ${String(s)}`
       );
     }
   }
 }
-function bt(t) {
+function vt(t) {
   return {
     toContain: (e) => {
-      lt(t, e);
+      ct(t, e);
     },
     toMatchSnapshot: () => {
-      pt(t);
+      dt(t);
     },
     promptCheck: async (e) => {
       await st(t, e);
     },
     toBeClassifiedAs: async (e, n) => {
-      await ut(t, e, n);
+      await at(t, e, n);
     }
   };
 }
 export {
-  Tt as describe,
-  bt as expect,
-  vt as it,
-  $ as prompt,
-  $t as promptFile,
-  St as runCli
+  Ct as describe,
+  vt as expect,
+  wt as it,
+  y as prompt,
+  bt as promptFile,
+  xt as runCli
 };

package/dist/katt.js

@@ -1,5 +1,5 @@
 #!/usr/bin/env node
-import { o as r } from "./runCli-CDRmZ5hw.js";
+import { t as r } from "./runCli-B3oIBxOl.js";
 r().then((e) => {
   process.exit(e);
 }).catch((e) => {

package/dist/runCli-B3oIBxOl.js

@@ -0,0 +1,317 @@
+import { fileURLToPath as Z, pathToFileURL as y } from "node:url";
+import { readdir as N } from "node:fs/promises";
+import { resolve as M } from "node:path";
+import { AsyncLocalStorage as v } from "node:async_hooks";
+import { readFileSync as X } from "node:fs";
+const H = /\.eval\.(js|ts)$/, R = /* @__PURE__ */ new Set([".git", "node_modules"]);
+async function w(t) {
+  const e = await N(t, { withFileTypes: !0 }), n = [];
+  return await Promise.all(
+    e.map(async (i) => {
+      const s = M(t, i.name);
+      if (i.isDirectory()) {
+        if (R.has(i.name))
+          return;
+        n.push(...await w(s));
+        return;
+      }
+      i.isFile() && H.test(i.name) && n.push(s);
+    })
+  ), n;
+}
+const j = new v(), W = {
+  describeStack: [],
+  itStack: [],
+  tokenUsageStack: [],
+  modelStack: []
+};
+let b = 0, L = 0;
+const p = [], f = [];
+let k = 0;
+function o() {
+  return j.getStore() ?? W;
+}
+function J(t) {
+  return {
+    describeStack: [...t.describeStack],
+    itStack: [...t.itStack],
+    tokenUsageStack: [...t.tokenUsageStack],
+    modelStack: [...t.modelStack]
+  };
+}
+function U() {
+  return b += 1, `d${b}`;
+}
+function G() {
+  return L += 1, `i${L}`;
+}
+function B(t) {
+  return t.tokenUsageStack.length - 1;
+}
+function F(t) {
+  return t.modelStack.length - 1;
+}
+function bt(t, e) {
+  const n = e ?? J(o());
+  return j.run(n, t);
+}
+function Lt() {
+  return J(o());
+}
+function $t(t) {
+  o().describeStack.push({ id: U(), description: t });
+}
+function vt() {
+  o().describeStack.pop();
+}
+function O() {
+  return o().describeStack.map((t) => t.description).join(" > ");
+}
+function wt() {
+  return [...o().describeStack];
+}
+function jt(t) {
+  const e = o();
+  e.itStack.push({ id: G(), description: t }), e.tokenUsageStack.push(0), e.modelStack.push(void 0);
+}
+function Jt() {
+  const t = o();
+  t.itStack.pop(), t.tokenUsageStack.pop(), t.modelStack.pop();
+}
+function Y() {
+  return o().itStack.map((t) => t.description).join(" > ");
+}
+function Bt() {
+  return [...o().itStack];
+}
+function Ft(t) {
+  if (!Number.isFinite(t) || t <= 0)
+    return;
+  const e = o(), n = B(e);
+  n < 0 || (e.tokenUsageStack[n] += t);
+}
+function Q() {
+  const t = o(), e = B(t);
+  return e < 0 ? 0 : t.tokenUsageStack[e] ?? 0;
+}
+function xt(t) {
+  if (t.length === 0)
+    return;
+  const e = o(), n = F(e);
+  n < 0 || (e.modelStack[n] = t);
+}
+function V() {
+  const t = o(), e = F(t);
+  if (!(e < 0))
+    return t.modelStack[e];
+}
+function Tt(t) {
+  p.push(t);
+}
+function Zt() {
+  k += 1;
+}
+function K() {
+  return k;
+}
+function P() {
+  k = 0;
+}
+function yt(t) {
+  f.push(t);
+}
+function E() {
+  return [...f];
+}
+function Nt() {
+  return f.length;
+}
+function z() {
+  f.length = 0;
+}
+async function D() {
+  const t = [];
+  for (; p.length > 0; ) {
+    const e = p.splice(0, p.length), n = await Promise.allSettled(e);
+    t.push(...n);
+  }
+  return t;
+}
+const _ = "\x1B[1;36m", q = "\x1B[33m", tt = "\x1B[38;5;208m", et = "\x1B[1;38;5;208m", m = "\x1B[0m";
+function r(t) {
+  return `${_}${t}${m}`;
+}
+function C(t) {
+  return `${q}${t}${m}`;
+}
+function $(t) {
+  return `${tt}${t}${m}`;
+}
+function nt(t) {
+  return `${et}${t}${m}`;
+}
+let A = "";
+function it() {
+  A = "";
+}
+function ot({
+  suitePath: t,
+  casePath: e,
+  didPass: n,
+  durationMs: i,
+  model: s,
+  tokenUsage: c
+}) {
+  const l = t.length > 0 ? t : "(root)", u = e.length > 0 ? e : "(root)";
+  A !== l && (console.log(`Suite "${r(l)}"`), A = l);
+  const g = n ? "✅ Passed in" : "❌ Failed in", I = [
+    `Test "${r(u)}"`,
+    `- ${g} ${r(`${i}ms`)}`
+  ];
+  s && I.push(`- Model ${r(s)}`), (c ?? 0) > 0 && I.push(`- Tokens used ${r(String(c))}`), I.push("---"), console.log(I.join(`
+`));
+}
+function Mt(t, e, n = "(root)") {
+  const i = Y();
+  ot({
+    suitePath: O(),
+    casePath: i.length > 0 ? i : n,
+    didPass: t,
+    durationMs: e,
+    model: V(),
+    tokenUsage: Q()
+  });
+}
+const st = new v(), h = new URL("data:application/json;base64,ewogICJuYW1lIjogImthdHQiLAogICJ2ZXJzaW9uIjogIjAuMC41IiwKICAiZGVzY3JpcHRpb24iOiAiQ0xJIHRvb2wgdGhhdCB0ZXN0cyB0aGUgb3V0cHV0IG9mIGFnZW50aWMgQUkgdG9vbHMiLAogICJrZXl3b3JkcyI6IFsKICAgICJjbGkiLAogICAgImFpIiwKICAgICJhZ2VudGljLWFpIiwKICAgICJ0ZXN0aW5nIiwKICAgICJldmFsdWF0aW9uIgogIF0sCiAgImF1dGhvciI6ICJSYXBoYWVsIFBvcnRvIChodHRwczovL2dpdGh1Yi5jb20vcmFwaGFlbHBvcikiLAogICJsaWNlbnNlIjogIk1JVCIsCiAgInR5cGUiOiAibW9kdWxlIiwKICAibWFpbiI6ICJkaXN0L2luZGV4LmpzIiwKICAiZXhwb3J0cyI6IHsKICAgICIuIjogewogICAgICAidHlwZXMiOiAiLi9kaXN0L2luZGV4LmQudHMiLAogICAgICAiaW1wb3J0IjogIi4vZGlzdC9pbmRleC5qcyIKICAgIH0KICB9LAogICJiaW4iOiB7CiAgICAia2F0dCI6ICJkaXN0L2thdHQuanMiCiAgfSwKICAic2NyaXB0cyI6IHsKICAgICJidWlsZCI6ICJ2aXRlIGJ1aWxkIiwKICAgICJkZXYiOiAidHN4IHNyYy9pbmRleC50cyIsCiAgICAibGludCI6ICJiaW9tZSBsaW50IC4vc3JjIiwKICAgICJmb3JtYXQiOiAiYmlvbWUgZm9ybWF0IC0td3JpdGUgLi9zcmMiLAogICAgInRlc3QiOiAidml0ZXN0IiwKICAgICJ0eXBlY2hlY2siOiAidHNjIC1wIHRzY29uZmlnLmpzb24gLS1ub0VtaXQiLAogICAgInRlc3Q6YnVpbGQiOiAibm9kZSAuL2Rpc3Qva2F0dC5qcyIKICB9LAogICJ0eXBlcyI6ICJkaXN0L2luZGV4LmQudHMiLAogICJkZXZEZXBlbmRlbmNpZXMiOiB7CiAgICAiQGJpb21lanMvYmlvbWUiOiAiMS45LjQiLAogICAgIkB0eXBlcy9ub2RlIjogIjI1LjIuMCIsCiAgICAidHN4IjogIjQuMjEuMCIsCiAgICAidHlwZXNjcmlwdCI6ICI1LjguMiIsCiAgICAidml0ZSI6ICI3LjMuMSIsCiAgICAidml0ZS1wbHVnaW4tZHRzIjogIjQuNS40IiwKICAgICJ2aXRlc3QiOiAiMy4yLjQiLAogICAgInZzY29kZS1qc29ucnBjIjogIl44LjIuMSIKICB9LAogICJkZXBlbmRlbmNpZXMiOiB7CiAgICAiQGdpdGh1Yi9jb3BpbG90LXNkayI6ICJeMC4xLjIxIgogIH0sCiAgImJ1Z3MiOiB7CiAgICAidXJsIjogImh0dHBzOi8vZ2l0aHViLmNvbS9yYXBoYWVscG9yL2thdHQvaXNzdWVzIgogIH0sCiAgImhvbWVwYWdlIjogImh0dHBzOi8vZ2l0aHViLmNvbS9yYXBoYWVscG9yL2thdHQiCn0K", import.meta.url);
+let d;
+function at() {
+  if (d !== void 0)
+    return d;
+  try {
+    const t = h.protocol === "data:" ? rt(h) : X(Z(h), "utf8"), e = JSON.parse(t);
+    d = typeof e.version == "string" ? e.version : "unknown";
+  } catch {
+    d = "unknown";
+  }
+  return d;
+}
+function rt(t) {
+  const e = t.pathname.indexOf(",");
+  if (e < 0)
+    throw new Error("Invalid data URL.");
+  const n = t.pathname.slice(0, e), i = t.pathname.slice(e + 1);
+  return n.includes(";base64") ? Buffer.from(i, "base64").toString("utf8") : decodeURIComponent(i);
+}
+function ct() {
+  const t = " ██╗  ██╗ █████╗ ████████╗████████╗", e = " ██║ ██╔╝██╔══██╗╚══██╔══╝╚══██╔══╝", n = " █████╔╝ ███████║   ██║      ██║", i = " ██╔═██╗ ██╔══██║   ██║      ██║", s = " ██║  ██╗██║  ██║   ██║      ██║", c = " ╚═╝  ╚═╝╚═╝  ╚═╝   ╚═╝      ╚═╝", l = `v${at()}`, u = Math.max(
+    0,
+    Math.floor((t.length - l.length) / 2)
+  ), g = `${" ".repeat(u)}${l}`;
+  console.log(`
+${C(t)}
+${C(e)}
+${C(n)}
+${$(i)}
+${$(s)}
+${nt(c)}
+${C(g)}
+`);
+}
+let x = !1;
+function lt(t) {
+  x = t;
+}
+function Xt() {
+  return x;
+}
+function ut(t) {
+  const e = String(t.getHours()).padStart(2, "0"), n = String(t.getMinutes()).padStart(2, "0"), i = String(t.getSeconds()).padStart(2, "0");
+  return `${e}:${n}:${i}`;
+}
+function gt(t) {
+  return t.includes("--update-snapshots") || t.includes("-u");
+}
+function It(t) {
+  return [t.describePath, t.itPath].filter((e) => e.length > 0).join(" > ");
+}
+function dt(t) {
+  for (const e of t)
+    console.error(`Error executing ${e.file}: ${String(e.reason)}`);
+}
+function Ct(t) {
+  for (const e of t)
+    console.error(`Error executing async test: ${String(e.reason)}`);
+}
+function pt(t) {
+  console.error("❌ Failed tests:");
+  for (const [e, n] of t.entries()) {
+    const i = It(n), s = i.length > 0 ? `${i}: ` : "";
+    console.error(`${e + 1}. ${s}${n.message}`);
+  }
+}
+function ft(t, e, n, i) {
+  return [
+    "---",
+    `${r("Files")}  ${t} passed`,
+    `${r("Evals")}  ${e} passed`,
+    `${r("Start at")}  ${ut(n)}`,
+    `${r("Duration")}  ${i}ms`
+  ].join(`
+`);
+}
+async function Ht() {
+  const t = process.argv.slice(2), e = gt(t);
+  lt(e), ct();
+  const n = /* @__PURE__ */ new Date();
+  it(), z(), P();
+  const i = await w(process.cwd());
+  if (i.length === 0)
+    return console.log("No .eval.js or .eval.ts files found."), 1;
+  const c = (await Promise.allSettled(
+    i.map(
+      (a) => st.run(
+        { evalFile: a },
+        () => import(y(a).href)
+      )
+    )
+  )).map((a, S) => ({ result: a, file: i[S] })).filter(({ result: a }) => a.status === "rejected").map(({ result: a, file: S }) => ({
+    file: S,
+    reason: a.status === "rejected" ? a.reason : void 0
+  }));
+  if (c.length > 0)
+    return dt(c), 1;
+  const u = (await D()).filter(
+    (a) => a.status === "rejected"
+  );
+  if (u.length > 0)
+    return Ct(u), 1;
+  const g = E();
+  if (g.length > 0)
+    return pt(g), 1;
+  const I = K(), T = Date.now() - n.getTime();
+  return console.log(
+    ft(i.length, I, n, T)
+  ), 0;
+}
+export {
+  Tt as a,
+  Jt as b,
+  Lt as c,
+  Zt as d,
+  $t as e,
+  vt as f,
+  Nt as g,
+  st as h,
+  Ft as i,
+  yt as j,
+  Y as k,
+  Mt as l,
+  O as m,
+  Xt as n,
+  wt as o,
+  jt as p,
+  Bt as q,
+  bt as r,
+  xt as s,
+  Ht as t
+};

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "katt",
-  "version": "0.0.5",
+  "version": "0.0.6",
   "description": "CLI tool that tests the output of agentic AI tools",
   "keywords": [
     "cli",

package/dist/runCli-CDRmZ5hw.js

@@ -1,288 +0,0 @@
-import { fileURLToPath as x, pathToFileURL as y } from "node:url";
-import { readdir as X } from "node:fs/promises";
-import { resolve as H } from "node:path";
-import { AsyncLocalStorage as j } from "node:async_hooks";
-import { readFileSync as M } from "node:fs";
-const R = /\.eval\.(js|ts)$/, W = /* @__PURE__ */ new Set([".git", "node_modules"]);
-async function v(t) {
-  const e = await X(t, { withFileTypes: !0 }), n = [];
-  return await Promise.all(
-    e.map(async (o) => {
-      const a = H(t, o.name);
-      if (o.isDirectory()) {
-        if (W.has(o.name))
-          return;
-        n.push(...await v(a));
-        return;
-      }
-      o.isFile() && R.test(o.name) && n.push(a);
-    })
-  ), n;
-}
-const J = new j(), G = {
-  describeStack: [],
-  itStack: [],
-  tokenUsageStack: [],
-  modelStack: []
-};
-let L = 0, $ = 0;
-const p = [], m = [];
-let k = 0;
-function s() {
-  return J.getStore() ?? G;
-}
-function B(t) {
-  return {
-    describeStack: [...t.describeStack],
-    itStack: [...t.itStack],
-    tokenUsageStack: [...t.tokenUsageStack],
-    modelStack: [...t.modelStack]
-  };
-}
-function U() {
-  return L += 1, `d${L}`;
-}
-function O() {
-  return $ += 1, `i${$}`;
-}
-function mt(t, e) {
-  const n = e ?? B(s());
-  return J.run(n, t);
-}
-function ht() {
-  return B(s());
-}
-function St(t) {
-  s().describeStack.push({ id: U(), description: t });
-}
-function At() {
-  s().describeStack.pop();
-}
-function Y() {
-  return s().describeStack.map((t) => t.description).join(" > ");
-}
-function kt(t) {
-  s().itStack.push({ id: O(), description: t }), s().tokenUsageStack.push(0), s().modelStack.push(void 0);
-}
-function bt() {
-  s().itStack.pop(), s().tokenUsageStack.pop(), s().modelStack.pop();
-}
-function Q() {
-  return s().itStack.map((t) => t.description).join(" > ");
-}
-function Lt(t) {
-  if (!Number.isFinite(t) || t <= 0)
-    return;
-  const e = s(), n = e.tokenUsageStack.length - 1;
-  n < 0 || (e.tokenUsageStack[n] += t);
-}
-function V() {
-  const t = s(), e = t.tokenUsageStack.length - 1;
-  return e < 0 ? 0 : t.tokenUsageStack[e] ?? 0;
-}
-function $t(t) {
-  if (t.length === 0)
-    return;
-  const e = s(), n = e.modelStack.length - 1;
-  n < 0 || (e.modelStack[n] = t);
-}
-function K() {
-  const t = s(), e = t.modelStack.length - 1;
-  if (!(e < 0))
-    return t.modelStack[e];
-}
-function wt(t) {
-  p.push(t);
-}
-function jt() {
-  k += 1;
-}
-function P() {
-  return k;
-}
-function z() {
-  k = 0;
-}
-function vt(t) {
-  m.push(t);
-}
-function E() {
-  return [...m];
-}
-function Jt() {
-  return m.length;
-}
-function D() {
-  m.length = 0;
-}
-async function _() {
-  const t = [];
-  for (; p.length > 0; ) {
-    const e = p.splice(0, p.length), n = await Promise.allSettled(e);
-    t.push(...n);
-  }
-  return t;
-}
-const q = "\x1B[1;36m", tt = "\x1B[33m", et = "\x1B[38;5;208m", nt = "\x1B[1;38;5;208m", h = "\x1B[0m";
-function c(t) {
-  return `${q}${t}${h}`;
-}
-function f(t) {
-  return `${tt}${t}${h}`;
-}
-function w(t) {
-  return `${et}${t}${h}`;
-}
-function ot(t) {
-  return `${nt}${t}${h}`;
-}
-let A = "";
-function it() {
-  A = "";
-}
-function st({
-  suitePath: t,
-  casePath: e,
-  didPass: n,
-  durationMs: o,
-  model: a,
-  tokenUsage: r
-}) {
-  const l = t.length > 0 ? t : "(root)", g = e.length > 0 ? e : "(root)";
-  A !== l && (console.log(`Suite "${c(l)}"`), A = l);
-  const I = n ? "✅ Passed in" : "❌ Failed in", d = [
-    `Test "${c(g)}"`,
-    `- ${I} ${c(`${o}ms`)}`
-  ];
-  a && d.push(`- Model ${c(a)}`), (r ?? 0) > 0 && d.push(`- Tokens used ${c(String(r))}`), d.push("---"), console.log(d.join(`
-`));
-}
-function Bt(t, e, n = "(root)") {
-  const o = Q();
-  st({
-    suitePath: Y(),
-    casePath: o.length > 0 ? o : n,
-    didPass: t,
-    durationMs: e,
-    model: K(),
-    tokenUsage: V()
-  });
-}
-const at = new j(), S = new URL("data:application/json;base64,ewogICJuYW1lIjogImthdHQiLAogICJ2ZXJzaW9uIjogIjAuMC40IiwKICAiZGVzY3JpcHRpb24iOiAiQ0xJIHRvb2wgdGhhdCB0ZXN0cyB0aGUgb3V0cHV0IG9mIGFnZW50aWMgQUkgdG9vbHMiLAogICJrZXl3b3JkcyI6IFsKICAgICJjbGkiLAogICAgImFpIiwKICAgICJhZ2VudGljLWFpIiwKICAgICJ0ZXN0aW5nIiwKICAgICJldmFsdWF0aW9uIgogIF0sCiAgImF1dGhvciI6ICJSYXBoYWVsIFBvcnRvIChodHRwczovL2dpdGh1Yi5jb20vcmFwaGFlbHBvcikiLAogICJsaWNlbnNlIjogIk1JVCIsCiAgInR5cGUiOiAibW9kdWxlIiwKICAibWFpbiI6ICJkaXN0L2luZGV4LmpzIiwKICAiZXhwb3J0cyI6IHsKICAgICIuIjogewogICAgICAidHlwZXMiOiAiLi9kaXN0L2luZGV4LmQudHMiLAogICAgICAiaW1wb3J0IjogIi4vZGlzdC9pbmRleC5qcyIKICAgIH0KICB9LAogICJiaW4iOiB7CiAgICAia2F0dCI6ICJkaXN0L2thdHQuanMiCiAgfSwKICAic2NyaXB0cyI6IHsKICAgICJidWlsZCI6ICJ2aXRlIGJ1aWxkIiwKICAgICJkZXYiOiAidHN4IHNyYy9pbmRleC50cyIsCiAgICAibGludCI6ICJiaW9tZSBsaW50IC4vc3JjIiwKICAgICJmb3JtYXQiOiAiYmlvbWUgZm9ybWF0IC0td3JpdGUgLi9zcmMiLAogICAgInRlc3QiOiAidml0ZXN0IiwKICAgICJ0eXBlY2hlY2siOiAidHNjIC1wIHRzY29uZmlnLmpzb24gLS1ub0VtaXQiLAogICAgInRlc3Q6YnVpbGQiOiAibm9kZSAuL2Rpc3Qva2F0dC5qcyIKICB9LAogICJ0eXBlcyI6ICJkaXN0L2luZGV4LmQudHMiLAogICJkZXZEZXBlbmRlbmNpZXMiOiB7CiAgICAiQGJpb21lanMvYmlvbWUiOiAiMS45LjQiLAogICAgIkB0eXBlcy9ub2RlIjogIjI1LjIuMCIsCiAgICAidHN4IjogIjQuMjEuMCIsCiAgICAidHlwZXNjcmlwdCI6ICI1LjguMiIsCiAgICAidml0ZSI6ICI3LjMuMSIsCiAgICAidml0ZS1wbHVnaW4tZHRzIjogIjQuNS40IiwKICAgICJ2aXRlc3QiOiAiMy4yLjQiLAogICAgInZzY29kZS1qc29ucnBjIjogIl44LjIuMSIKICB9LAogICJkZXBlbmRlbmNpZXMiOiB7CiAgICAiQGdpdGh1Yi9jb3BpbG90LXNkayI6ICJeMC4xLjIxIgogIH0sCiAgImJ1Z3MiOiB7CiAgICAidXJsIjogImh0dHBzOi8vZ2l0aHViLmNvbS9yYXBoYWVscG9yL2thdHQvaXNzdWVzIgogIH0sCiAgImhvbWVwYWdlIjogImh0dHBzOi8vZ2l0aHViLmNvbS9yYXBoYWVscG9yL2thdHQiCn0K", import.meta.url);
-let C;
-function ct() {
-  if (C !== void 0)
-    return C;
-  try {
-    const t = S.protocol === "data:" ? rt(S) : M(x(S), "utf8"), e = JSON.parse(t);
-    C = typeof e.version == "string" ? e.version : "unknown";
-  } catch {
-    C = "unknown";
-  }
-  return C;
-}
-function rt(t) {
-  const e = t.pathname.indexOf(",");
-  if (e < 0)
-    throw new Error("Invalid data URL.");
-  const n = t.pathname.slice(0, e), o = t.pathname.slice(e + 1);
-  return n.includes(";base64") ? Buffer.from(o, "base64").toString("utf8") : decodeURIComponent(o);
-}
-function lt() {
-  const t = " ██╗  ██╗ █████╗ ████████╗████████╗", e = " ██║ ██╔╝██╔══██╗╚══██╔══╝╚══██╔══╝", n = " █████╔╝ ███████║   ██║      ██║", o = " ██╔═██╗ ██╔══██║   ██║      ██║", a = " ██║  ██╗██║  ██║   ██║      ██║", r = " ╚═╝  ╚═╝╚═╝  ╚═╝   ╚═╝      ╚═╝", l = `v${ct()}`, g = Math.max(
-    0,
-    Math.floor((t.length - l.length) / 2)
-  ), I = `${" ".repeat(g)}${l}`;
-  console.log(`
-${f(t)}
-${f(e)}
-${f(n)}
-${w(o)}
-${w(a)}
-${ot(r)}
-${f(I)}
-`);
-}
-let Z = !1;
-function ut(t) {
-  Z = t;
-}
-function Zt() {
-  return Z;
-}
-function gt(t) {
-  const e = String(t.getHours()).padStart(2, "0"), n = String(t.getMinutes()).padStart(2, "0"), o = String(t.getSeconds()).padStart(2, "0");
-  return `${e}:${n}:${o}`;
-}
-async function Ft() {
-  const t = process.argv.slice(2), e = t.includes("--update-snapshots") || t.includes("-u");
-  ut(e), lt();
-  const n = /* @__PURE__ */ new Date();
-  it(), D(), z();
-  const o = await v(process.cwd());
-  if (o.length === 0)
-    return console.log("No .eval.js or .eval.ts files found."), 1;
-  const r = (await Promise.allSettled(
-    o.map(
-      (i) => at.run(
-        { evalFile: i },
-        () => import(y(i).href)
-      )
-    )
-  )).map((i, u) => ({ result: i, file: o[u] })).filter(({ result: i }) => i.status === "rejected");
-  if (r.length > 0) {
-    for (const i of r) {
-      const u = i.result.status === "rejected" ? i.result.reason : void 0;
-      console.error(`Error executing ${i.file}: ${String(u)}`);
-    }
-    return 1;
-  }
-  const g = (await _()).filter(
-    (i) => i.status === "rejected"
-  );
-  if (g.length > 0) {
-    for (const i of g)
-      i.status === "rejected" && console.error(`Error executing async test: ${String(i.reason)}`);
-    return 1;
-  }
-  const I = E();
-  if (I.length > 0) {
-    console.error("❌ Failed tests:");
-    for (const [i, u] of I.entries()) {
-      const b = [u.describePath, u.itPath].filter((T) => T.length > 0).join(" > "), N = b.length > 0 ? `${b}: ` : "";
-      console.error(`${i + 1}. ${N}${u.message}`);
-    }
-    return 1;
-  }
-  const d = P(), F = Date.now() - n.getTime();
-  return console.log(
-    [
-      "---",
-      `${c("Files")}  ${o.length} passed`,
-      `${c("Evals")}  ${d} passed`,
-      `${c("Start at")}  ${gt(n)}`,
-      `${c("Duration")}  ${F}ms`
-    ].join(`
-`)
-  ), 0;
-}
-export {
-  jt as a,
-  wt as b,
-  ht as c,
-  bt as d,
-  St as e,
-  At as f,
-  Jt as g,
-  Lt as h,
-  at as i,
-  vt as j,
-  Q as k,
-  Bt as l,
-  Y as m,
-  Zt as n,
-  Ft as o,
-  kt as p,
-  mt as r,
-  $t as s
-};