@evelinka-ai/core 0.1.3

package/dist/lib/budgets/token-usage-pg.js

@@ -0,0 +1,305 @@
+import { Pool } from "pg";
+import { logger } from "../observability/logger.js";
+const log = logger.child({ module: "token-usage" });
+let pool;
+let ddlDone;
+function getPool() {
+    const url = process.env.TOKEN_USAGE_POSTGRES_URL ?? process.env.WORKFLOW_POSTGRES_URL;
+    if (!url)
+        return undefined;
+    if (!pool) {
+        pool = new Pool({
+            connectionString: url,
+            max: 2,
+            connectionTimeoutMillis: 5_000,
+            query_timeout: 10_000,
+            statement_timeout: 10_000,
+        });
+    }
+    return pool;
+}
+const DDL = `CREATE TABLE IF NOT EXISTS token_usage (
+  id bigserial PRIMARY KEY,
+  session_id text NOT NULL,
+  root_id text NOT NULL,
+  flow text,
+  model text NOT NULL,
+  turn_id text,
+  step_index int,
+  at timestamptz NOT NULL,
+  input_tokens int NOT NULL DEFAULT 0,
+  output_tokens int NOT NULL DEFAULT 0,
+  cache_read_tokens int NOT NULL DEFAULT 0,
+  cache_write_tokens int NOT NULL DEFAULT 0,
+  instance text NOT NULL DEFAULT 'eve'
+);
+CREATE INDEX IF NOT EXISTS token_usage_at_idx ON token_usage (at);
+CREATE INDEX IF NOT EXISTS token_usage_root_idx ON token_usage (root_id);
+ALTER TABLE token_usage ADD COLUMN IF NOT EXISTS instance text NOT NULL DEFAULT 'eve';
+CREATE INDEX IF NOT EXISTS token_usage_instance_idx ON token_usage (instance);`;
+function ensureDdl(p) {
+    if (!ddlDone) {
+        ddlDone = p
+            .query(DDL)
+            .then(() => undefined)
+            .catch((err) => {
+            ddlDone = undefined;
+            throw err;
+        });
+    }
+    return ddlDone;
+}
+const INS = `INSERT INTO token_usage
+  (session_id, root_id, flow, model, turn_id, step_index, at,
+   input_tokens, output_tokens, cache_read_tokens, cache_write_tokens, instance)
+  VALUES ($1,$2,$3,$4,$5,$6,$7,$8,$9,$10,$11,$12)`;
+const usageListeners = new Set();
+export function onUsageRecorded(fn) {
+    usageListeners.add(fn);
+    return () => usageListeners.delete(fn);
+}
+export async function recordUsage(r) {
+    for (const fn of usageListeners) {
+        try {
+            fn(r);
+        }
+        catch (err) {
+            log.warn({ err: String(err) }, "usage listener threw — ignored");
+        }
+    }
+    const p = getPool();
+    if (!p)
+        return;
+    try {
+        await ensureDdl(p);
+        await p.query(INS, [
+            r.sessionId,
+            r.rootId,
+            r.flow,
+            r.model,
+            r.turnId,
+            r.stepIndex,
+            r.at,
+            r.input,
+            r.output,
+            r.cacheRead,
+            r.cacheWrite,
+            r.instance,
+        ]);
+    }
+    catch (err) {
+        log.error({ err }, "token usage write failed — swallowed");
+    }
+}
+const EMPTY_AGG = {
+    totals: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, requests: 0, sessions: 0 },
+    series: [],
+    seriesByModel: [],
+    byModel: [],
+    byFlow: [],
+    byInstance: [],
+    byRoot: [],
+};
+export async function readUsageAggregate(q) {
+    const p = getPool();
+    if (!p)
+        return EMPTY_AGG;
+    try {
+        await ensureDdl(p);
+        const args = [q.from, q.to];
+        let where = "at >= $1::timestamptz AND at < $2::timestamptz";
+        if (q.instance) {
+            args.push(q.instance);
+            where += ` AND instance = $${args.length}`;
+        }
+        const trunc = q.bucket === "hour" ? "hour" : "day";
+        const [totals, series, seriesByModel, byModel, byFlow, byInstance, byRoot] = await Promise.all([
+            p.query(`SELECT COALESCE(SUM(input_tokens),0)::int i, COALESCE(SUM(output_tokens),0)::int o,
+                COALESCE(SUM(cache_read_tokens),0)::int cr, COALESCE(SUM(cache_write_tokens),0)::int cw,
+                COUNT(*)::int reqs, COUNT(DISTINCT session_id)::int sess
+         FROM token_usage WHERE ${where}`, args),
+            p.query(`SELECT date_trunc('${trunc}', at) b,
+                SUM(input_tokens)::int i, SUM(output_tokens)::int o,
+                SUM(cache_read_tokens)::int cr, SUM(cache_write_tokens)::int cw
+         FROM token_usage WHERE ${where} GROUP BY b ORDER BY b`, args),
+            p.query(`SELECT date_trunc('${trunc}', at) b, model,
+                SUM(input_tokens + output_tokens)::int total
+         FROM token_usage WHERE ${where} GROUP BY b, model ORDER BY b`, args),
+            p.query(`SELECT model key, SUM(input_tokens)::int i, SUM(output_tokens)::int o,
+                SUM(input_tokens + output_tokens)::int total, COUNT(*)::int reqs
+         FROM token_usage WHERE ${where} GROUP BY model ORDER BY total DESC`, args),
+            p.query(`SELECT flow key, SUM(input_tokens)::int i, SUM(output_tokens)::int o,
+                SUM(input_tokens + output_tokens)::int total, COUNT(*)::int reqs
+         FROM token_usage WHERE ${where} GROUP BY flow ORDER BY total DESC`, args),
+            p.query(`SELECT instance key, SUM(input_tokens)::int i, SUM(output_tokens)::int o,
+                SUM(input_tokens + output_tokens)::int total, COUNT(*)::int reqs
+         FROM token_usage WHERE ${where} GROUP BY instance ORDER BY total DESC`, args),
+            p.query(`SELECT root_id, SUM(input_tokens)::int i, SUM(output_tokens)::int o,
+                SUM(input_tokens + output_tokens)::int total, COUNT(*)::int reqs
+         FROM token_usage WHERE ${where} GROUP BY root_id`, args),
+        ]);
+        const t = totals.rows[0] ?? {};
+        return {
+            totals: {
+                input: t.i ?? 0,
+                output: t.o ?? 0,
+                cacheRead: t.cr ?? 0,
+                cacheWrite: t.cw ?? 0,
+                requests: t.reqs ?? 0,
+                sessions: t.sess ?? 0,
+            },
+            series: series.rows.map((r) => ({
+                at: new Date(r.b).toISOString(),
+                input: r.i,
+                output: r.o,
+                cacheRead: r.cr,
+                cacheWrite: r.cw,
+            })),
+            seriesByModel: seriesByModel.rows.map((r) => ({
+                at: new Date(r.b).toISOString(),
+                model: r.model,
+                total: r.total,
+            })),
+            byModel: byModel.rows.map((r) => ({
+                key: r.key,
+                input: r.i,
+                output: r.o,
+                total: r.total,
+                requests: r.reqs,
+            })),
+            byFlow: byFlow.rows.map((r) => ({
+                key: r.key,
+                input: r.i,
+                output: r.o,
+                total: r.total,
+                requests: r.reqs,
+            })),
+            byInstance: byInstance.rows.map((r) => ({
+                key: r.key,
+                input: r.i,
+                output: r.o,
+                total: r.total,
+                requests: r.reqs,
+            })),
+            byRoot: byRoot.rows.map((r) => ({
+                rootId: r.root_id,
+                input: r.i,
+                output: r.o,
+                total: r.total,
+                requests: r.reqs,
+            })),
+        };
+    }
+    catch (err) {
+        log.error({ err }, "usage aggregate read failed — returning empty");
+        return EMPTY_AGG;
+    }
+}
+export async function readRunUsageByModel(rootId, flow) {
+    const p = getPool();
+    if (!p)
+        return [];
+    try {
+        await ensureDdl(p);
+        const args = [rootId];
+        let where = "root_id = $1";
+        if (flow) {
+            args.push(flow);
+            where += " AND flow = $2";
+        }
+        const res = await p.query(`SELECT model key, COALESCE(SUM(input_tokens),0)::int i, COALESCE(SUM(output_tokens),0)::int o,
+              COALESCE(SUM(cache_read_tokens),0)::int cr, COALESCE(SUM(cache_write_tokens),0)::int cw,
+              COALESCE(SUM(input_tokens + output_tokens),0)::int total, COUNT(*)::int reqs
+       FROM (
+         (SELECT DISTINCT ON (session_id, turn_id, step_index)
+                 model, input_tokens, output_tokens, cache_read_tokens, cache_write_tokens
+          FROM token_usage WHERE ${where} AND turn_id IS NOT NULL AND step_index IS NOT NULL
+          ORDER BY session_id, turn_id, step_index, id DESC)
+         UNION ALL
+         (SELECT model, input_tokens, output_tokens, cache_read_tokens, cache_write_tokens
+          FROM token_usage WHERE ${where} AND (turn_id IS NULL OR step_index IS NULL))
+       ) dedup GROUP BY model ORDER BY total DESC`, args);
+        return res.rows.map((r) => ({
+            key: r.key,
+            input: r.i,
+            output: r.o,
+            cacheRead: r.cr,
+            cacheWrite: r.cw,
+            total: r.total,
+            requests: r.reqs,
+        }));
+    }
+    catch (err) {
+        log.error({ err, rootId }, "readRunUsageByModel failed — returning empty");
+        return [];
+    }
+}
+export async function readRootDayTotal(rootId, dayIso) {
+    const p = getPool();
+    if (!p)
+        return 0;
+    try {
+        await ensureDdl(p);
+        const res = await p.query(`SELECT COALESCE(SUM(input_tokens + output_tokens),0)::bigint total
+       FROM token_usage
+       WHERE root_id = $1
+         AND at >= $2::timestamptz AND at < $2::timestamptz + interval '24 hours'`, [rootId, `${dayIso}T00:00:00Z`]);
+        return Number(res.rows[0]?.total ?? 0);
+    }
+    catch (err) {
+        log.warn({ err: String(err), rootId }, "root day total read failed — returning 0");
+        return 0;
+    }
+}
+export async function readUsageLogsPage(q) {
+    const p = getPool();
+    if (!p)
+        return { rows: [], nextCursor: null, hasMore: false };
+    try {
+        await ensureDdl(p);
+        const conds = [];
+        const args = [];
+        if (q.before != null) {
+            args.push(q.before);
+            conds.push(`id < $${args.length}`);
+        }
+        if (q.flow) {
+            args.push(q.flow);
+            conds.push(`flow = $${args.length}`);
+        }
+        if (q.model) {
+            args.push(q.model);
+            conds.push(`model = $${args.length}`);
+        }
+        if (q.instance) {
+            args.push(q.instance);
+            conds.push(`instance = $${args.length}`);
+        }
+        const where = conds.length ? `WHERE ${conds.join(" AND ")}` : "";
+        args.push(q.limit + 1);
+        const res = await p.query(`SELECT id, session_id, root_id, flow, model, at,
+              input_tokens, output_tokens, cache_read_tokens, cache_write_tokens, instance
+       FROM token_usage ${where} ORDER BY id DESC LIMIT $${args.length}`, args);
+        const all = res.rows.map((r) => ({
+            id: Number(r.id),
+            sessionId: r.session_id,
+            rootId: r.root_id,
+            flow: r.flow,
+            model: r.model,
+            at: new Date(r.at).toISOString(),
+            input: r.input_tokens,
+            output: r.output_tokens,
+            cacheRead: r.cache_read_tokens,
+            cacheWrite: r.cache_write_tokens,
+            instance: r.instance,
+        }));
+        const hasMore = all.length > q.limit;
+        const rows = hasMore ? all.slice(0, q.limit) : all;
+        return { rows, nextCursor: hasMore ? (rows[rows.length - 1]?.id ?? null) : null, hasMore };
+    }
+    catch (err) {
+        log.error({ err }, "usage logs read failed — returning empty");
+        return { rows: [], nextCursor: null, hasMore: false };
+    }
+}
+//# sourceMappingURL=token-usage-pg.js.map

