adonis-ai-kit 0.0.0-stage → 1.1.0

package/dist/src/budget/cost_tracker.js

@@ -0,0 +1,199 @@
+import { DEFAULT_MODEL_PRICING, computeCost, lookupPricing, normalizeModelName, toSafeTokens, } from './pricing.js';
+const DEFAULT_MAX_RECORDS = 10_000;
+const MAX_WARNED_MODELS = 1_000;
+function newAggregate() {
+    return { tokens: 0, cost: 0, calls: 0, totalDurationMs: 0 };
+}
+function addTo(agg, record) {
+    agg.tokens += record.totalTokens;
+    agg.cost += record.costUsd;
+    agg.calls += 1;
+    agg.totalDurationMs += record.durationMs;
+}
+/**
+ * Tracks token usage and monetary cost per agent, model, and session.
+ * Provides aggregations and logging without framework dependencies.
+ *
+ * All keyed state uses `Map`, so agent/model names such as `__proto__` or `constructor`
+ * cannot pollute prototypes. Memory is bounded by `maxRecords`.
+ */
+export class CostTracker {
+    ring = [];
+    ringStart = 0;
+    ringSize = 0;
+    maxRecords;
+    sink;
+    byAgent = new Map();
+    byModel = new Map();
+    total = newAggregate();
+    customPricing = new Map();
+    defaultPricing = new Map(Object.entries(DEFAULT_MODEL_PRICING));
+    warnedModels = new Set();
+    logger;
+    pricingLoader;
+    constructor(opts = {}) {
+        const max = opts.maxRecords ?? DEFAULT_MAX_RECORDS;
+        if (!Number.isInteger(max) || max < 0) {
+            throw new RangeError(`[CostTracker] maxRecords must be a non-negative integer, got ${max}`);
+        }
+        this.maxRecords = max;
+        this.sink = opts.sink;
+        if (opts.pricing) {
+            this.setPricing(opts.pricing);
+        }
+        this.logger = opts.logger;
+        this.pricingLoader = opts.pricingLoader;
+    }
+    setLogger(logger) {
+        this.logger = logger;
+    }
+    setPricing(pricing) {
+        for (const [model, price] of Object.entries(pricing)) {
+            this.customPricing.set(model, price);
+        }
+    }
+    registerPricingLoader(loader) {
+        this.pricingLoader = loader;
+    }
+    async loadPricing() {
+        if (!this.pricingLoader)
+            return;
+        try {
+            const loaded = await this.pricingLoader();
+            this.setPricing(loaded);
+        }
+        catch (err) {
+            this.logger?.warn?.('[CostTracker] Failed to load dynamic pricing:', err);
+        }
+    }
+    /**
+     * Backwards-compatible initialization alias.
+     */
+    async initFromCatalog() {
+        return this.loadPricing();
+    }
+    record(entry) {
+        const inputTokens = toSafeTokens(entry.inputTokens);
+        const outputTokens = toSafeTokens(entry.outputTokens);
+        const details = {
+            cacheReadTokens: toSafeTokens(entry.cacheReadTokens),
+            cacheWriteTokens: toSafeTokens(entry.cacheWriteTokens),
+            reasoningTokens: toSafeTokens(entry.reasoningTokens),
+        };
+        const costUsd = this.calculateCost(entry.model, inputTokens, outputTokens, details);
+        const record = {
+            ...entry,
+            agentName: String(entry.agentName || 'unknown'),
+            inputTokens,
+            outputTokens,
+            cacheReadTokens: details.cacheReadTokens,
+            cacheWriteTokens: details.cacheWriteTokens,
+            reasoningTokens: details.reasoningTokens,
+            totalTokens: toSafeTokens(entry.totalTokens) || inputTokens + outputTokens,
+            durationMs: toSafeTokens(entry.durationMs),
+            costUsd,
+            timestamp: Date.now(),
+        };
+        this.aggregate(record);
+        this.store(record);
+        if (this.sink) {
+            try {
+                this.sink(record);
+            }
+            catch (err) {
+                this.logger?.warn?.('[CostTracker] Sink failed:', err);
+            }
+        }
+        return costUsd;
+    }
+    /**
+     * Cost in USD. `inputTokens` is the total input (cached included), `outputTokens` the total
+     * output (reasoning included); `details` carves cache/reasoning tokens out at their own rates.
+     */
+    calculateCost(model, inputTokens, outputTokens, details = {}) {
+        const pricing = lookupPricing(model, this.customPricing) ?? lookupPricing(model, this.defaultPricing);
+        if (!pricing) {
+            const shortModel = normalizeModelName(model);
+            if (!this.warnedModels.has(shortModel) && this.warnedModels.size < MAX_WARNED_MODELS) {
+                this.warnedModels.add(shortModel);
+                this.logger?.warn?.(`[CostTracker] Unknown model "${shortModel}" (full: "${model}"). Cost tracking will report $0. Consider configuring pricing.`);
+            }
+            return 0;
+        }
+        return computeCost(pricing, inputTokens, outputTokens, details);
+    }
+    getByAgent() {
+        return Object.fromEntries([...this.byAgent].map(([name, v]) => [
+            name,
+            {
+                tokens: v.tokens,
+                cost: v.cost,
+                calls: v.calls,
+                avgDurationMs: v.calls > 0 ? Math.round(v.totalDurationMs / v.calls) : 0,
+                totalDurationMs: v.totalDurationMs,
+            },
+        ]));
+    }
+    getByModel() {
+        return Object.fromEntries([...this.byModel].map(([model, v]) => [
+            model,
+            { tokens: v.tokens, cost: v.cost, calls: v.calls },
+        ]));
+    }
+    getTotalCost() {
+        return this.total.cost;
+    }
+    getTotalTokens() {
+        return this.total.tokens;
+    }
+    getTotalCalls() {
+        return this.total.calls;
+    }
+    /** Most recent records (at most `maxRecords`), oldest first. */
+    getRecords() {
+        const out = [];
+        for (let i = 0; i < this.ringSize; i++) {
+            out.push(this.ring[(this.ringStart + i) % this.maxRecords]);
+        }
+        return out;
+    }
+    reset() {
+        this.ring = [];
+        this.ringStart = 0;
+        this.ringSize = 0;
+        this.byAgent.clear();
+        this.byModel.clear();
+        this.total = newAggregate();
+        this.warnedModels.clear();
+    }
+    toLogString() {
+        const byAgent = this.getByAgent();
+        const lines = Object.entries(byAgent).map(([name, { tokens, cost, calls, avgDurationMs }]) => `  ${name}: ${calls} calls, ${tokens} tokens, $${cost.toFixed(4)}, avg ${avgDurationMs}ms`);
+        lines.push(`  TOTAL: ${this.total.calls} calls, ${this.getTotalTokens()} tokens, $${this.getTotalCost().toFixed(4)}`);
+        return lines.join('\n');
+    }
+    aggregate(record) {
+        addTo(this.total, record);
+        let agent = this.byAgent.get(record.agentName);
+        if (!agent)
+            this.byAgent.set(record.agentName, (agent = newAggregate()));
+        addTo(agent, record);
+        const modelKey = normalizeModelName(record.model);
+        let model = this.byModel.get(modelKey);
+        if (!model)
+            this.byModel.set(modelKey, (model = newAggregate()));
+        addTo(model, record);
+    }
+    store(record) {
+        if (this.maxRecords === 0)
+            return;
+        if (this.ringSize < this.maxRecords) {
+            this.ring[(this.ringStart + this.ringSize) % this.maxRecords] = record;
+            this.ringSize++;
+            return;
+        }
+        this.ring[this.ringStart] = record;
+        this.ringStart = (this.ringStart + 1) % this.maxRecords;
+    }
+}
+//# sourceMappingURL=cost_tracker.js.map

