@hydra-acp/budgeter 0.1.3 → 0.1.5

package/dist/enforce.js

@@ -1,81 +1,10 @@
-// Builds outbound emit_message payloads and stop responses for the
-// router. Centralises the "what does a warning/rejection look like on
-// the wire" decisions so the router only deals with verdicts.
-export class Enforcer {
-    client;
-    log;
-    trace;
-    constructor(client, log, trace) {
-        this.client = client;
-        this.log = log;
-        this.trace = trace;
-    }
-    // Emit a warning to every client attached to the session. Sent as a
-    // session/update with sessionUpdate="agent_message_chunk" so it
-    // surfaces in the conversation log on every well-behaved client. A
-    // _meta.hydra-acp.budgeter marker lets renderer-aware clients style it
-    // differently (or filter it out, if they prefer their own UI).
-    async warn(sessionId, warn) {
-        const text = warn.body ? `\n\n${warn.title}\n${warn.body}\n\n` : `\n\n${warn.title}\n\n`;
-        const envelope = {
-            sessionId,
-            update: {
-                sessionUpdate: "agent_message_chunk",
-                content: { type: "text", text },
-                _meta: {
-                    "hydra-acp": {
-                        budgeter: {
-                            title: warn.title,
-                            ...(warn.body !== undefined ? { body: warn.body } : {}),
-                        },
-                    },
-                },
-            },
-        };
-        if (this.trace) {
-            this.trace.emits.push({
-                sessionId,
-                method: "session/update",
-                envelope,
-            });
-        }
-        try {
-            await this.client.request("hydra-acp/emit_message", {
-                sessionId,
-                method: "session/update",
-                envelope,
-                route: "chain",
-            });
-        }
-        catch (err) {
-            this.log.warn(`emit_message warn for ${sessionId} failed: ${err.message}`);
-        }
-    }
-    // Build the payload that the router returns as the stop action's
-    // payload. Becomes the response delivered to the originator of the
-    // session/prompt — clients render the stopReason and the _meta.
-    buildRejectPayload(reject) {
-        return {
-            stopReason: reject.stopReason ?? "refusal",
-            _meta: {
-                "hydra-acp": {
-                    budgeter: { message: reject.message },
-                },
-            },
-        };
-    }
-    // Convenience: combine a warn + reject into the appropriate side
-    // effects. Called by the router on a prompt_request verdict that
-    // includes both — the warn fires asynchronously so the stop payload
-    // can be returned to the daemon without blocking on emit_message.
-    async dispatch(sessionId, verdict) {
-        if (verdict.warn) {
-            void this.warn(sessionId, verdict.warn);
-        }
-        if (verdict.reject) {
-            return this.buildRejectPayload(verdict.reject);
-        }
-        return undefined;
-    }
-}
-//# sourceMappingURL=enforce.js.map
+class i{constructor(t,e,r){this.client=t;this.log=e;this.trace=r}client;log;trace;async warn(t,e){const r=e.body?`
+
+${e.title}
+${e.body}
+
+`:`
+
+${e.title}
+
+`,n={sessionId:t,update:{sessionUpdate:"agent_message_chunk",content:{type:"text",text:r},_meta:{"hydra-acp":{budgeter:{title:e.title,...e.body!==void 0?{body:e.body}:{}}}}}};this.trace&&this.trace.emits.push({sessionId:t,method:"session/update",envelope:n});try{await this.client.request("hydra-acp/message/emit",{sessionId:t,method:"session/update",envelope:n,route:"chain"})}catch(s){this.log.warn(`emit_message warn for ${t} failed: ${s.message}`)}}buildRejectPayload(t){return{stopReason:t.stopReason??"refusal",_meta:{"hydra-acp":{budgeter:{message:t.message}}}}}async dispatch(t,e){if(e.warn&&this.warn(t,e.warn),e.reject)return this.buildRejectPayload(e.reject)}}export{i as Enforcer};

package/dist/index.js