package/dist/lib/budgets/token-usage-pg.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"token-usage-pg.js","sourceRoot":"","sources":["../../../src/lib/budgets/token-usage-pg.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,IAAI,CAAC;AAC1B,OAAO,EAAE,MAAM,EAAE,MAAM,4BAA4B,CAAC;AAEpD,MAAM,GAAG,GAAG,MAAM,CAAC,KAAK,CAAC,EAAE,MAAM,EAAE,aAAa,EAAE,CAAC,CAAC;AAMpD,IAAI,IAAsB,CAAC;AAC3B,IAAI,OAAkC,CAAC;AAEvC,SAAS,OAAO;IAKd,MAAM,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC,wBAAwB,IAAI,OAAO,CAAC,GAAG,CAAC,qBAAqB,CAAC;IACtF,IAAI,CAAC,GAAG;QAAE,OAAO,SAAS,CAAC;IAC3B,IAAI,CAAC,IAAI,EAAE,CAAC;QAGV,IAAI,GAAG,IAAI,IAAI,CAAC;YACd,gBAAgB,EAAE,GAAG;YACrB,GAAG,EAAE,CAAC;YACN,uBAAuB,EAAE,KAAK;YAC9B,aAAa,EAAE,MAAM;YACrB,iBAAiB,EAAE,MAAM;SAC1B,CAAC,CAAC;IACL,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,MAAM,GAAG,GAAG;;;;;;;;;;;;;;;;;;+EAkBmE,CAAC;AAEhF,SAAS,SAAS,CAAC,CAAO;IAIxB,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,OAAO,GAAG,CAAC;aACR,KAAK,CAAC,GAAG,CAAC;aACV,IAAI,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC;aACrB,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;YACb,OAAO,GAAG,SAAS,CAAC;YACpB,MAAM,GAAG,CAAC;QACZ,CAAC,CAAC,CAAC;IACP,CAAC;IACD,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,MAAM,GAAG,GAAG;;;kDAGsC,CAAC;AAuBnD,MAAM,cAAc,GAAG,IAAI,GAAG,EAAiB,CAAC;AAChD,MAAM,UAAU,eAAe,CAAC,EAAiB;IAC/C,cAAc,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IACvB,OAAO,GAAG,EAAE,CAAC,cAAc,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;AACzC,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,WAAW,CAAC,CAAW;IAC3C,KAAK,MAAM,EAAE,IAAI,cAAc,EAAE,CAAC;QAChC,IAAI,CAAC;YACH,EAAE,CAAC,CAAC,CAAC,CAAC;QACR,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,GAAG,CAAC,IAAI,CAAC,EAAE,GAAG,EAAE,MAAM,CAAC,GAAG,CAAC,EAAE,EAAE,gCAAgC,CAAC,CAAC;QACnE,CAAC;IACH,CAAC;IACD,MAAM,CAAC,GAAG,OAAO,EAAE,CAAC;IACpB,IAAI,CAAC,CAAC;QAAE,OAAO;IACf,IAAI,CAAC;QACH,MAAM,SAAS,CAAC,CAAC,CAAC,CAAC;QACnB,MAAM,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE;YACjB,CAAC,CAAC,SAAS;YACX,CAAC,CAAC,MAAM;YACR,CAAC,CAAC,IAAI;YACN,CAAC,CAAC,KAAK;YACP,CAAC,CAAC,MAAM;YACR,CAAC,CAAC,SAAS;YACX,CAAC,CAAC,EAAE;YACJ,CAAC,CAAC,KAAK;YACP,CAAC,CAAC,MAAM;YACR,CAAC,CAAC,SAAS;YACX,CAAC,CAAC,UAAU;YACZ,CAAC,CAAC,QAAQ;SACX,CAAC,CAAC;IACL,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,GAAG,CAAC,KAAK,CAAC,EAAE,GAAG,EAAE,EAAE,sCAAsC,CAAC,CAAC;IAC7D,CAAC;AACH,CAAC;AAiDD,MAAM,SAAS,GAAmB;IAChC,MAAM,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE;IACtF,MAAM,EAAE,EAAE;IACV,aAAa,EAAE,EAAE;IACjB,OAAO,EAAE,EAAE;IACX,MAAM,EAAE,EAAE;IACV,UAAU,EAAE,EAAE;IACd,MAAM,EAAE,EAAE;CACX,CAAC;AAEF,MAAM,CAAC,KAAK,UAAU,kBAAkB,CAAC,CAAiB;IACxD,MAAM,CAAC,GAAG,OAAO,EAAE,CAAC;IACpB,IAAI,CAAC,CAAC;QAAE,OAAO,SAAS,CAAC;IACzB,IAAI,CAAC;QACH,MAAM,SAAS,CAAC,CAAC,CAAC,CAAC;QACnB,MAAM,IAAI,GAAc,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;QACvC,IAAI,KAAK,GAAG,gDAAgD,CAAC;QAC7D,IAAI,CAAC,CAAC,QAAQ,EAAE,CAAC;YACf,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;YACtB,KAAK,IAAI,oBAAoB,IAAI,CAAC,MAAM,EAAE,CAAC;QAC7C,CAAC;QACD,MAAM,KAAK,GAAG,CAAC,CAAC,MAAM,KAAK,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC;QAEnD,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,aAAa,EAAE,OAAO,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;YAC7F,CAAC,CAAC,KAAK,CACL;;;kCAG0B,KAAK,EAAE,EACjC,IAAI,CACL;YACD,CAAC,CAAC,KAAK,CACL,sBAAsB,KAAK;;;kCAGD,KAAK,wBAAwB,EACvD,IAAI,CACL;YACD,CAAC,CAAC,KAAK,CACL,sBAAsB,KAAK;;kCAED,KAAK,+BAA+B,EAC9D,IAAI,CACL;YACD,CAAC,CAAC,KAAK,CACL;;kCAE0B,KAAK,qCAAqC,EACpE,IAAI,CACL;YACD,CAAC,CAAC,KAAK,CACL;;kCAE0B,KAAK,oCAAoC,EACnE,IAAI,CACL;YACD,CAAC,CAAC,KAAK,CACL;;kCAE0B,KAAK,wCAAwC,EACvE,IAAI,CACL;YACD,CAAC,CAAC,KAAK,CACL;;kCAE0B,KAAK,mBAAmB,EAClD,IAAI,CACL;SACF,CAAC,CAAC;QAEH,MAAM,CAAC,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QAC/B,OAAO;YACL,MAAM,EAAE;gBACN,KAAK,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;gBACf,MAAM,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;gBAChB,SAAS,EAAE,CAAC,CAAC,EAAE,IAAI,CAAC;gBACpB,UAAU,EAAE,CAAC,CAAC,EAAE,IAAI,CAAC;gBACrB,QAAQ,EAAE,CAAC,CAAC,IAAI,IAAI,CAAC;gBACrB,QAAQ,EAAE,CAAC,CAAC,IAAI,IAAI,CAAC;aACtB;YACD,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;gBAC9B,EAAE,EAAE,IAAI,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE;gBAC/B,KAAK,EAAE,CAAC,CAAC,CAAC;gBACV,MAAM,EAAE,CAAC,CAAC,CAAC;gBACX,SAAS,EAAE,CAAC,CAAC,EAAE;gBACf,UAAU,EAAE,CAAC,CAAC,EAAE;aACjB,CAAC,CAAC;YACH,aAAa,EAAE,aAAa,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;gBAC5C,EAAE,EAAE,IAAI,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE;gBAC/B,KAAK,EAAE,CAAC,CAAC,KAAK;gBACd,KAAK,EAAE,CAAC,CAAC,KAAK;aACf,CAAC,CAAC;YACH,OAAO,EAAE,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;gBAChC,GAAG,EAAE,CAAC,CAAC,GAAG;gBACV,KAAK,EAAE,CAAC,CAAC,CAAC;gBACV,MAAM,EAAE,CAAC,CAAC,CAAC;gBACX,KAAK,EAAE,CAAC,CAAC,KAAK;gBACd,QAAQ,EAAE,CAAC,CAAC,IAAI;aACjB,CAAC,CAAC;YACH,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;gBAC9B,GAAG,EAAE,CAAC,CAAC,GAAG;gBACV,KAAK,EAAE,CAAC,CAAC,CAAC;gBACV,MAAM,EAAE,CAAC,CAAC,CAAC;gBACX,KAAK,EAAE,CAAC,CAAC,KAAK;gBACd,QAAQ,EAAE,CAAC,CAAC,IAAI;aACjB,CAAC,CAAC;YACH,UAAU,EAAE,UAAU,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;gBACtC,GAAG,EAAE,CAAC,CAAC,GAAG;gBACV,KAAK,EAAE,CAAC,CAAC,CAAC;gBACV,MAAM,EAAE,CAAC,CAAC,CAAC;gBACX,KAAK,EAAE,CAAC,CAAC,KAAK;gBACd,QAAQ,EAAE,CAAC,CAAC,IAAI;aACjB,CAAC,CAAC;YACH,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;gBAC9B,MAAM,EAAE,CAAC,CAAC,OAAO;gBACjB,KAAK,EAAE,CAAC,CAAC,CAAC;gBACV,MAAM,EAAE,CAAC,CAAC,CAAC;gBACX,KAAK,EAAE,CAAC,CAAC,KAAK;gBACd,QAAQ,EAAE,CAAC,CAAC,IAAI;aACjB,CAAC,CAAC;SACJ,CAAC;IACJ,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,GAAG,CAAC,KAAK,CAAC,EAAE,GAAG,EAAE,EAAE,+CAA+C,CAAC,CAAC;QACpE,OAAO,SAAS,CAAC;IACnB,CAAC;AACH,CAAC;AAOD,MAAM,CAAC,KAAK,UAAU,mBAAmB,CACvC,MAAc,EACd,IAAa;IAEb,MAAM,CAAC,GAAG,OAAO,EAAE,CAAC;IACpB,IAAI,CAAC,CAAC;QAAE,OAAO,EAAE,CAAC;IAClB,IAAI,CAAC;QACH,MAAM,SAAS,CAAC,CAAC,CAAC,CAAC;QACnB,MAAM,IAAI,GAAc,CAAC,MAAM,CAAC,CAAC;QACjC,IAAI,KAAK,GAAG,cAAc,CAAC;QAC3B,IAAI,IAAI,EAAE,CAAC;YACT,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAChB,KAAK,IAAI,gBAAgB,CAAC;QAC5B,CAAC;QAOD,MAAM,GAAG,GAAG,MAAM,CAAC,CAAC,KAAK,CACvB;;;;;;mCAM6B,KAAK;;;;mCAIL,KAAK;kDACU,EAC5C,IAAI,CACL,CAAC;QACF,OAAO,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YAC1B,GAAG,EAAE,CAAC,CAAC,GAAG;YACV,KAAK,EAAE,CAAC,CAAC,CAAC;YACV,MAAM,EAAE,CAAC,CAAC,CAAC;YACX,SAAS,EAAE,CAAC,CAAC,EAAE;YACf,UAAU,EAAE,CAAC,CAAC,EAAE;YAChB,KAAK,EAAE,CAAC,CAAC,KAAK;YACd,QAAQ,EAAE,CAAC,CAAC,IAAI;SACjB,CAAC,CAAC,CAAC;IACN,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,GAAG,CAAC,KAAK,CAAC,EAAE,GAAG,EAAE,MAAM,EAAE,EAAE,8CAA8C,CAAC,CAAC;QAC3E,OAAO,EAAE,CAAC;IACZ,CAAC;AACH,CAAC;AAMD,MAAM,CAAC,KAAK,UAAU,gBAAgB,CAAC,MAAc,EAAE,MAAc;IACnE,MAAM,CAAC,GAAG,OAAO,EAAE,CAAC;IACpB,IAAI,CAAC,CAAC;QAAE,OAAO,CAAC,CAAC;IACjB,IAAI,CAAC;QACH,MAAM,SAAS,CAAC,CAAC,CAAC,CAAC;QAOnB,MAAM,GAAG,GAAG,MAAM,CAAC,CAAC,KAAK,CACvB;;;kFAG4E,EAC5E,CAAC,MAAM,EAAE,GAAG,MAAM,YAAY,CAAC,CAChC,CAAC;QACF,OAAO,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,KAAK,IAAI,CAAC,CAAC,CAAC;IACzC,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,GAAG,CAAC,IAAI,CAAC,EAAE,GAAG,EAAE,MAAM,CAAC,GAAG,CAAC,EAAE,MAAM,EAAE,EAAE,0CAA0C,CAAC,CAAC;QACnF,OAAO,CAAC,CAAC;IACX,CAAC;AACH,CAAC;AA+BD,MAAM,CAAC,KAAK,UAAU,iBAAiB,CAAC,CAAY;IAClD,MAAM,CAAC,GAAG,OAAO,EAAE,CAAC;IACpB,IAAI,CAAC,CAAC;QAAE,OAAO,EAAE,IAAI,EAAE,EAAE,EAAE,UAAU,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;IAC9D,IAAI,CAAC;QACH,MAAM,SAAS,CAAC,CAAC,CAAC,CAAC;QACnB,MAAM,KAAK,GAAa,EAAE,CAAC;QAC3B,MAAM,IAAI,GAAc,EAAE,CAAC;QAC3B,IAAI,CAAC,CAAC,MAAM,IAAI,IAAI,EAAE,CAAC;YACrB,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC;YACpB,KAAK,CAAC,IAAI,CAAC,SAAS,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;QACrC,CAAC;QACD,IAAI,CAAC,CAAC,IAAI,EAAE,CAAC;YACX,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;YAClB,KAAK,CAAC,IAAI,CAAC,WAAW,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;QACvC,CAAC;QACD,IAAI,CAAC,CAAC,KAAK,EAAE,CAAC;YACZ,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;YACnB,KAAK,CAAC,IAAI,CAAC,YAAY,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;QACxC,CAAC;QACD,IAAI,CAAC,CAAC,QAAQ,EAAE,CAAC;YACf,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;YACtB,KAAK,CAAC,IAAI,CAAC,eAAe,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;QAC3C,CAAC;QACD,MAAM,KAAK,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACjE,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC;QACvB,MAAM,GAAG,GAAG,MAAM,CAAC,CAAC,KAAK,CACvB;;0BAEoB,KAAK,4BAA4B,IAAI,CAAC,MAAM,EAAE,EAClE,IAAI,CACL,CAAC;QACF,MAAM,GAAG,GAAkB,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YAC9C,EAAE,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC;YAChB,SAAS,EAAE,CAAC,CAAC,UAAU;YACvB,MAAM,EAAE,CAAC,CAAC,OAAO;YACjB,IAAI,EAAE,CAAC,CAAC,IAAI;YACZ,KAAK,EAAE,CAAC,CAAC,KAAK;YACd,EAAE,EAAE,IAAI,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,WAAW,EAAE;YAChC,KAAK,EAAE,CAAC,CAAC,YAAY;YACrB,MAAM,EAAE,CAAC,CAAC,aAAa;YACvB,SAAS,EAAE,CAAC,CAAC,iBAAiB;YAC9B,UAAU,EAAE,CAAC,CAAC,kBAAkB;YAChC,QAAQ,EAAE,CAAC,CAAC,QAAQ;SACrB,CAAC,CAAC,CAAC;QACJ,MAAM,OAAO,GAAG,GAAG,CAAC,MAAM,GAAG,CAAC,CAAC,KAAK,CAAC;QACrC,MAAM,IAAI,GAAG,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC;QACnD,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,EAAE,EAAE,IAAI,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,OAAO,EAAE,CAAC;IAC7F,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,GAAG,CAAC,KAAK,CAAC,EAAE,GAAG,EAAE,EAAE,0CAA0C,CAAC,CAAC;QAC/D,OAAO,EAAE,IAAI,EAAE,EAAE,EAAE,UAAU,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;IACxD,CAAC;AACH,CAAC"}