package/dist/src/budget/cost_tracker.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cost_tracker.js","sourceRoot":"","sources":["../../../src/budget/cost_tracker.ts"],"names":[],"mappings":"AAOA,OAAO,EACL,qBAAqB,EACrB,WAAW,EACX,aAAa,EACb,kBAAkB,EAClB,YAAY,GACb,MAAM,cAAc,CAAA;AA2BrB,MAAM,mBAAmB,GAAG,MAAM,CAAA;AAClC,MAAM,iBAAiB,GAAG,KAAK,CAAA;AAE/B,SAAS,YAAY;IACnB,OAAO,EAAE,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,eAAe,EAAE,CAAC,EAAE,CAAA;AAC7D,CAAC;AAED,SAAS,KAAK,CAAC,GAAc,EAAE,MAAuB;IACpD,GAAG,CAAC,MAAM,IAAI,MAAM,CAAC,WAAW,CAAA;IAChC,GAAG,CAAC,IAAI,IAAI,MAAM,CAAC,OAAO,CAAA;IAC1B,GAAG,CAAC,KAAK,IAAI,CAAC,CAAA;IACd,GAAG,CAAC,eAAe,IAAI,MAAM,CAAC,UAAU,CAAA;AAC1C,CAAC;AAED;;;;;;GAMG;AACH,MAAM,OAAO,WAAW;IACd,IAAI,GAAoC,EAAE,CAAA;IAC1C,SAAS,GAAG,CAAC,CAAA;IACb,QAAQ,GAAG,CAAC,CAAA;IACZ,UAAU,CAAQ;IAClB,IAAI,CAAoC;IAExC,OAAO,GAAG,IAAI,GAAG,EAAqB,CAAA;IACtC,OAAO,GAAG,IAAI,GAAG,EAAqB,CAAA;IACtC,KAAK,GAAc,YAAY,EAAE,CAAA;IAEjC,aAAa,GAAG,IAAI,GAAG,EAAwB,CAAA;IAC/C,cAAc,GAAG,IAAI,GAAG,CAAuB,MAAM,CAAC,OAAO,CAAC,qBAAqB,CAAC,CAAC,CAAA;IACrF,YAAY,GAAG,IAAI,GAAG,EAAU,CAAA;IAChC,MAAM,CAAS;IACf,aAAa,CAA8C;IAEnE,YAAY,OAA2B,EAAE;QACvC,MAAM,GAAG,GAAG,IAAI,CAAC,UAAU,IAAI,mBAAmB,CAAA;QAClD,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,GAAG,GAAG,CAAC,EAAE,CAAC;YACtC,MAAM,IAAI,UAAU,CAAC,gEAAgE,GAAG,EAAE,CAAC,CAAA;QAC7F,CAAC;QACD,IAAI,CAAC,UAAU,GAAG,GAAG,CAAA;QACrB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC,IAAI,CAAA;QACrB,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACjB,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;QAC/B,CAAC;QACD,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAA;QACzB,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,aAAa,CAAA;IACzC,CAAC;IAED,SAAS,CAAC,MAAc;QACtB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAA;IACtB,CAAC;IAED,UAAU,CAAC,OAAqC;QAC9C,KAAK,MAAM,CAAC,KAAK,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;YACrD,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,KAAK,EAAE,KAAK,CAAC,CAAA;QACtC,CAAC;IACH,CAAC;IAED,qBAAqB,CAAC,MAAmD;QACvE,IAAI,CAAC,aAAa,GAAG,MAAM,CAAA;IAC7B,CAAC;IAED,KAAK,CAAC,WAAW;QACf,IAAI,CAAC,IAAI,CAAC,aAAa;YAAE,OAAM;QAC/B,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,aAAa,EAAE,CAAA;YACzC,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAA;QACzB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,+CAA+C,EAAE,GAAG,CAAC,CAAA;QAC3E,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,eAAe;QACnB,OAAO,IAAI,CAAC,WAAW,EAAE,CAAA;IAC3B,CAAC;IAED,MAAM,CAAC,KAAqD;QAC1D,MAAM,WAAW,GAAG,YAAY,CAAC,KAAK,CAAC,WAAW,CAAC,CAAA;QACnD,MAAM,YAAY,GAAG,YAAY,CAAC,KAAK,CAAC,YAAY,CAAC,CAAA;QACrD,MAAM,OAAO,GAAsB;YACjC,eAAe,EAAE,YAAY,CAAC,KAAK,CAAC,eAAe,CAAC;YACpD,gBAAgB,EAAE,YAAY,CAAC,KAAK,CAAC,gBAAgB,CAAC;YACtD,eAAe,EAAE,YAAY,CAAC,KAAK,CAAC,eAAe,CAAC;SACrD,CAAA;QACD,MAAM,OAAO,GAAG,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,OAAO,CAAC,CAAA;QAEnF,MAAM,MAAM,GAAoB;YAC9B,GAAG,KAAK;YACR,SAAS,EAAE,MAAM,CAAC,KAAK,CAAC,SAAS,IAAI,SAAS,CAAC;YAC/C,WAAW;YACX,YAAY;YACZ,eAAe,EAAE,OAAO,CAAC,eAAe;YACxC,gBAAgB,EAAE,OAAO,CAAC,gBAAgB;YAC1C,eAAe,EAAE,OAAO,CAAC,eAAe;YACxC,WAAW,EAAE,YAAY,CAAC,KAAK,CAAC,WAAW,CAAC,IAAI,WAAW,GAAG,YAAY;YAC1E,UAAU,EAAE,YAAY,CAAC,KAAK,CAAC,UAAU,CAAC;YAC1C,OAAO;YACP,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;SACtB,CAAA;QAED,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAA;QACtB,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAA;QAElB,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;YACd,IAAI,CAAC;gBACH,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;YACnB,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,IAAI,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,4BAA4B,EAAE,GAAG,CAAC,CAAA;YACxD,CAAC;QACH,CAAC;QACD,OAAO,OAAO,CAAA;IAChB,CAAC;IAED;;;OAGG;IACH,aAAa,CACX,KAAa,EACb,WAAmB,EACnB,YAAoB,EACpB,UAA6B,EAAE;QAE/B,MAAM,OAAO,GACX,aAAa,CAAC,KAAK,EAAE,IAAI,CAAC,aAAa,CAAC,IAAI,aAAa,CAAC,KAAK,EAAE,IAAI,CAAC,cAAc,CAAC,CAAA;QAEvF,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,UAAU,GAAG,kBAAkB,CAAC,KAAK,CAAC,CAAA;YAC5C,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,IAAI,CAAC,YAAY,CAAC,IAAI,GAAG,iBAAiB,EAAE,CAAC;gBACrF,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,UAAU,CAAC,CAAA;gBACjC,IAAI,CAAC,MAAM,EAAE,IAAI,EAAE,CACjB,gCAAgC,UAAU,aAAa,KAAK,iEAAiE,CAC9H,CAAA;YACH,CAAC;YACD,OAAO,CAAC,CAAA;QACV,CAAC;QAED,OAAO,WAAW,CAAC,OAAO,EAAE,WAAW,EAAE,YAAY,EAAE,OAAO,CAAC,CAAA;IACjE,CAAC;IAED,UAAU;QACR,OAAO,MAAM,CAAC,WAAW,CACvB,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC;YACnC,IAAI;YACJ;gBACE,MAAM,EAAE,CAAC,CAAC,MAAM;gBAChB,IAAI,EAAE,CAAC,CAAC,IAAI;gBACZ,KAAK,EAAE,CAAC,CAAC,KAAK;gBACd,aAAa,EAAE,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,eAAe,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;gBACxE,eAAe,EAAE,CAAC,CAAC,eAAe;aACnC;SACF,CAAC,CACH,CAAA;IACH,CAAC;IAED,UAAU;QACR,OAAO,MAAM,CAAC,WAAW,CACvB,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC;YACpC,KAAK;YACL,EAAE,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC,KAAK,EAAE;SACnD,CAAC,CACH,CAAA;IACH,CAAC;IAED,YAAY;QACV,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAA;IACxB,CAAC;IAED,cAAc;QACZ,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,CAAA;IAC1B,CAAC;IAED,aAAa;QACX,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,CAAA;IACzB,CAAC;IAED,gEAAgE;IAChE,UAAU;QACR,MAAM,GAAG,GAAsB,EAAE,CAAA;QACjC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAC,EAAE,EAAE,CAAC;YACvC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,SAAS,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,UAAU,CAAE,CAAC,CAAA;QAC9D,CAAC;QACD,OAAO,GAAG,CAAA;IACZ,CAAC;IAED,KAAK;QACH,IAAI,CAAC,IAAI,GAAG,EAAE,CAAA;QACd,IAAI,CAAC,SAAS,GAAG,CAAC,CAAA;QAClB,IAAI,CAAC,QAAQ,GAAG,CAAC,CAAA;QACjB,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAA;QACpB,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAA;QACpB,IAAI,CAAC,KAAK,GAAG,YAAY,EAAE,CAAA;QAC3B,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,CAAA;IAC3B,CAAC;IAED,WAAW;QACT,MAAM,OAAO,GAAG,IAAI,CAAC,UAAU,EAAE,CAAA;QACjC,MAAM,KAAK,GAAG,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,GAAG,CACvC,CAAC,CAAC,IAAI,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,aAAa,EAAE,CAAC,EAAE,EAAE,CACjD,KAAK,IAAI,KAAK,KAAK,WAAW,MAAM,aAAa,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,aAAa,IAAI,CAC7F,CAAA;QACD,KAAK,CAAC,IAAI,CACR,YAAY,IAAI,CAAC,KAAK,CAAC,KAAK,WAAW,IAAI,CAAC,cAAc,EAAE,aAAa,IAAI,CAAC,YAAY,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAC1G,CAAA;QACD,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;IACzB,CAAC;IAEO,SAAS,CAAC,MAAuB;QACvC,KAAK,CAAC,IAAI,CAAC,KAAK,EAAE,MAAM,CAAC,CAAA;QAEzB,IAAI,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,SAAS,CAAC,CAAA;QAC9C,IAAI,CAAC,KAAK;YAAE,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC,KAAK,GAAG,YAAY,EAAE,CAAC,CAAC,CAAA;QACxE,KAAK,CAAC,KAAK,EAAE,MAAM,CAAC,CAAA;QAEpB,MAAM,QAAQ,GAAG,kBAAkB,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;QACjD,IAAI,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAA;QACtC,IAAI,CAAC,KAAK;YAAE,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC,KAAK,GAAG,YAAY,EAAE,CAAC,CAAC,CAAA;QAChE,KAAK,CAAC,KAAK,EAAE,MAAM,CAAC,CAAA;IACtB,CAAC;IAEO,KAAK,CAAC,MAAuB;QACnC,IAAI,IAAI,CAAC,UAAU,KAAK,CAAC;YAAE,OAAM;QACjC,IAAI,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;YACpC,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,MAAM,CAAA;YACtE,IAAI,CAAC,QAAQ,EAAE,CAAA;YACf,OAAM;QACR,CAAC;QACD,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,MAAM,CAAA;QAClC,IAAI,CAAC,SAAS,GAAG,CAAC,IAAI,CAAC,SAAS,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,UAAU,CAAA;IACzD,CAAC;CACF"}