@@ -1,65 +1,5 @@
 #!/usr/bin/env node
-import { readFileSync } from "node:fs";
-import { dirname, resolve } from "node:path";
-import { fileURLToPath } from "node:url";
-import { loadConfig } from "./config.js";
-import { BudgeterBridge } from "./bridge.js";
-import { stateFilePath } from "./paths.js";
-import { DEFAULT_RULE } from "./rule.js";
-import { deleteStateFile } from "./tracker.js";
-import { logger, setDebug } from "./util/log.js";
-const log = logger("main");
-function readVersion() {
-    try {
-        const here = dirname(fileURLToPath(import.meta.url));
-        const pkg = JSON.parse(readFileSync(resolve(here, "../package.json"), "utf8"));
-        return pkg.version ?? "unknown";
-    }
-    catch {
-        return "unknown";
-    }
-}
-function runReset() {
-    deleteStateFile(stateFilePath());
-    process.stdout.write("hydra-acp-budgeter accumulated cost reset\n");
-}
-async function runTransformer() {
-    const config = loadConfig();
-    setDebug(config.debug);
-    const statePath = stateFilePath();
-    const bridge = new BudgeterBridge({
-        daemonWsUrl: config.hydraWsUrl,
-        token: config.hydraToken,
-        softLimit: config.softLimit,
-        hardLimit: config.hardLimit,
-        currency: config.currency,
-        rule: DEFAULT_RULE,
-        statePath,
-    });
-    bridge.start();
-    const shutdown = (sig) => {
-        log.info(`${sig} received — shutting down`);
-        bridge.stop();
-        setTimeout(() => process.exit(0), 200).unref();
-    };
-    process.on("SIGINT", () => shutdown("SIGINT"));
-    process.on("SIGTERM", () => shutdown("SIGTERM"));
-    log.info(`hydra-acp-budgeter up; daemon=${config.hydraDaemonUrl} soft=${config.softLimit} hard=${config.hardLimit} ${config.currency} state=${statePath}`);
-}
-async function main() {
-    const argv = process.argv.slice(2);
-    if (argv.includes("--version") || argv.includes("-v")) {
-        process.stdout.write(`hydra-acp-budgeter ${readVersion()}\n`);
-        return;
-    }
-    if (argv[0] === "reset") {
-        runReset();
-        return;
-    }
-    await runTransformer();
-}
-main().catch((err) => {
-    process.stderr.write(`hydra-acp-budgeter: ${err.message}\n`);
-    process.exit(1);
-});
-//# sourceMappingURL=index.js.map
+import{readFileSync as a}from"node:fs";import{dirname as c,resolve as d}from"node:path";import{fileURLToPath as m}from"node:url";import{loadConfig as u}from"./config.js";import{BudgeterBridge as p}from"./bridge.js";import{stateFilePath as n}from"./paths.js";import{DEFAULT_RULE as f}from"./rule.js";import{deleteStateFile as g}from"./tracker.js";import{logger as h,setDebug as l}from"./util/log.js";const s=h("main");function y(){try{const r=c(m(import.meta.url));return JSON.parse(a(d(r,"../package.json"),"utf8")).version??"unknown"}catch{return"unknown"}}function v(){g(n()),process.stdout.write(`hydra-acp-budgeter accumulated cost reset
+`)}async function w(){const r=u();l(r.debug);const e=n(),t=new p({daemonWsUrl:r.hydraWsUrl,token:r.hydraToken,softLimit:r.softLimit,hardLimit:r.hardLimit,currency:r.currency,rule:f,statePath:e});t.start();const o=i=>{s.info(`${i} received \u2014 shutting down`),t.stop(),setTimeout(()=>process.exit(0),200).unref()};process.on("SIGINT",()=>o("SIGINT")),process.on("SIGTERM",()=>o("SIGTERM")),s.info(`hydra-acp-budgeter up; daemon=${r.hydraDaemonUrl} soft=${r.softLimit} hard=${r.hardLimit} ${r.currency} state=${e}`)}async function L(){const r=process.argv.slice(2);if(r.includes("--version")||r.includes("-v")){process.stdout.write(`hydra-acp-budgeter ${y()}
+`);return}if(r[0]==="reset"){v();return}await w()}L().catch(r=>{process.stderr.write(`hydra-acp-budgeter: ${r.message}
+`),process.exit(1)});