package/dist/lib/budgets/usage.d.ts

@@ -0,0 +1,103 @@
+import { type CostKind } from "../models/model-catalog.js";
+import { type AggregateQuery, type LogsQuery, type UsageBreakdownRow, type UsageRootRow } from "./token-usage-pg.js";
+export declare class UsageQueryError extends Error {
+}
+export declare function parseUsageQuery(rawUrl: string, now: Date): AggregateQuery;
+export declare function parseLogsQuery(rawUrl: string): LogsQuery;
+export interface SourceBreakdownRow {
+    key: string;
+    input: number;
+    output: number;
+    total: number;
+    requests: number;
+}
+export declare function enrichBySource(byRoot: UsageRootRow[], index: Map<string, string>): SourceBreakdownRow[];
+export interface PricedBreakdown extends UsageBreakdownRow {
+    costUsd: number;
+    costKind: CostKind;
+}
+export declare function priceModels(rows: UsageBreakdownRow[], env?: NodeJS.ProcessEnv): {
+    rows: PricedBreakdown[];
+    realUsd: number;
+    notionalUsd: number;
+};
+export declare function buildUsage(rawUrl: string, now: Date): Promise<{
+    byModel: PricedBreakdown[];
+    byFlow: (UsageBreakdownRow & {
+        costUsd: number;
+        costKind: "mixed";
+    })[];
+    byInstance: (UsageBreakdownRow & {
+        costUsd: number;
+        costKind: "mixed";
+    })[];
+    bySource: (SourceBreakdownRow & {
+        costUsd: number;
+        costKind: "mixed";
+    })[];
+    cost: {
+        realUsd: number;
+        notionalUsd: number;
+    };
+    generatedAt: string;
+    totals: {
+        input: number;
+        output: number;
+        cacheRead: number;
+        cacheWrite: number;
+        requests: number;
+        sessions: number;
+    };
+    series: {
+        at: string;
+        input: number;
+        output: number;
+        cacheRead: number;
+        cacheWrite: number;
+    }[];
+    seriesByModel: {
+        at: string;
+        model: string;
+        total: number;
+    }[];
+    range: AggregateQuery;
+}>;
+export declare function buildRunUsage(rootId: string, rows: UsageBreakdownRow[], now: Date, env?: NodeJS.ProcessEnv): {
+    rootId: string;
+    byModel: {
+        model: string;
+        input: number;
+        output: number;
+        cacheRead: number;
+        cacheWrite: number;
+        calls: number;
+    }[];
+    total: {
+        input: number;
+        output: number;
+        cacheRead: number;
+        cacheWrite: number;
+        calls: number;
+    };
+    generatedAt: string;
+};
+export declare function buildUsageLogs(rawUrl: string, now: Date): Promise<{
+    rows: {
+        source: string;
+        id: number;
+        sessionId: string;
+        rootId: string;
+        flow: string | null;
+        model: string;
+        at: string;
+        input: number;
+        output: number;
+        cacheRead: number;
+        cacheWrite: number;
+        instance: string;
+    }[];
+    nextCursor: number | null;
+    hasMore: boolean;
+    generatedAt: string;
+}>;
+//# sourceMappingURL=usage.d.ts.map