package/dist/src/budget/index.d.ts

@@ -0,0 +1,6 @@
+export * from './types.js';
+export * from './pricing.js';
+export * from './cost_tracker.js';
+export * from './budget_allocator.js';
+export * from './budget_enforcer.js';
+//# sourceMappingURL=index.d.ts.map

package/dist/src/budget/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/budget/index.ts"],"names":[],"mappings":"AAAA,cAAc,YAAY,CAAA;AAC1B,cAAc,cAAc,CAAA;AAC5B,cAAc,mBAAmB,CAAA;AACjC,cAAc,uBAAuB,CAAA;AACrC,cAAc,sBAAsB,CAAA"}

package/dist/src/budget/index.js

@@ -0,0 +1,6 @@
+export * from './types.js';
+export * from './pricing.js';
+export * from './cost_tracker.js';
+export * from './budget_allocator.js';
+export * from './budget_enforcer.js';
+//# sourceMappingURL=index.js.map

package/dist/src/budget/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/budget/index.ts"],"names":[],"mappings":"AAAA,cAAc,YAAY,CAAA;AAC1B,cAAc,cAAc,CAAA;AAC5B,cAAc,mBAAmB,CAAA;AACjC,cAAc,uBAAuB,CAAA;AACrC,cAAc,sBAAsB,CAAA"}

package/dist/src/budget/pricing.d.ts

@@ -0,0 +1,35 @@
+import type { ModelPricing, TokenUsageDetails } from './types.js';
+/**
+ * Default pricing catalog for popular LLM models (USD per 1M tokens).
+ *
+ * Prices drift: treat this table as a fallback and prefer `CostTracker.setPricing()` /
+ * `registerPricingLoader()` (e.g. backed by models.dev) for authoritative values.
+ */
+export declare const DEFAULT_MODEL_PRICING: Readonly<Record<string, ModelPricing>>;
+/**
+ * Extracts normalized model name (removes provider prefix like 'openai/gpt-4o' -> 'gpt-4o').
+ */
+export declare function normalizeModelName(model: string): string;
+/**
+ * Candidate pricing keys for a model id, from most to least specific:
+ * `claude-3-7-sonnet-20250219` -> `claude-3-7-sonnet`, `gpt-4o-2024-08-06` -> `gpt-4o`,
+ * `claude-3-7-sonnet-latest` -> `claude-3-7-sonnet`.
+ */
+export declare function pricingKeyCandidates(model: string): string[];
+/**
+ * Looks up pricing in `table`, tolerating dated/`-latest` variants of a known model.
+ * Uses own-property lookups only, so keys like `__proto__` or `constructor` never match.
+ */
+export declare function lookupPricing(model: string, table: ReadonlyMap<string, ModelPricing> | Readonly<Record<string, ModelPricing>>): ModelPricing | undefined;
+/** Coerces untrusted numbers (NaN, Infinity, negatives, non-numbers) to a safe non-negative value. */
+export declare function toSafeTokens(value: unknown): number;
+/**
+ * Computes USD cost for a call.
+ *
+ * `inputTokens` is the total input (cache reads/writes included) and `outputTokens` is the total
+ * output (reasoning included), matching the AI SDK usage convention. Cached tokens are billed at
+ * their dedicated rates and removed from the regular input charge; reasoning tokens are billed at
+ * `reasoningCostPer1M` when set, otherwise at the output rate.
+ */
+export declare function computeCost(pricing: ModelPricing, inputTokens: number, outputTokens: number, details?: TokenUsageDetails): number;
+//# sourceMappingURL=pricing.d.ts.map