package/dist/paths.js

@@ -1,12 +1 @@
-import { homedir } from "node:os";
-import { resolve } from "node:path";
-// hydra-acp injects HYDRA_ACP_HOME when it spawns transformers; this file
-// is invoked both in that role and as a one-shot subcommand from the
-// user's shell, so fall back to ~/.hydra-acp when the env isn't set.
-function hydraHome() {
-    return process.env.HYDRA_ACP_HOME ?? resolve(homedir(), ".hydra-acp");
-}
-export function stateFilePath() {
-    return resolve(hydraHome(), "budgeter-cost.json");
-}
-//# sourceMappingURL=paths.js.map
+import{homedir as t}from"node:os";import{resolve as r}from"node:path";function e(){return process.env.HYDRA_ACP_HOME??r(t(),".hydra-acp")}function i(){return r(e(),"budgeter-cost.json")}export{i as stateFilePath};

package/dist/router.js

@@ -1,161 +1 @@
-// Translates incoming transformer/message and transformer/session_event
-// envelopes into BudgetEvents, runs them through the rule, and dispatches
-// the resulting verdicts via the Enforcer. Unlike the notifier's
-// EventRouter (one instance per session), this is a singleton — the
-// transformer process holds one connection for every session, so the
-// router keeps a sessionId → SessionMeta map.
-export class EventRouter {
-    rule;
-    tracker;
-    enforcer;
-    log;
-    metas = new Map();
-    constructor(rule, tracker, enforcer, log) {
-        this.rule = rule;
-        this.tracker = tracker;
-        this.enforcer = enforcer;
-        this.log = log;
-    }
-    setMeta(sessionId, meta) {
-        this.metas.set(sessionId, meta);
-    }
-    forgetSession(sessionId) {
-        this.metas.delete(sessionId);
-    }
-    // Fired by the bridge when an agent→client session/update notification
-    // is being routed. usage_update is the only kind we care about for cost
-    // tracking; other kinds may still flow through the rule so a custom
-    // rule can observe (but the default rule ignores them).
-    async onResponseUpdate(sessionId, update) {
-        const kind = typeof update.sessionUpdate === "string" ? update.sessionUpdate : "";
-        // session_info_update may carry an updated title; mirror the notifier
-        // and refresh meta so subsequent warning messages reflect the latest.
-        if (kind === "session_info_update") {
-            this.applySessionInfoUpdate(sessionId, update);
-        }
-        if (kind !== "usage_update") {
-            return;
-        }
-        this.tracker.applyUsageUpdate(sessionId, update);
-        const transition = this.tracker.consumeStateTransition();
-        if (transition !== undefined) {
-            await this.fire({
-                sessionId,
-                kind: "threshold_cross",
-                raw: { to: transition, from: this.tracker.state },
-                meta: this.metaFor(sessionId),
-                budget: this.tracker.snapshotFor(sessionId),
-            });
-        }
-        // Also run a plain usage_update event so a custom rule can act on
-        // every tick (e.g. emit a status notification every $1 spent). The
-        // default rule ignores this kind.
-        await this.fire({
-            sessionId,
-            kind: "usage_update",
-            raw: update,
-            meta: this.metaFor(sessionId),
-            budget: this.tracker.snapshotFor(sessionId),
-        });
-    }
-    // Fired by the bridge for a request:session/prompt intercept. Returns
-    // the stop payload to send back to the daemon, or undefined to let the
-    // prompt continue. The rule decides — the default rule rejects when
-    // we're in the "hard" state.
-    async onPromptRequest(sessionId, envelope) {
-        const verdict = await this.runRule({
-            sessionId,
-            kind: "prompt_request",
-            raw: { envelope },
-            meta: this.metaFor(sessionId),
-            budget: this.tracker.snapshotFor(sessionId),
-        });
-        if (!verdict) {
-            return undefined;
-        }
-        return this.enforcer.dispatch(sessionId, verdict);
-    }
-    // Fired by the bridge for lifecycle:session.opened and session.closed.
-    async onLifecycle(event, sessionId) {
-        if (event === "session.opened") {
-            await this.fire({
-                sessionId,
-                kind: "session_opened",
-                raw: {},
-                meta: this.metaFor(sessionId),
-                budget: this.tracker.snapshotFor(sessionId),
-            });
-            return;
-        }
-        if (event === "session.closed") {
-            // Note: we deliberately do NOT drop the session's cost from the
-            // tracker — total spend is sticky across session.closed so the
-            // budget reflects every dollar this transformer has seen. Reset
-            // via `hydra-acp-budgeter reset` (deletes the state file) when
-            // you want to zero it.
-            await this.fire({
-                sessionId,
-                kind: "session_closed",
-                raw: {},
-                meta: this.metaFor(sessionId),
-                budget: this.tracker.snapshotFor(sessionId),
-            });
-            this.forgetSession(sessionId);
-        }
-    }
-    async fire(ev) {
-        const verdict = await this.runRule(ev);
-        if (!verdict) {
-            return;
-        }
-        // For non-prompt events, only the warn side can fire — reject without
-        // a prompt to attach it to is a no-op. Log and skip so a misconfigured
-        // rule doesn't fail silently.
-        if (verdict.reject && ev.kind !== "prompt_request") {
-            this.log.warn(`rule returned reject on kind=${ev.kind} for ${ev.sessionId.slice(0, 12)} — only valid on prompt_request; ignored`);
-        }
-        if (verdict.warn) {
-            await this.enforcer.warn(ev.sessionId, verdict.warn);
-        }
-    }
-    async runRule(ev) {
-        try {
-            return await this.rule(ev);
-        }
-        catch (err) {
-            this.log.warn(`rule threw on kind=${ev.kind} sessionId=${ev.sessionId.slice(0, 12)}: ${err.message}; skipping`);
-            return null;
-        }
-    }
-    metaFor(sessionId) {
-        return this.metas.get(sessionId) ?? {};
-    }
-    applySessionInfoUpdate(sessionId, update) {
-        const cur = { ...this.metaFor(sessionId) };
-        let changed = false;
-        if (typeof update.title === "string" && cur.title !== update.title) {
-            cur.title = update.title;
-            changed = true;
-        }
-        const agentId = readHydraAgentId(update._meta);
-        if (agentId !== undefined && cur.agentId !== agentId) {
-            cur.agentId = agentId;
-            changed = true;
-        }
-        if (changed) {
-            this.metas.set(sessionId, cur);
-        }
-    }
-}
-function readHydraAgentId(meta) {
-    if (!meta || typeof meta !== "object" || Array.isArray(meta)) {
-        return undefined;
-    }
-    const ns = meta["hydra-acp"];
-    if (!ns || typeof ns !== "object" || Array.isArray(ns)) {
-        return undefined;
-    }
-    const v = ns.agentId;
-    return typeof v === "string" ? v : undefined;
-}
-//# sourceMappingURL=router.js.map
+class a{constructor(t,e,r,n){this.rule=t;this.tracker=e;this.enforcer=r;this.log=n}rule;tracker;enforcer;log;metas=new Map;setMeta(t,e){this.metas.set(t,e)}forgetSession(t){this.metas.delete(t)}async onResponseUpdate(t,e){const r=typeof e.sessionUpdate=="string"?e.sessionUpdate:"";if(r==="session_info_update"&&this.applySessionInfoUpdate(t,e),r!=="usage_update")return;this.tracker.applyUsageUpdate(t,e);const n=this.tracker.consumeStateTransition();n!==void 0&&await this.fire({sessionId:t,kind:"threshold_cross",raw:{to:n,from:this.tracker.state},meta:this.metaFor(t),budget:this.tracker.snapshotFor(t)}),await this.fire({sessionId:t,kind:"usage_update",raw:e,meta:this.metaFor(t),budget:this.tracker.snapshotFor(t)})}async onPromptRequest(t,e){const r=await this.runRule({sessionId:t,kind:"prompt_request",raw:{envelope:e},meta:this.metaFor(t),budget:this.tracker.snapshotFor(t)});if(r)return this.enforcer.dispatch(t,r)}async onLifecycle(t,e){if(t==="session.opened"){await this.fire({sessionId:e,kind:"session_opened",raw:{},meta:this.metaFor(e),budget:this.tracker.snapshotFor(e)});return}t==="session.closed"&&(await this.fire({sessionId:e,kind:"session_closed",raw:{},meta:this.metaFor(e),budget:this.tracker.snapshotFor(e)}),this.forgetSession(e))}async fire(t){const e=await this.runRule(t);e&&(e.reject&&t.kind!=="prompt_request"&&this.log.warn(`rule returned reject on kind=${t.kind} for ${t.sessionId.slice(0,12)} \u2014 only valid on prompt_request; ignored`),e.warn&&await this.enforcer.warn(t.sessionId,e.warn))}async runRule(t){try{return await this.rule(t)}catch(e){return this.log.warn(`rule threw on kind=${t.kind} sessionId=${t.sessionId.slice(0,12)}: ${e.message}; skipping`),null}}metaFor(t){return this.metas.get(t)??{}}applySessionInfoUpdate(t,e){const r={...this.metaFor(t)};let n=!1;typeof e.title=="string"&&r.title!==e.title&&(r.title=e.title,n=!0);const s=o(e._meta);s!==void 0&&r.agentId!==s&&(r.agentId=s,n=!0),n&&this.metas.set(t,r)}}function o(i){if(!i||typeof i!="object"||Array.isArray(i))return;const t=i["hydra-acp"];if(!t||typeof t!="object"||Array.isArray(t))return;const e=t.agentId;return typeof e=="string"?e:void 0}export{a as EventRouter};