package/dist/lib/budgets/usage.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"usage.d.ts","sourceRoot":"","sources":["../../../src/lib/budgets/usage.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,QAAQ,EAAyC,MAAM,4BAA4B,CAAC;AAElG,OAAO,EACL,KAAK,cAAc,EACnB,KAAK,SAAS,EAGd,KAAK,iBAAiB,EACtB,KAAK,YAAY,EAClB,MAAM,qBAAqB,CAAC;AAM7B,qBAAa,eAAgB,SAAQ,KAAK;CAAG;AAQ7C,wBAAgB,eAAe,CAAC,MAAM,EAAE,MAAM,EAAE,GAAG,EAAE,IAAI,GAAG,cAAc,CAezE;AAED,wBAAgB,cAAc,CAAC,MAAM,EAAE,MAAM,GAAG,SAAS,CAiBxD;AAED,MAAM,WAAW,kBAAkB;IACjC,GAAG,EAAE,MAAM,CAAC;IACZ,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,EAAE,MAAM,CAAC;CAClB;AAKD,wBAAgB,cAAc,CAC5B,MAAM,EAAE,YAAY,EAAE,EACtB,KAAK,EAAE,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,GACzB,kBAAkB,EAAE,CAYtB;AAQD,MAAM,WAAW,eAAgB,SAAQ,iBAAiB;IACxD,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,EAAE,QAAQ,CAAC;CACpB;AAgBD,wBAAgB,WAAW,CACzB,IAAI,EAAE,iBAAiB,EAAE,EACzB,GAAG,GAAE,MAAM,CAAC,UAAwB,GACnC;IAAE,IAAI,EAAE,eAAe,EAAE,CAAC;IAAC,OAAO,EAAE,MAAM,CAAC;IAAC,WAAW,EAAE,MAAM,CAAA;CAAE,CAenE;AAED,wBAAsB,UAAU,CAAC,MAAM,EAAE,MAAM,EAAE,GAAG,EAAE,IAAI;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAuBzD;AAOD,wBAAgB,aAAa,CAC3B,MAAM,EAAE,MAAM,EACd,IAAI,EAAE,iBAAiB,EAAE,EACzB,GAAG,EAAE,IAAI,EACT,GAAG,GAAE,MAAM,CAAC,UAAwB;;;eAKzB,MAAM;eACN,MAAM;gBACL,MAAM;mBACH,MAAM;oBACL,MAAM;eACX,MAAM;;;;;;;;;;EAgClB;AAED,wBAAsB,cAAc,CAAC,MAAM,EAAE,MAAM,EAAE,GAAG,EAAE,IAAI;;;;;;;;;;;;;;;;;;GAW7D"}