package/dist/src/budget/pricing.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"pricing.d.ts","sourceRoot":"","sources":["../../../src/budget/pricing.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAA;AAEjE;;;;;GAKG;AACH,eAAO,MAAM,qBAAqB,EAAE,QAAQ,CAAC,MAAM,CAAC,MAAM,EAAE,YAAY,CAAC,CA4ChC,CAAA;AAEzC;;GAEG;AACH,wBAAgB,kBAAkB,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM,CAGxD;AAED;;;;GAIG;AACH,wBAAgB,oBAAoB,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM,EAAE,CAU5D;AAED;;;GAGG;AACH,wBAAgB,aAAa,CAC3B,KAAK,EAAE,MAAM,EACb,KAAK,EAAE,WAAW,CAAC,MAAM,EAAE,YAAY,CAAC,GAAG,QAAQ,CAAC,MAAM,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC,GAChF,YAAY,GAAG,SAAS,CAsB1B;AAED,sGAAsG;AACtG,wBAAgB,YAAY,CAAC,KAAK,EAAE,OAAO,GAAG,MAAM,CAEnD;AAED;;;;;;;GAOG;AACH,wBAAgB,WAAW,CACzB,OAAO,EAAE,YAAY,EACrB,WAAW,EAAE,MAAM,EACnB,YAAY,EAAE,MAAM,EACpB,OAAO,GAAE,iBAAsB,GAC9B,MAAM,CAsBR"}

package/dist/src/budget/pricing.js

@@ -0,0 +1,129 @@
+/**
+ * Default pricing catalog for popular LLM models (USD per 1M tokens).
+ *
+ * Prices drift: treat this table as a fallback and prefer `CostTracker.setPricing()` /
+ * `registerPricingLoader()` (e.g. backed by models.dev) for authoritative values.
+ */
+export const DEFAULT_MODEL_PRICING = Object.freeze({
+    // OpenAI
+    'gpt-4.1': { input: 2.0, output: 8.0, cacheReadCostPer1M: 0.5 },
+    'gpt-4.1-mini': { input: 0.4, output: 1.6, cacheReadCostPer1M: 0.1 },
+    'gpt-4.1-nano': { input: 0.1, output: 0.4, cacheReadCostPer1M: 0.025 },
+    'gpt-4o': { input: 2.5, output: 10.0, cacheReadCostPer1M: 1.25 },
+    'gpt-4o-mini': { input: 0.15, output: 0.6, cacheReadCostPer1M: 0.075 },
+    'gpt-4-turbo': { input: 10.0, output: 30.0 },
+    'o1': { input: 15.0, output: 60.0, cacheReadCostPer1M: 7.5 },
+    'o1-mini': { input: 1.1, output: 4.4, cacheReadCostPer1M: 0.55 },
+    'o3': { input: 2.0, output: 8.0, cacheReadCostPer1M: 0.5 },
+    'o3-mini': { input: 1.1, output: 4.4, cacheReadCostPer1M: 0.55 },
+    'o4-mini': { input: 1.1, output: 4.4, cacheReadCostPer1M: 0.275 },
+    // Anthropic (cache read = 0.1x input, 5-minute cache write = 1.25x input)
+    'claude-opus-4-1': { input: 15.0, output: 75.0, cacheReadCostPer1M: 1.5, cacheWriteCostPer1M: 18.75 },
+    'claude-opus-4': { input: 15.0, output: 75.0, cacheReadCostPer1M: 1.5, cacheWriteCostPer1M: 18.75 },
+    'claude-sonnet-4-5': { input: 3.0, output: 15.0, cacheReadCostPer1M: 0.3, cacheWriteCostPer1M: 3.75 },
+    'claude-sonnet-4': { input: 3.0, output: 15.0, cacheReadCostPer1M: 0.3, cacheWriteCostPer1M: 3.75 },
+    'claude-haiku-4-5': { input: 1.0, output: 5.0, cacheReadCostPer1M: 0.1, cacheWriteCostPer1M: 1.25 },
+    'claude-3-7-sonnet': { input: 3.0, output: 15.0, cacheReadCostPer1M: 0.3, cacheWriteCostPer1M: 3.75 },
+    'claude-3-5-sonnet': { input: 3.0, output: 15.0, cacheReadCostPer1M: 0.3, cacheWriteCostPer1M: 3.75 },
+    'claude-3-5-haiku': { input: 0.8, output: 4.0, cacheReadCostPer1M: 0.08, cacheWriteCostPer1M: 1.0 },
+    'claude-haiku-3.5': { input: 0.8, output: 4.0, cacheReadCostPer1M: 0.08, cacheWriteCostPer1M: 1.0 },
+    'claude-3-opus': { input: 15.0, output: 75.0, cacheReadCostPer1M: 1.5, cacheWriteCostPer1M: 18.75 },
+    // Google (prices for prompts <= 200k tokens; reasoning is billed as output)
+    'gemini-2.5-pro': { input: 1.25, output: 10.0, cacheReadCostPer1M: 0.31 },
+    'gemini-2.5-flash': { input: 0.3, output: 2.5, cacheReadCostPer1M: 0.075 },
+    'gemini-2.5-flash-lite': { input: 0.1, output: 0.4, cacheReadCostPer1M: 0.025 },
+    'gemini-2.0-flash': { input: 0.1, output: 0.4, cacheReadCostPer1M: 0.025 },
+    'gemini-2.0-flash-lite': { input: 0.075, output: 0.3 },
+    'gemini-1.5-pro': { input: 1.25, output: 5.0 },
+    'gemini-1.5-flash': { input: 0.075, output: 0.3 },
+    // Groq / Open-source / Meta
+    'llama-3.3-70b-versatile': { input: 0.59, output: 0.79 },
+    'llama-3.3-70b': { input: 0.59, output: 0.79 },
+    'llama-3.1-70b': { input: 0.59, output: 0.79 },
+    'llama-3.1-8b': { input: 0.05, output: 0.08 },
+    // DeepSeek
+    'deepseek-chat': { input: 0.14, output: 0.28 },
+    'deepseek-reasoner': { input: 0.55, output: 2.19 },
+});
+/**
+ * Extracts normalized model name (removes provider prefix like 'openai/gpt-4o' -> 'gpt-4o').
+ */
+export function normalizeModelName(model) {
+    if (!model)
+        return '';
+    return model.split('/').pop() ?? model;
+}
+/**
+ * Candidate pricing keys for a model id, from most to least specific:
+ * `claude-3-7-sonnet-20250219` -> `claude-3-7-sonnet`, `gpt-4o-2024-08-06` -> `gpt-4o`,
+ * `claude-3-7-sonnet-latest` -> `claude-3-7-sonnet`.
+ */
+export function pricingKeyCandidates(model) {
+    const short = normalizeModelName(model);
+    const candidates = [model, short];
+    const stripped = short
+        .replace(/-latest$/, '')
+        .replace(/-\d{4}-\d{2}-\d{2}$/, '')
+        .replace(/-\d{8}$/, '')
+        .replace(/@\d{8}$/, '');
+    if (stripped !== short)
+        candidates.push(stripped);
+    return [...new Set(candidates.filter(Boolean))];
+}
+/**
+ * Looks up pricing in `table`, tolerating dated/`-latest` variants of a known model.
+ * Uses own-property lookups only, so keys like `__proto__` or `constructor` never match.
+ */
+export function lookupPricing(model, table) {
+    const get = (key) => table instanceof Map
+        ? table.get(key)
+        : Object.hasOwn(table, key)
+            ? table[key]
+            : undefined;
+    const candidates = pricingKeyCandidates(model);
+    for (const key of candidates) {
+        const hit = get(key);
+        if (hit)
+            return hit;
+    }
+    // Longest known key that is a dash-delimited prefix of the model (e.g. `gpt-4.1-2025-04-14`).
+    const short = candidates[candidates.length - 1];
+    const keys = table instanceof Map ? [...table.keys()] : Object.keys(table);
+    let best;
+    for (const key of keys) {
+        if (short.startsWith(`${key}-`) && (!best || key.length > best.length))
+            best = key;
+    }
+    return best ? get(best) : undefined;
+}
+/** Coerces untrusted numbers (NaN, Infinity, negatives, non-numbers) to a safe non-negative value. */
+export function toSafeTokens(value) {
+    return typeof value === 'number' && Number.isFinite(value) && value > 0 ? value : 0;
+}
+/**
+ * Computes USD cost for a call.
+ *
+ * `inputTokens` is the total input (cache reads/writes included) and `outputTokens` is the total
+ * output (reasoning included), matching the AI SDK usage convention. Cached tokens are billed at
+ * their dedicated rates and removed from the regular input charge; reasoning tokens are billed at
+ * `reasoningCostPer1M` when set, otherwise at the output rate.
+ */
+export function computeCost(pricing, inputTokens, outputTokens, details = {}) {
+    const input = toSafeTokens(inputTokens);
+    const output = toSafeTokens(outputTokens);
+    const cacheRead = Math.min(toSafeTokens(details.cacheReadTokens), input);
+    const cacheWrite = Math.min(toSafeTokens(details.cacheWriteTokens), input - cacheRead);
+    const reasoning = Math.min(toSafeTokens(details.reasoningTokens), output);
+    const cacheReadRate = pricing.cacheReadCostPer1M ?? pricing.cacheRead ?? pricing.input;
+    const cacheWriteRate = pricing.cacheWriteCostPer1M ?? pricing.cacheWrite ?? pricing.input;
+    const reasoningRate = pricing.reasoningCostPer1M ?? pricing.output;
+    const plainInput = input - cacheRead - cacheWrite;
+    const plainOutput = output - reasoning;
+    const total = plainInput * pricing.input +
+        cacheRead * cacheReadRate +
+        cacheWrite * cacheWriteRate +
+        plainOutput * pricing.output +
+        reasoning * reasoningRate;
+    return total / 1_000_000;
+}
+//# sourceMappingURL=pricing.js.map