package/dist/rule.js

@@ -1,86 +1 @@
-const CURRENCY_SYMBOLS = {
-    USD: "$",
-    EUR: "€",
-    GBP: "£",
-    JPY: "¥",
-    CNY: "¥",
-    CAD: "C$",
-    AUD: "A$",
-    NZD: "NZ$",
-    CHF: "CHF ",
-    HKD: "HK$",
-    SGD: "S$",
-    INR: "₹",
-    KRW: "₩",
-    BRL: "R$",
-    MXN: "MX$",
-};
-function fmtMoney(amount, currency) {
-    const fixed = amount.toFixed(2);
-    const sym = CURRENCY_SYMBOLS[currency.toUpperCase()];
-    if (sym) {
-        return `${sym}${fixed}`;
-    }
-    return `${currency} ${fixed}`;
-}
-// The rule the budgeter runs on every event. Strategy:
-//   - threshold_cross to "soft": warn (one-shot, the tracker only fires
-//     crosses once per transition)
-//   - threshold_cross to "hard": warn (also one-shot)
-//   - prompt_request in "hard" state: reject with stopReason "refusal"
-//     and an explanation in the message
-//   - session_opened in "hard" state: warn the new session so the user
-//     understands why their next prompt will bounce
-//   - usage_update and session_closed: do nothing
-export const DEFAULT_RULE = (ev) => {
-    const { budget } = ev;
-    const totalStr = fmtMoney(budget.total, budget.currency);
-    const softStr = fmtMoney(budget.soft, budget.currency);
-    const hardStr = fmtMoney(budget.hard, budget.currency);
-    if (ev.kind === "threshold_cross") {
-        const to = (ev.raw.to ?? "ok");
-        if (to === "hard") {
-            return {
-                warn: {
-                    title: `🛑 Budget hard limit hit`,
-                    body: `Spent ${totalStr} ≥ ${hardStr} hard limit. Further prompts will be rejected until the budget is reset.`,
-                },
-            };
-        }
-        return null;
-    }
-    // Warn on every turn that reports a cost while over the soft limit.
-    if (ev.kind === "usage_update" && budget.state !== "ok" && typeof ev.raw.cost?.amount === "number") {
-        const label = budget.state === "hard" ? "🛑 Over hard limit" : "💰 Over soft limit";
-        return {
-            warn: {
-                title: `${label} · ${totalStr} spent`,
-                body: budget.state === "hard"
-                    ? `Hard limit ${hardStr} reached. Prompts will be rejected until budget is reset.`
-                    : `Soft limit ${softStr} reached (hard: ${hardStr}).`,
-            },
-        };
-    }
-    if (ev.kind === "prompt_request" && budget.state === "hard") {
-        return {
-            warn: {
-                title: `🛑 Prompt blocked — budget exceeded`,
-                body: `Spent ${totalStr} ≥ ${hardStr} hard limit. Reset the budget or raise HYDRA_ACP_BUDGETER_HARD to continue.`,
-            },
-            reject: {
-                message: `Budget exceeded: spent ${totalStr} ≥ ${hardStr} hard limit. Reset the budget or raise HYDRA_ACP_BUDGETER_HARD to continue.`,
-                stopReason: "refusal",
-            },
-        };
-    }
-    if (ev.kind === "session_opened" && budget.state === "hard") {
-        return {
-            warn: {
-                title: `🛑 Session opened over budget`,
-                body: `Total spend ${totalStr} ≥ ${hardStr}. Prompts on this session will be rejected.`,
-            },
-        };
-    }
-    return null;
-};
-//# sourceMappingURL=rule.js.map
+const d={USD:"$",EUR:"\u20AC",GBP:"\xA3",JPY:"\xA5",CNY:"\xA5",CAD:"C$",AUD:"A$",NZD:"NZ$",CHF:"CHF ",HKD:"HK$",SGD:"S$",INR:"\u20B9",KRW:"\u20A9",BRL:"R$",MXN:"MX$"};function o(e,t){const r=e.toFixed(2),s=d[t.toUpperCase()];return s?`${s}${r}`:`${t} ${r}`}const u=e=>{const{budget:t}=e,r=o(t.total,t.currency),s=o(t.soft,t.currency),n=o(t.hard,t.currency);return e.kind==="threshold_cross"?(e.raw.to??"ok")==="hard"?{warn:{title:"\u{1F6D1} Budget hard limit hit",body:`Spent ${r} \u2265 ${n} hard limit. Further prompts will be rejected until the budget is reset.`}}:null:e.kind==="usage_update"&&t.state!=="ok"&&typeof e.raw.cost?.amount=="number"?{warn:{title:`${t.state==="hard"?"\u{1F6D1} Over hard limit":"\u{1F4B0} Over soft limit"} \xB7 ${r} spent`,body:t.state==="hard"?`Hard limit ${n} reached. Prompts will be rejected until budget is reset.`:`Soft limit ${s} reached (hard: ${n}).`}}:e.kind==="prompt_request"&&t.state==="hard"?{warn:{title:"\u{1F6D1} Prompt blocked \u2014 budget exceeded",body:`Spent ${r} \u2265 ${n} hard limit. Reset the budget or raise HYDRA_ACP_BUDGETER_HARD to continue.`},reject:{message:`Budget exceeded: spent ${r} \u2265 ${n} hard limit. Reset the budget or raise HYDRA_ACP_BUDGETER_HARD to continue.`,stopReason:"refusal"}}:e.kind==="session_opened"&&t.state==="hard"?{warn:{title:"\u{1F6D1} Session opened over budget",body:`Total spend ${r} \u2265 ${n}. Prompts on this session will be rejected.`}}:null};export{u as DEFAULT_RULE};