package/dist/lib/budgets/usage.js

@@ -0,0 +1,151 @@
+import { costKindFor, isCatalogModel, priceFor } from "../models/model-catalog.js";
+import { readSessionsIndexed } from "../observability/recorder.js";
+import { readUsageAggregate, readUsageLogsPage, } from "./token-usage-pg.js";
+const WINDOW_MS = 7 * 24 * 3600 * 1000;
+export class UsageQueryError extends Error {
+}
+function parseDate(value, field) {
+    const d = new Date(value);
+    if (Number.isNaN(d.getTime()))
+        throw new UsageQueryError(`invalid ${field}: ${value}`);
+    return d;
+}
+export function parseUsageQuery(rawUrl, now) {
+    const u = new URL(rawUrl, "http://localhost");
+    const bucket = u.searchParams.get("bucket") === "hour" ? "hour" : "day";
+    const fromParam = u.searchParams.get("from");
+    const toParam = u.searchParams.get("to");
+    const to = toParam ? parseDate(toParam, "to") : now;
+    const from = fromParam ? parseDate(fromParam, "from") : new Date(to.getTime() - WINDOW_MS);
+    return {
+        from: from.toISOString(),
+        to: to.toISOString(),
+        bucket,
+        instance: u.searchParams.get("instance") || undefined,
+    };
+}
+export function parseLogsQuery(rawUrl) {
+    const u = new URL(rawUrl, "http://localhost");
+    let limit = 50;
+    const l = u.searchParams.get("limit");
+    if (l != null) {
+        const n = Number(l);
+        if (Number.isFinite(n) && n > 0)
+            limit = Math.min(n, 200);
+    }
+    const b = u.searchParams.get("before");
+    const before = b != null && Number.isFinite(Number(b)) ? Number(b) : undefined;
+    return {
+        limit,
+        before,
+        flow: u.searchParams.get("flow") ?? undefined,
+        model: u.searchParams.get("model") ?? undefined,
+        instance: u.searchParams.get("instance") || undefined,
+    };
+}
+export function enrichBySource(byRoot, index) {
+    const acc = new Map();
+    for (const r of byRoot) {
+        const key = index.get(r.rootId) ?? "unknown";
+        const cur = acc.get(key) ?? { key, input: 0, output: 0, total: 0, requests: 0 };
+        cur.input += r.input;
+        cur.output += r.output;
+        cur.total += r.total;
+        cur.requests += r.requests;
+        acc.set(key, cur);
+    }
+    return [...acc.values()].sort((a, b) => b.total - a.total);
+}
+function sourceIndex() {
+    const m = new Map();
+    for (const row of readSessionsIndexed())
+        m.set(row.id, row.source.kind);
+    return m;
+}
+function catalogId(recorded, env) {
+    if (isCatalogModel(recorded, env))
+        return recorded;
+    const i = recorded.indexOf("/");
+    return i >= 0 ? recorded.slice(i + 1) : recorded;
+}
+export function priceModels(rows, env = process.env) {
+    let realUsd = 0;
+    let notionalUsd = 0;
+    const priced = rows.map((row) => {
+        const id = catalogId(row.key ?? "", env);
+        const price = priceFor(id, env);
+        const costKind = costKindFor(id, env);
+        const costUsd = price
+            ? (row.input * price.inputPerMTok + row.output * price.outputPerMTok) / 1_000_000
+            : 0;
+        if (costKind === "real")
+            realUsd += costUsd;
+        else
+            notionalUsd += costUsd;
+        return { ...row, costUsd, costKind };
+    });
+    return { rows: priced, realUsd, notionalUsd };
+}
+export async function buildUsage(rawUrl, now) {
+    const q = parseUsageQuery(rawUrl, now);
+    const agg = await readUsageAggregate(q);
+    const { byRoot, byModel, byFlow, byInstance, ...rest } = agg;
+    const priced = priceModels(byModel);
+    const mixed = (r) => ({
+        ...r,
+        costUsd: 0,
+        costKind: "mixed",
+    });
+    return {
+        range: q,
+        ...rest,
+        byModel: priced.rows,
+        byFlow: byFlow.map(mixed),
+        byInstance: byInstance.map(mixed),
+        bySource: enrichBySource(byRoot, sourceIndex()).map(mixed),
+        cost: { realUsd: priced.realUsd, notionalUsd: priced.notionalUsd },
+        generatedAt: now.toISOString(),
+    };
+}
+export function buildRunUsage(rootId, rows, now, env = process.env) {
+    const acc = new Map();
+    for (const r of rows) {
+        const model = catalogId(r.key ?? "unknown", env) || "unknown";
+        const cur = acc.get(model) ?? {
+            model,
+            input: 0,
+            output: 0,
+            cacheRead: 0,
+            cacheWrite: 0,
+            calls: 0,
+        };
+        cur.input += r.input;
+        cur.output += r.output;
+        cur.cacheRead += r.cacheRead ?? 0;
+        cur.cacheWrite += r.cacheWrite ?? 0;
+        cur.calls += r.requests;
+        acc.set(model, cur);
+    }
+    const byModel = [...acc.values()].sort((a, b) => b.input + b.output - (a.input + a.output));
+    const total = byModel.reduce((t, m) => ({
+        input: t.input + m.input,
+        output: t.output + m.output,
+        cacheRead: t.cacheRead + m.cacheRead,
+        cacheWrite: t.cacheWrite + m.cacheWrite,
+        calls: t.calls + m.calls,
+    }), { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, calls: 0 });
+    return { rootId, byModel, total, generatedAt: now.toISOString() };
+}
+export async function buildUsageLogs(rawUrl, now) {
+    const q = parseLogsQuery(rawUrl);
+    const page = await readUsageLogsPage(q);
+    const idx = sourceIndex();
+    const rows = page.rows.map((r) => ({ ...r, source: idx.get(r.rootId) ?? "unknown" }));
+    return {
+        rows,
+        nextCursor: page.nextCursor,
+        hasMore: page.hasMore,
+        generatedAt: now.toISOString(),
+    };
+}
+//# sourceMappingURL=usage.js.map