package/dist/src/budget/pricing.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"pricing.js","sourceRoot":"","sources":["../../../src/budget/pricing.ts"],"names":[],"mappings":"AAEA;;;;;GAKG;AACH,MAAM,CAAC,MAAM,qBAAqB,GAA2C,MAAM,CAAC,MAAM,CAAC;IACzF,SAAS;IACT,SAAS,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,kBAAkB,EAAE,GAAG,EAAE;IAC/D,cAAc,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,kBAAkB,EAAE,GAAG,EAAE;IACpE,cAAc,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,kBAAkB,EAAE,KAAK,EAAE;IACtE,QAAQ,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE,kBAAkB,EAAE,IAAI,EAAE;IAChE,aAAa,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,EAAE,kBAAkB,EAAE,KAAK,EAAE;IACtE,aAAa,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE;IAC5C,IAAI,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,kBAAkB,EAAE,GAAG,EAAE;IAC5D,SAAS,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,kBAAkB,EAAE,IAAI,EAAE;IAChE,IAAI,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,kBAAkB,EAAE,GAAG,EAAE;IAC1D,SAAS,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,kBAAkB,EAAE,IAAI,EAAE;IAChE,SAAS,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,kBAAkB,EAAE,KAAK,EAAE;IAEjE,0EAA0E;IAC1E,iBAAiB,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,kBAAkB,EAAE,GAAG,EAAE,mBAAmB,EAAE,KAAK,EAAE;IACrG,eAAe,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,kBAAkB,EAAE,GAAG,EAAE,mBAAmB,EAAE,KAAK,EAAE;IACnG,mBAAmB,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE,kBAAkB,EAAE,GAAG,EAAE,mBAAmB,EAAE,IAAI,EAAE;IACrG,iBAAiB,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE,kBAAkB,EAAE,GAAG,EAAE,mBAAmB,EAAE,IAAI,EAAE;IACnG,kBAAkB,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,kBAAkB,EAAE,GAAG,EAAE,mBAAmB,EAAE,IAAI,EAAE;IACnG,mBAAmB,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE,kBAAkB,EAAE,GAAG,EAAE,mBAAmB,EAAE,IAAI,EAAE;IACrG,mBAAmB,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE,kBAAkB,EAAE,GAAG,EAAE,mBAAmB,EAAE,IAAI,EAAE;IACrG,kBAAkB,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,kBAAkB,EAAE,IAAI,EAAE,mBAAmB,EAAE,GAAG,EAAE;IACnG,kBAAkB,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,kBAAkB,EAAE,IAAI,EAAE,mBAAmB,EAAE,GAAG,EAAE;IACnG,eAAe,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,kBAAkB,EAAE,GAAG,EAAE,mBAAmB,EAAE,KAAK,EAAE;IAEnG,4EAA4E;IAC5E,gBAAgB,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,kBAAkB,EAAE,IAAI,EAAE;IACzE,kBAAkB,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,kBAAkB,EAAE,KAAK,EAAE;IAC1E,uBAAuB,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,kBAAkB,EAAE,KAAK,EAAE;IAC/E,kBAAkB,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,kBAAkB,EAAE,KAAK,EAAE;IAC1E,uBAAuB,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,EAAE;IACtD,gBAAgB,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,EAAE;IAC9C,kBAAkB,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,EAAE;IAEjD,4BAA4B;IAC5B,yBAAyB,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE;IACxD,eAAe,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE;IAC9C,eAAe,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE;IAC9C,cAAc,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE;IAE7C,WAAW;IACX,eAAe,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE;IAC9C,mBAAmB,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE;CACZ,CAAC,CAAA;AAEzC;;GAEG;AACH,MAAM,UAAU,kBAAkB,CAAC,KAAa;IAC9C,IAAI,CAAC,KAAK;QAAE,OAAO,EAAE,CAAA;IACrB,OAAO,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,IAAI,KAAK,CAAA;AACxC,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,oBAAoB,CAAC,KAAa;IAChD,MAAM,KAAK,GAAG,kBAAkB,CAAC,KAAK,CAAC,CAAA;IACvC,MAAM,UAAU,GAAG,CAAC,KAAK,EAAE,KAAK,CAAC,CAAA;IACjC,MAAM,QAAQ,GAAG,KAAK;SACnB,OAAO,CAAC,UAAU,EAAE,EAAE,CAAC;SACvB,OAAO,CAAC,qBAAqB,EAAE,EAAE,CAAC;SAClC,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC;SACtB,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC,CAAA;IACzB,IAAI,QAAQ,KAAK,KAAK;QAAE,UAAU,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;IACjD,OAAO,CAAC,GAAG,IAAI,GAAG,CAAC,UAAU,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAA;AACjD,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,aAAa,CAC3B,KAAa,EACb,KAAiF;IAEjF,MAAM,GAAG,GAAG,CAAC,GAAW,EAA4B,EAAE,CACpD,KAAK,YAAY,GAAG;QAClB,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC;QAChB,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,GAAG,CAAC;YACzB,CAAC,CAAE,KAAsC,CAAC,GAAG,CAAC;YAC9C,CAAC,CAAC,SAAS,CAAA;IAEjB,MAAM,UAAU,GAAG,oBAAoB,CAAC,KAAK,CAAC,CAAA;IAC9C,KAAK,MAAM,GAAG,IAAI,UAAU,EAAE,CAAC;QAC7B,MAAM,GAAG,GAAG,GAAG,CAAC,GAAG,CAAC,CAAA;QACpB,IAAI,GAAG;YAAE,OAAO,GAAG,CAAA;IACrB,CAAC;IAED,8FAA8F;IAC9F,MAAM,KAAK,GAAG,UAAU,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,CAAC,CAAA;IAC/C,MAAM,IAAI,GAAG,KAAK,YAAY,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;IAC1E,IAAI,IAAwB,CAAA;IAC5B,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;QACvB,IAAI,KAAK,CAAC,UAAU,CAAC,GAAG,GAAG,GAAG,CAAC,IAAI,CAAC,CAAC,IAAI,IAAI,GAAG,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;YAAE,IAAI,GAAG,GAAG,CAAA;IACpF,CAAC;IACD,OAAO,IAAI,CAAC,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS,CAAA;AACrC,CAAC;AAED,sGAAsG;AACtG,MAAM,UAAU,YAAY,CAAC,KAAc;IACzC,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,KAAK,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAA;AACrF,CAAC;AAED;;;;;;;GAOG;AACH,MAAM,UAAU,WAAW,CACzB,OAAqB,EACrB,WAAmB,EACnB,YAAoB,EACpB,UAA6B,EAAE;IAE/B,MAAM,KAAK,GAAG,YAAY,CAAC,WAAW,CAAC,CAAA;IACvC,MAAM,MAAM,GAAG,YAAY,CAAC,YAAY,CAAC,CAAA;IACzC,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,OAAO,CAAC,eAAe,CAAC,EAAE,KAAK,CAAC,CAAA;IACxE,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,OAAO,CAAC,gBAAgB,CAAC,EAAE,KAAK,GAAG,SAAS,CAAC,CAAA;IACtF,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,OAAO,CAAC,eAAe,CAAC,EAAE,MAAM,CAAC,CAAA;IAEzE,MAAM,aAAa,GAAG,OAAO,CAAC,kBAAkB,IAAI,OAAO,CAAC,SAAS,IAAI,OAAO,CAAC,KAAK,CAAA;IACtF,MAAM,cAAc,GAAG,OAAO,CAAC,mBAAmB,IAAI,OAAO,CAAC,UAAU,IAAI,OAAO,CAAC,KAAK,CAAA;IACzF,MAAM,aAAa,GAAG,OAAO,CAAC,kBAAkB,IAAI,OAAO,CAAC,MAAM,CAAA;IAElE,MAAM,UAAU,GAAG,KAAK,GAAG,SAAS,GAAG,UAAU,CAAA;IACjD,MAAM,WAAW,GAAG,MAAM,GAAG,SAAS,CAAA;IAEtC,MAAM,KAAK,GACT,UAAU,GAAG,OAAO,CAAC,KAAK;QAC1B,SAAS,GAAG,aAAa;QACzB,UAAU,GAAG,cAAc;QAC3B,WAAW,GAAG,OAAO,CAAC,MAAM;QAC5B,SAAS,GAAG,aAAa,CAAA;IAE3B,OAAO,KAAK,GAAG,SAAS,CAAA;AAC1B,CAAC"}

package/dist/src/budget/types.d.ts

@@ -0,0 +1,91 @@
+export interface Logger {
+    info(message: string, ...args: unknown[]): void;
+    warn(message: string, ...args: unknown[]): void;
+    error(message: string, ...args: unknown[]): void;
+    debug?(message: string, ...args: unknown[]): void;
+}
+export interface ModelPricing {
+    /** Price in USD per 1M input tokens (non-cached input) */
+    input: number;
+    /** Price in USD per 1M output tokens (reasoning tokens included unless `reasoningCostPer1M` is set) */
+    output: number;
+    /**
+     * Price in USD per 1M cached-read input tokens.
+     * When omitted, cached reads are billed at the regular `input` price (conservative).
+     */
+    cacheReadCostPer1M?: number;
+    /**
+     * Price in USD per 1M cache-write input tokens.
+     * When omitted, cache writes are billed at the regular `input` price (conservative).
+     */
+    cacheWriteCostPer1M?: number;
+    /**
+     * Price in USD per 1M reasoning tokens, when a provider bills them separately from output.
+     * When omitted, reasoning tokens are billed at the `output` price.
+     */
+    reasoningCostPer1M?: number;
+    /** @deprecated Use `cacheReadCostPer1M`. */
+    cacheRead?: number;
+    /** @deprecated Use `cacheWriteCostPer1M`. */
+    cacheWrite?: number;
+}
+/**
+ * Token breakdown used for cost calculation.
+ *
+ * Follows the AI SDK convention: `inputTokens` is the TOTAL input (cached tokens included)
+ * and `outputTokens` is the TOTAL output (reasoning tokens included).
+ */
+export interface TokenUsageDetails {
+    cacheReadTokens?: number;
+    cacheWriteTokens?: number;
+    reasoningTokens?: number;
+}
+export interface AgentCostRecord {
+    agentName: string;
+    model: string;
+    inputTokens: number;
+    outputTokens: number;
+    cacheReadTokens?: number;
+    cacheWriteTokens?: number;
+    reasoningTokens?: number;
+    totalTokens: number;
+    durationMs: number;
+    costUsd: number;
+    timestamp: number;
+    metadata?: Record<string, unknown>;
+}
+export interface AgentCostSummary {
+    tokens: number;
+    cost: number;
+    calls: number;
+    avgDurationMs: number;
+    totalDurationMs: number;
+}
+export interface BudgetModifier {
+    factor: string;
+    pctChange: number;
+}
+export interface BudgetAllocation {
+    perAgent: number;
+    perSession: number;
+    modifiers: BudgetModifier[];
+    totalModifierPct: number;
+    reason: string;
+}
+export interface BudgetEnforcerOptions {
+    perAgentMaxTokens?: number;
+    perSessionMaxTokens?: number;
+    perAgentMaxCostUsd?: number;
+    perSessionMaxCostUsd?: number;
+}
+export interface BudgetSummary {
+    sessionTokensUsed: number;
+    sessionMaxTokens: number;
+    sessionCostUsd: number;
+    sessionMaxCostUsd: number;
+    agentUsage: Record<string, {
+        tokens: number;
+        costUsd: number;
+    }>;
+}
+//# sourceMappingURL=types.d.ts.map