package/dist/lib/budgets/usage.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"usage.js","sourceRoot":"","sources":["../../../src/lib/budgets/usage.ts"],"names":[],"mappings":"AAAA,OAAO,EAAiB,WAAW,EAAE,cAAc,EAAE,QAAQ,EAAE,MAAM,4BAA4B,CAAC;AAClG,OAAO,EAAE,mBAAmB,EAAE,MAAM,8BAA8B,CAAC;AACnE,OAAO,EAGL,kBAAkB,EAClB,iBAAiB,GAGlB,MAAM,qBAAqB,CAAC;AAE7B,MAAM,SAAS,GAAG,CAAC,GAAG,EAAE,GAAG,IAAI,GAAG,IAAI,CAAC;AAIvC,MAAM,OAAO,eAAgB,SAAQ,KAAK;CAAG;AAE7C,SAAS,SAAS,CAAC,KAAa,EAAE,KAAa;IAC7C,MAAM,CAAC,GAAG,IAAI,IAAI,CAAC,KAAK,CAAC,CAAC;IAC1B,IAAI,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC;QAAE,MAAM,IAAI,eAAe,CAAC,WAAW,KAAK,KAAK,KAAK,EAAE,CAAC,CAAC;IACvF,OAAO,CAAC,CAAC;AACX,CAAC;AAED,MAAM,UAAU,eAAe,CAAC,MAAc,EAAE,GAAS;IACvD,MAAM,CAAC,GAAG,IAAI,GAAG,CAAC,MAAM,EAAE,kBAAkB,CAAC,CAAC;IAC9C,MAAM,MAAM,GAAG,CAAC,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,KAAK,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC;IACxE,MAAM,SAAS,GAAG,CAAC,CAAC,YAAY,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IAC7C,MAAM,OAAO,GAAG,CAAC,CAAC,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IACzC,MAAM,EAAE,GAAG,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC;IACpD,MAAM,IAAI,GAAG,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,SAAS,CAAC,CAAC;IAC3F,OAAO;QACL,IAAI,EAAE,IAAI,CAAC,WAAW,EAAE;QACxB,EAAE,EAAE,EAAE,CAAC,WAAW,EAAE;QACpB,MAAM;QAGN,QAAQ,EAAE,CAAC,CAAC,YAAY,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,SAAS;KACtD,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,cAAc,CAAC,MAAc;IAC3C,MAAM,CAAC,GAAG,IAAI,GAAG,CAAC,MAAM,EAAE,kBAAkB,CAAC,CAAC;IAC9C,IAAI,KAAK,GAAG,EAAE,CAAC;IACf,MAAM,CAAC,GAAG,CAAC,CAAC,YAAY,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;IACtC,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC;QACd,MAAM,CAAC,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC;QACpB,IAAI,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC;YAAE,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;IAC5D,CAAC;IACD,MAAM,CAAC,GAAG,CAAC,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IACvC,MAAM,MAAM,GAAG,CAAC,IAAI,IAAI,IAAI,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IAC/E,OAAO;QACL,KAAK;QACL,MAAM;QACN,IAAI,EAAE,CAAC,CAAC,YAAY,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,SAAS;QAC7C,KAAK,EAAE,CAAC,CAAC,YAAY,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,SAAS;QAC/C,QAAQ,EAAE,CAAC,CAAC,YAAY,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,SAAS;KACtD,CAAC;AACJ,CAAC;AAaD,MAAM,UAAU,cAAc,CAC5B,MAAsB,EACtB,KAA0B;IAE1B,MAAM,GAAG,GAAG,IAAI,GAAG,EAA8B,CAAC;IAClD,KAAK,MAAM,CAAC,IAAI,MAAM,EAAE,CAAC;QACvB,MAAM,GAAG,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,SAAS,CAAC;QAC7C,MAAM,GAAG,GAAG,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC;QAChF,GAAG,CAAC,KAAK,IAAI,CAAC,CAAC,KAAK,CAAC;QACrB,GAAG,CAAC,MAAM,IAAI,CAAC,CAAC,MAAM,CAAC;QACvB,GAAG,CAAC,KAAK,IAAI,CAAC,CAAC,KAAK,CAAC;QACrB,GAAG,CAAC,QAAQ,IAAI,CAAC,CAAC,QAAQ,CAAC;QAC3B,GAAG,CAAC,GAAG,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;IACpB,CAAC;IACD,OAAO,CAAC,GAAG,GAAG,CAAC,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC;AAC7D,CAAC;AAED,SAAS,WAAW;IAClB,MAAM,CAAC,GAAG,IAAI,GAAG,EAAkB,CAAC;IACpC,KAAK,MAAM,GAAG,IAAI,mBAAmB,EAAE;QAAE,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,EAAE,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;IACxE,OAAO,CAAC,CAAC;AACX,CAAC;AAYD,SAAS,SAAS,CAAC,QAAgB,EAAE,GAAsB;IACzD,IAAI,cAAc,CAAC,QAAQ,EAAE,GAAG,CAAC;QAAE,OAAO,QAAQ,CAAC;IACnD,MAAM,CAAC,GAAG,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;IAChC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC;AACnD,CAAC;AAKD,MAAM,UAAU,WAAW,CACzB,IAAyB,EACzB,MAAyB,OAAO,CAAC,GAAG;IAEpC,IAAI,OAAO,GAAG,CAAC,CAAC;IAChB,IAAI,WAAW,GAAG,CAAC,CAAC;IACpB,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;QAC9B,MAAM,EAAE,GAAG,SAAS,CAAC,GAAG,CAAC,GAAG,IAAI,EAAE,EAAE,GAAG,CAAC,CAAC;QACzC,MAAM,KAAK,GAAG,QAAQ,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC;QAChC,MAAM,QAAQ,GAAG,WAAW,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC;QACtC,MAAM,OAAO,GAAG,KAAK;YACnB,CAAC,CAAC,CAAC,GAAG,CAAC,KAAK,GAAG,KAAK,CAAC,YAAY,GAAG,GAAG,CAAC,MAAM,GAAG,KAAK,CAAC,aAAa,CAAC,GAAG,SAAS;YACjF,CAAC,CAAC,CAAC,CAAC;QACN,IAAI,QAAQ,KAAK,MAAM;YAAE,OAAO,IAAI,OAAO,CAAC;;YACvC,WAAW,IAAI,OAAO,CAAC;QAC5B,OAAO,EAAE,GAAG,GAAG,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC;IACvC,CAAC,CAAC,CAAC;IACH,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,WAAW,EAAE,CAAC;AAChD,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,UAAU,CAAC,MAAc,EAAE,GAAS;IACxD,MAAM,CAAC,GAAG,eAAe,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IACvC,MAAM,GAAG,GAAG,MAAM,kBAAkB,CAAC,CAAC,CAAC,CAAC;IACxC,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,UAAU,EAAE,GAAG,IAAI,EAAE,GAAG,GAAG,CAAC;IAC7D,MAAM,MAAM,GAAG,WAAW,CAAC,OAAO,CAAC,CAAC;IAIpC,MAAM,KAAK,GAAG,CAA8B,CAAI,EAAE,EAAE,CAAC,CAAC;QACpD,GAAG,CAAC;QACJ,OAAO,EAAE,CAAC;QACV,QAAQ,EAAE,OAAgB;KAC3B,CAAC,CAAC;IACH,OAAO;QACL,KAAK,EAAE,CAAC;QACR,GAAG,IAAI;QACP,OAAO,EAAE,MAAM,CAAC,IAAI;QACpB,MAAM,EAAE,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC;QACzB,UAAU,EAAE,UAAU,CAAC,GAAG,CAAC,KAAK,CAAC;QACjC,QAAQ,EAAE,cAAc,CAAC,MAAM,EAAE,WAAW,EAAE,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC;QAC1D,IAAI,EAAE,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE,WAAW,EAAE,MAAM,CAAC,WAAW,EAAE;QAClE,WAAW,EAAE,GAAG,CAAC,WAAW,EAAE;KAC/B,CAAC;AACJ,CAAC;AAOD,MAAM,UAAU,aAAa,CAC3B,MAAc,EACd,IAAyB,EACzB,GAAS,EACT,MAAyB,OAAO,CAAC,GAAG;IAEpC,MAAM,GAAG,GAAG,IAAI,GAAG,EAUhB,CAAC;IACJ,KAAK,MAAM,CAAC,IAAI,IAAI,EAAE,CAAC;QACrB,MAAM,KAAK,GAAG,SAAS,CAAC,CAAC,CAAC,GAAG,IAAI,SAAS,EAAE,GAAG,CAAC,IAAI,SAAS,CAAC;QAC9D,MAAM,GAAG,GAAG,GAAG,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI;YAC5B,KAAK;YACL,KAAK,EAAE,CAAC;YACR,MAAM,EAAE,CAAC;YACT,SAAS,EAAE,CAAC;YACZ,UAAU,EAAE,CAAC;YACb,KAAK,EAAE,CAAC;SACT,CAAC;QACF,GAAG,CAAC,KAAK,IAAI,CAAC,CAAC,KAAK,CAAC;QACrB,GAAG,CAAC,MAAM,IAAI,CAAC,CAAC,MAAM,CAAC;QACvB,GAAG,CAAC,SAAS,IAAI,CAAC,CAAC,SAAS,IAAI,CAAC,CAAC;QAClC,GAAG,CAAC,UAAU,IAAI,CAAC,CAAC,UAAU,IAAI,CAAC,CAAC;QACpC,GAAG,CAAC,KAAK,IAAI,CAAC,CAAC,QAAQ,CAAC;QACxB,GAAG,CAAC,GAAG,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;IACtB,CAAC;IACD,MAAM,OAAO,GAAG,CAAC,GAAG,GAAG,CAAC,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC;IAC5F,MAAM,KAAK,GAAG,OAAO,CAAC,MAAM,CAC1B,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;QACT,KAAK,EAAE,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK;QACxB,MAAM,EAAE,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,MAAM;QAC3B,SAAS,EAAE,CAAC,CAAC,SAAS,GAAG,CAAC,CAAC,SAAS;QACpC,UAAU,EAAE,CAAC,CAAC,UAAU,GAAG,CAAC,CAAC,UAAU;QACvC,KAAK,EAAE,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK;KACzB,CAAC,EACF,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,CAC/D,CAAC;IACF,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,WAAW,EAAE,GAAG,CAAC,WAAW,EAAE,EAAE,CAAC;AACpE,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,cAAc,CAAC,MAAc,EAAE,GAAS;IAC5D,MAAM,CAAC,GAAG,cAAc,CAAC,MAAM,CAAC,CAAC;IACjC,MAAM,IAAI,GAAG,MAAM,iBAAiB,CAAC,CAAC,CAAC,CAAC;IACxC,MAAM,GAAG,GAAG,WAAW,EAAE,CAAC;IAC1B,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,MAAM,EAAE,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,SAAS,EAAE,CAAC,CAAC,CAAC;IACtF,OAAO;QACL,IAAI;QACJ,UAAU,EAAE,IAAI,CAAC,UAAU;QAC3B,OAAO,EAAE,IAAI,CAAC,OAAO;QACrB,WAAW,EAAE,GAAG,CAAC,WAAW,EAAE;KAC/B,CAAC;AACJ,CAAC"}

package/dist/lib/control-plane/call-context.d.ts

@@ -0,0 +1,7 @@
+export interface CallContext {
+    sessionId: string;
+    rootId: string;
+}
+export declare function withCallContext<T>(ctx: CallContext, fn: () => T): T;
+export declare function currentCallContext(): CallContext | undefined;
+//# sourceMappingURL=call-context.d.ts.map

package/dist/lib/control-plane/call-context.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"call-context.d.ts","sourceRoot":"","sources":["../../../src/lib/control-plane/call-context.ts"],"names":[],"mappings":"AAQA,MAAM,WAAW,WAAW;IAC1B,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,EAAE,MAAM,CAAC;CAChB;AAID,wBAAgB,eAAe,CAAC,CAAC,EAAE,GAAG,EAAE,WAAW,EAAE,EAAE,EAAE,MAAM,CAAC,GAAG,CAAC,CAEnE;AAED,wBAAgB,kBAAkB,IAAI,WAAW,GAAG,SAAS,CAE5D"}

package/dist/lib/control-plane/call-context.js

@@ -0,0 +1,9 @@
+import { AsyncLocalStorage } from "node:async_hooks";
+const als = new AsyncLocalStorage();
+export function withCallContext(ctx, fn) {
+    return als.run(ctx, fn);
+}
+export function currentCallContext() {
+    return als.getStore();
+}
+//# sourceMappingURL=call-context.js.map