package/dist/src/budget/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../src/budget/types.ts"],"names":[],"mappings":"AAAA,MAAM,WAAW,MAAM;IACrB,IAAI,CAAC,OAAO,EAAE,MAAM,EAAE,GAAG,IAAI,EAAE,OAAO,EAAE,GAAG,IAAI,CAAA;IAC/C,IAAI,CAAC,OAAO,EAAE,MAAM,EAAE,GAAG,IAAI,EAAE,OAAO,EAAE,GAAG,IAAI,CAAA;IAC/C,KAAK,CAAC,OAAO,EAAE,MAAM,EAAE,GAAG,IAAI,EAAE,OAAO,EAAE,GAAG,IAAI,CAAA;IAChD,KAAK,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,GAAG,IAAI,EAAE,OAAO,EAAE,GAAG,IAAI,CAAA;CAClD;AAED,MAAM,WAAW,YAAY;IAC3B,0DAA0D;IAC1D,KAAK,EAAE,MAAM,CAAA;IACb,uGAAuG;IACvG,MAAM,EAAE,MAAM,CAAA;IACd;;;OAGG;IACH,kBAAkB,CAAC,EAAE,MAAM,CAAA;IAC3B;;;OAGG;IACH,mBAAmB,CAAC,EAAE,MAAM,CAAA;IAC5B;;;OAGG;IACH,kBAAkB,CAAC,EAAE,MAAM,CAAA;IAC3B,4CAA4C;IAC5C,SAAS,CAAC,EAAE,MAAM,CAAA;IAClB,6CAA6C;IAC7C,UAAU,CAAC,EAAE,MAAM,CAAA;CACpB;AAED;;;;;GAKG;AACH,MAAM,WAAW,iBAAiB;IAChC,eAAe,CAAC,EAAE,MAAM,CAAA;IACxB,gBAAgB,CAAC,EAAE,MAAM,CAAA;IACzB,eAAe,CAAC,EAAE,MAAM,CAAA;CACzB;AAED,MAAM,WAAW,eAAe;IAC9B,SAAS,EAAE,MAAM,CAAA;IACjB,KAAK,EAAE,MAAM,CAAA;IACb,WAAW,EAAE,MAAM,CAAA;IACnB,YAAY,EAAE,MAAM,CAAA;IACpB,eAAe,CAAC,EAAE,MAAM,CAAA;IACxB,gBAAgB,CAAC,EAAE,MAAM,CAAA;IACzB,eAAe,CAAC,EAAE,MAAM,CAAA;IACxB,WAAW,EAAE,MAAM,CAAA;IACnB,UAAU,EAAE,MAAM,CAAA;IAClB,OAAO,EAAE,MAAM,CAAA;IACf,SAAS,EAAE,MAAM,CAAA;IACjB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;CACnC;AAED,MAAM,WAAW,gBAAgB;IAC/B,MAAM,EAAE,MAAM,CAAA;IACd,IAAI,EAAE,MAAM,CAAA;IACZ,KAAK,EAAE,MAAM,CAAA;IACb,aAAa,EAAE,MAAM,CAAA;IACrB,eAAe,EAAE,MAAM,CAAA;CACxB;AAED,MAAM,WAAW,cAAc;IAC7B,MAAM,EAAE,MAAM,CAAA;IACd,SAAS,EAAE,MAAM,CAAA;CAClB;AAED,MAAM,WAAW,gBAAgB;IAC/B,QAAQ,EAAE,MAAM,CAAA;IAChB,UAAU,EAAE,MAAM,CAAA;IAClB,SAAS,EAAE,cAAc,EAAE,CAAA;IAC3B,gBAAgB,EAAE,MAAM,CAAA;IACxB,MAAM,EAAE,MAAM,CAAA;CACf;AAED,MAAM,WAAW,qBAAqB;IACpC,iBAAiB,CAAC,EAAE,MAAM,CAAA;IAC1B,mBAAmB,CAAC,EAAE,MAAM,CAAA;IAC5B,kBAAkB,CAAC,EAAE,MAAM,CAAA;IAC3B,oBAAoB,CAAC,EAAE,MAAM,CAAA;CAC9B;AAED,MAAM,WAAW,aAAa;IAC5B,iBAAiB,EAAE,MAAM,CAAA;IACzB,gBAAgB,EAAE,MAAM,CAAA;IACxB,cAAc,EAAE,MAAM,CAAA;IACtB,iBAAiB,EAAE,MAAM,CAAA;IACzB,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE;QAAE,MAAM,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAA;KAAE,CAAC,CAAA;CAChE"}

package/dist/src/budget/types.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=types.js.map

package/dist/src/budget/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../src/budget/types.ts"],"names":[],"mappings":""}

package/dist/src/catalog/index.d.ts

@@ -0,0 +1,3 @@
+export * from './models_dev.js';
+export * from './source.js';
+//# sourceMappingURL=index.d.ts.map

package/dist/src/catalog/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/catalog/index.ts"],"names":[],"mappings":"AAAA,cAAc,iBAAiB,CAAA;AAC/B,cAAc,aAAa,CAAA"}

package/dist/src/catalog/index.js

@@ -0,0 +1,3 @@
+export * from './models_dev.js';
+export * from './source.js';
+//# sourceMappingURL=index.js.map

package/dist/src/catalog/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/catalog/index.ts"],"names":[],"mappings":"AAAA,cAAc,iBAAiB,CAAA;AAC/B,cAAc,aAAa,CAAA"}

package/dist/src/catalog/models_dev.d.ts