package/dist/lib/control-plane/call-context.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"call-context.js","sourceRoot":"","sources":["../../../src/lib/control-plane/call-context.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AAarD,MAAM,GAAG,GAAG,IAAI,iBAAiB,EAAe,CAAC;AAEjD,MAAM,UAAU,eAAe,CAAI,GAAgB,EAAE,EAAW;IAC9D,OAAO,GAAG,CAAC,GAAG,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;AAC1B,CAAC;AAED,MAAM,UAAU,kBAAkB;IAChC,OAAO,GAAG,CAAC,QAAQ,EAAE,CAAC;AACxB,CAAC"}

package/dist/lib/control-plane/catalog.d.ts

@@ -0,0 +1,33 @@
+import { type ToolDescriptorTS } from "./tools-config.js";
+export interface CatalogSubagent {
+    name: string;
+    title: string;
+    description: string;
+    tools: string[];
+    nests: string[];
+    hitl: string[];
+    launch: string;
+    hasSettings: boolean;
+    icon: string | null;
+    role: string | null;
+    kind?: "subagent" | "task";
+    groupBy: string | null;
+}
+export interface CatalogResponseTS {
+    subagents: CatalogSubagent[];
+    tools: ToolDescriptorTS[];
+    generatedAt: string;
+}
+interface ParsedInstructions {
+    title?: string;
+    description?: string;
+    launch?: string;
+    icon?: string;
+    role?: string;
+    groupBy?: string;
+}
+export declare function parseInstructions(md: string): ParsedInstructions;
+export declare function discoverSubagents(root?: string): CatalogSubagent[];
+export declare function buildCatalog(role?: string): CatalogResponseTS;
+export {};
+//# sourceMappingURL=catalog.d.ts.map

package/dist/lib/control-plane/catalog.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"catalog.d.ts","sourceRoot":"","sources":["../../../src/lib/control-plane/catalog.ts"],"names":[],"mappings":"AAuBA,OAAO,EAAiB,KAAK,gBAAgB,EAAmB,MAAM,mBAAmB,CAAC;AAI1F,MAAM,WAAW,eAAe;IAC9B,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,MAAM,CAAC;IACd,WAAW,EAAE,MAAM,CAAC;IACpB,KAAK,EAAE,MAAM,EAAE,CAAC;IAChB,KAAK,EAAE,MAAM,EAAE,CAAC;IAChB,IAAI,EAAE,MAAM,EAAE,CAAC;IACf,MAAM,EAAE,MAAM,CAAC;IAGf,WAAW,EAAE,OAAO,CAAC;IAGrB,IAAI,EAAE,MAAM,GAAG,IAAI,CAAC;IAIpB,IAAI,EAAE,MAAM,GAAG,IAAI,CAAC;IAIpB,IAAI,CAAC,EAAE,UAAU,GAAG,MAAM,CAAC;IAI3B,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;CACxB;AAED,MAAM,WAAW,iBAAiB;IAChC,SAAS,EAAE,eAAe,EAAE,CAAC;IAC7B,KAAK,EAAE,gBAAgB,EAAE,CAAC;IAC1B,WAAW,EAAE,MAAM,CAAC;CACrB;AA2CD,UAAU,kBAAkB;IAC1B,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB;AA+BD,wBAAgB,iBAAiB,CAAC,EAAE,EAAE,MAAM,GAAG,kBAAkB,CAwChE;AA6BD,wBAAgB,iBAAiB,CAAC,IAAI,SAAiB,GAAG,eAAe,EAAE,CAoB1E;AAQD,wBAAgB,YAAY,CAAC,IAAI,CAAC,EAAE,MAAM,GAAG,iBAAiB,CAU7D"}