@@ -0,0 +1,87 @@
+import type { ModelPricing } from '../budget/types.js';
+import { type ModelsDevLoadOptions, type ModelsDevRefreshOptions } from './source.js';
+export interface ModelsDevModelLimit {
+    context?: number;
+    output?: number;
+}
+export interface ModelsDevModelCost {
+    input?: number;
+    output?: number;
+    cache_read?: number;
+    cache_write?: number;
+    reasoning?: number;
+}
+export interface ModelsDevModelModalities {
+    input?: string[];
+    output?: string[];
+}
+export interface ModelsDevModel {
+    id: string;
+    name: string;
+    description?: string;
+    family?: string;
+    tool_call?: boolean;
+    structured_output?: boolean;
+    attachment?: boolean;
+    reasoning?: boolean;
+    temperature?: boolean;
+    knowledge?: string;
+    release_date?: string;
+    last_updated?: string;
+    modalities?: ModelsDevModelModalities;
+    open_weights?: boolean;
+    limit?: ModelsDevModelLimit;
+    cost?: ModelsDevModelCost;
+    headers?: Record<string, string>;
+    [key: string]: unknown;
+}
+export interface ModelsDevProvider {
+    id: string;
+    name: string;
+    env: string[];
+    api?: string;
+    npm?: string;
+    protocol?: string;
+    models: Record<string, ModelsDevModel>;
+    [key: string]: unknown;
+}
+export type ModelsDevCatalog = Record<string, ModelsDevProvider>;
+export interface ModelCatalogEntry {
+    id: string;
+    name: string;
+    providerId: string;
+    providerName: string;
+    contextLimit: number;
+    outputLimit: number;
+    toolCall: boolean;
+    structuredOutput: boolean;
+    pricing?: ModelPricing;
+    description?: string;
+    headers?: Record<string, string>;
+    rawModel?: ModelsDevModel;
+}
+/**
+ * Loads the models.dev catalog using in-memory and disk cache.
+ */
+export declare function loadModelsDev(options?: ModelsDevLoadOptions): Promise<ModelsDevCatalog>;
+/**
+ * Refreshes the models.dev catalog from remote source or validates local cache.
+ */
+export declare function refreshModelsDev(options?: ModelsDevRefreshOptions): Promise<ModelsDevCatalog>;
+/**
+ * Resets in-memory cached catalog.
+ */
+export declare function resetModelsDevCache(): void;
+/**
+ * Looks up a model by provider ID and model ID.
+ */
+export declare function findModel(providerId: string, modelId: string): Promise<ModelCatalogEntry | undefined>;
+/**
+ * Lists all providers in the catalog.
+ */
+export declare function listProviders(): Promise<ModelsDevProvider[]>;
+/**
+ * Lists models, optionally filtered by provider.
+ */
+export declare function listModels(providerId?: string): Promise<ModelCatalogEntry[]>;
+//# sourceMappingURL=models_dev.d.ts.map

package/dist/src/catalog/models_dev.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"models_dev.d.ts","sourceRoot":"","sources":["../../../src/catalog/models_dev.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAA;AACtD,OAAO,EAAmB,KAAK,oBAAoB,EAAE,KAAK,uBAAuB,EAAE,MAAM,aAAa,CAAA;AAEtG,MAAM,WAAW,mBAAmB;IAClC,OAAO,CAAC,EAAE,MAAM,CAAA;IAChB,MAAM,CAAC,EAAE,MAAM,CAAA;CAChB;AAED,MAAM,WAAW,kBAAkB;IACjC,KAAK,CAAC,EAAE,MAAM,CAAA;IACd,MAAM,CAAC,EAAE,MAAM,CAAA;IACf,UAAU,CAAC,EAAE,MAAM,CAAA;IACnB,WAAW,CAAC,EAAE,MAAM,CAAA;IACpB,SAAS,CAAC,EAAE,MAAM,CAAA;CACnB;AAED,MAAM,WAAW,wBAAwB;IACvC,KAAK,CAAC,EAAE,MAAM,EAAE,CAAA;IAChB,MAAM,CAAC,EAAE,MAAM,EAAE,CAAA;CAClB;AAED,MAAM,WAAW,cAAc;IAC7B,EAAE,EAAE,MAAM,CAAA;IACV,IAAI,EAAE,MAAM,CAAA;IACZ,WAAW,CAAC,EAAE,MAAM,CAAA;IACpB,MAAM,CAAC,EAAE,MAAM,CAAA;IACf,SAAS,CAAC,EAAE,OAAO,CAAA;IACnB,iBAAiB,CAAC,EAAE,OAAO,CAAA;IAC3B,UAAU,CAAC,EAAE,OAAO,CAAA;IACpB,SAAS,CAAC,EAAE,OAAO,CAAA;IACnB,WAAW,CAAC,EAAE,OAAO,CAAA;IACrB,SAAS,CAAC,EAAE,MAAM,CAAA;IAClB,YAAY,CAAC,EAAE,MAAM,CAAA;IACrB,YAAY,CAAC,EAAE,MAAM,CAAA;IACrB,UAAU,CAAC,EAAE,wBAAwB,CAAA;IACrC,YAAY,CAAC,EAAE,OAAO,CAAA;IACtB,KAAK,CAAC,EAAE,mBAAmB,CAAA;IAC3B,IAAI,CAAC,EAAE,kBAAkB,CAAA;IACzB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;IAChC,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAA;CACvB;AAED,MAAM,WAAW,iBAAiB;IAChC,EAAE,EAAE,MAAM,CAAA;IACV,IAAI,EAAE,MAAM,CAAA;IACZ,GAAG,EAAE,MAAM,EAAE,CAAA;IACb,GAAG,CAAC,EAAE,MAAM,CAAA;IACZ,GAAG,CAAC,EAAE,MAAM,CAAA;IACZ,QAAQ,CAAC,EAAE,MAAM,CAAA;IACjB,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,cAAc,CAAC,CAAA;IACtC,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAA;CACvB;AAED,MAAM,MAAM,gBAAgB,GAAG,MAAM,CAAC,MAAM,EAAE,iBAAiB,CAAC,CAAA;AAEhE,MAAM,WAAW,iBAAiB;IAChC,EAAE,EAAE,MAAM,CAAA;IACV,IAAI,EAAE,MAAM,CAAA;IACZ,UAAU,EAAE,MAAM,CAAA;IAClB,YAAY,EAAE,MAAM,CAAA;IACpB,YAAY,EAAE,MAAM,CAAA;IACpB,WAAW,EAAE,MAAM,CAAA;IACnB,QAAQ,EAAE,OAAO,CAAA;IACjB,gBAAgB,EAAE,OAAO,CAAA;IACzB,OAAO,CAAC,EAAE,YAAY,CAAA;IACtB,WAAW,CAAC,EAAE,MAAM,CAAA;IACpB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;IAChC,QAAQ,CAAC,EAAE,cAAc,CAAA;CAC1B;AAED;;GAEG;AACH,wBAAsB,aAAa,CAAC,OAAO,CAAC,EAAE,oBAAoB,GAAG,OAAO,CAAC,gBAAgB,CAAC,CAE7F;AAED;;GAEG;AACH,wBAAsB,gBAAgB,CAAC,OAAO,CAAC,EAAE,uBAAuB,GAAG,OAAO,CAAC,gBAAgB,CAAC,CAEnG;AAED;;GAEG;AACH,wBAAgB,mBAAmB,IAAI,IAAI,CAE1C;AAED;;GAEG;AACH,wBAAsB,SAAS,CAC7B,UAAU,EAAE,MAAM,EAClB,OAAO,EAAE,MAAM,GACd,OAAO,CAAC,iBAAiB,GAAG,SAAS,CAAC,CAExC;AAED;;GAEG;AACH,wBAAsB,aAAa,IAAI,OAAO,CAAC,iBAAiB,EAAE,CAAC,CAElE;AAED;;GAEG;AACH,wBAAsB,UAAU,CAAC,UAAU,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,iBAAiB,EAAE,CAAC,CAElF"}