@cendor/tokenguard 0.2.0

package/dist/index.js

@@ -0,0 +1,607 @@
+/**
+ * `@cendor/tokenguard` — pre-flight cost caps + free per-tag spend attribution for LLM calls. The
+ * TypeScript port of `cendor.tokenguard`.
+ *
+ * It **subscribes** to `@cendor/core`'s event bus and registers a pre-flight interceptor; it never
+ * patches a client itself (the locked architecture: one `instrument()`, many subscribers). Once a
+ * client is instrumented, `budget(...)` enforces a cap and `track(...)` attributes spend by tags —
+ * with zero per-call wiring.
+ *
+ * Parity notes vs. the Python original:
+ * - Python `contextvars` become two `node:async_hooks` `AsyncLocalStorage` scopes (`_tags`,
+ *   `_budgets`) — they propagate across `await` in the same async task, but NOT across worker
+ *   threads (same caveat as Python's contextvars vs. OS threads).
+ * - Python `with budget(...) as b:` / `@budget(...)` become {@link withBudget} (async-callback
+ *   scope) and {@link budget} (decorator: `budget(cfg)(fn)`); `with track(...)` becomes
+ *   {@link track} (async-callback scope). `track.report` is attached as an ergonomic alias.
+ * - `warnings.warn(UnpricedModelWarning)` becomes a capturable/escalatable warning channel:
+ *   register a listener via {@link onUnpricedWarning}; with no listener it falls back to
+ *   `console.warn`. Deduped once-per-model (cleared by {@link reset}).
+ */
+import { AsyncLocalStorage } from 'node:async_hooks';
+import { Dec, LLMCall, MISS, Money, Reroute, UnknownModelError, addInterceptor, bus, prices, tokens, } from '@cendor/core';
+// --------------------------------------------------------------------------- constants
+/** Output tokens are unknown pre-flight; reserve this many for the projection unless the request
+ * carries an explicit `max_tokens` / `max_completion_tokens`. */
+const DEFAULT_OUTPUT_RESERVE = 256;
+/** Valid string values for `budget({ onExceed })` (a callable is also accepted). */
+const ON_EXCEED = ['raise', 'block', 'truncate', 'downgrade', 'clamp'];
+/** Per-provider request kwarg that caps generated (reasoning + visible) output tokens, used by
+ * `onExceed: 'clamp'`. Providers absent here can't have the cap injected safely (clamp falls back
+ * to a hard block). */
+const CLAMP_KWARG = {
+    openai: 'max_completion_tokens',
+    anthropic: 'max_tokens',
+};
+const DEFAULT_MAX_RECORDS = 100_000;
+const DEFAULT_ON_UNPRICED = 'warn';
+// --------------------------------------------------------------------------- errors + warnings
+/** Raised when a call pushes an active budget over its cap (post-flight `raise`, or pre-flight
+ * `block` / `clamp`-fallback). */
+export class BudgetExceeded extends Error {
+    constructor(message) {
+        super(message);
+        this.name = 'BudgetExceeded';
+    }
+}
+/** Warned (once per model) when a USD budget is active but the call's model has no price — an
+ * unpriced model records `$0` toward USD spend, so a USD-only cap can't enforce against it. */
+export class UnpricedModelWarning extends Error {
+    constructor(message) {
+        super(message);
+        this.name = 'UnpricedModelWarning';
+    }
+}
+/** Internal graceful-degradation signal for `onExceed: 'truncate'`; never escapes a budget. */
+class Truncated extends Error {
+    constructor() {
+        super('truncated');
+        this.name = 'Truncated';
+    }
+}
+/** Internal — mirrors Python's `ValueError` for eager config validation (tests match a substring). */
+class ValueError extends Error {
+    constructor(message) {
+        super(message);
+        this.name = 'ValueError';
+    }
+}
+/** Internal — mirrors Python's `AssertionError` for {@link Report.assertUnder}. */
+class AssertionError extends Error {
+    constructor(message) {
+        super(message);
+        this.name = 'AssertionError';
+    }
+}
+const warningListeners = new Set();
+/**
+ * Register a listener for {@link UnpricedModelWarning}s (the TS analog of a Python warning filter).
+ * Returns an unsubscribe function. With at least one listener installed the warning is delivered to
+ * the listeners (and NOT to `console.warn`); a listener may re-throw to escalate the warning to an
+ * error, exactly like `simplefilter("error", UnpricedModelWarning)`.
+ */
+export function onUnpricedWarning(listener) {
+    warningListeners.add(listener);
+    return () => {
+        warningListeners.delete(listener);
+    };
+}
+/** One active budget (mutable — the post-flight subscriber accumulates spend into it). */
+export class Frame {
+    capUsd;
+    capTokens;
+    onExceed;
+    scope;
+    downgrade;
+    outputReserve;
+    reasoningReserve;
+    spentUsd = new Dec(0);
+    spentTokens = 0;
+    calls = 0;
+    constructor(init) {
+        this.capUsd = init.capUsd;
+        this.capTokens = init.capTokens;
+        this.onExceed = init.onExceed;
+        this.scope = init.scope;
+        this.downgrade = init.downgrade;
+        this.outputReserve = init.outputReserve;
+        this.reasoningReserve = init.reasoningReserve;
+    }
+}
+// --------------------------------------------------------------------------- module state
+const tagsStore = new AsyncLocalStorage();
+const budgetsStore = new AsyncLocalStorage();
+const records = [];
+const downgradeRows = [];
+const clampRows = [];
+let sink = null;
+let maxRecords = DEFAULT_MAX_RECORDS;
+let droppedCount = 0;
+let onUnpriced = DEFAULT_ON_UNPRICED;
+const warnedUnpriced = new Set();
+function currentTags() {
+    return tagsStore.getStore() ?? {};
+}
+function currentFrames() {
+    return budgetsStore.getStore() ?? [];
+}
+function warnUnpriced(model, mode) {
+    if (warnedUnpriced.has(model))
+        return;
+    warnedUnpriced.add(model);
+    const warning = new UnpricedModelWarning(`tokenguard: no price for model '${model}', so the active USD budget (on_exceed='${mode}') counts its calls as $0 and cannot enforce a USD cap on it. Add a rate (cendor.core.prices), use a tokens= cap instead, or configure(on_unpriced='raise') to reject unpriced calls under on_exceed='block'.`);
+    if (warningListeners.size === 0) {
+        console.warn(warning.message);
+        return;
+    }
+    for (const listener of warningListeners)
+        listener(warning);
+}
+function ensureSubscribed() {
+    bus.subscribe(onCall); // idempotent on the bus side
+    addInterceptor(_preflightInterceptor); // idempotent; pre-flight downgrade/clamp/block routing
+}
+// --------------------------------------------------------------------------- projection helpers
+/**
+ * Output tokens to assume pre-flight. Prefers the request's explicit output cap
+ * (`max_completion_tokens`, else `max_tokens`) using `!= null` checks so `max_tokens: 0` is honored
+ * (project 0 output), NOT treated as unset. Otherwise assumes `reserve + reasoningReserve`.
+ *
+ * Internal, but exported for parity with the Python test that asserts its behavior directly.
+ */
+export function _projectedOutput(call, reserve, reasoningReserve = 0) {
+    const kwargs = call.metadata.request_kwargs ?? {};
+    let explicit = kwargs.max_completion_tokens;
+    if (explicit == null)
+        explicit = kwargs.max_tokens;
+    if (explicit != null)
+        return Math.trunc(Number(explicit));
+    return reserve + reasoningReserve;
+}
+/** Project a call's cost pre-flight from its model + messages (+ an output reserve). May throw
+ * `UnknownModelError` (the KeyError-equivalent) for an unpriced model. */
+function estimateEvent(call, reserve, reasoningReserve = 0) {
+    const inputTokens = tokens.count(call.messages, call.model);
+    const projected = _projectedOutput(call, reserve, reasoningReserve);
+    return prices.estimate(call.model, inputTokens, { outputTokens: projected }).amount;
+}
+/** Pre-flight token projection: input tokens + the output reserve (max_tokens or default). */
+function projectTokens(call, reserve, reasoningReserve = 0) {
+    return (tokens.count(call.messages, call.model) + _projectedOutput(call, reserve, reasoningReserve));
+}
+// --------------------------------------------------------------------------- pre-flight interceptor
+/**
+ * Pre-flight enforcement, before the call runs: reroute (`downgrade`), clamp (`clamp`), or block
+ * (`block`). `raise`/`truncate`/callable are ignored here (post-flight only). Iterates frames
+ * innermost-first.
+ *
+ * Internal, but exported so tests can drive it directly on a synthesized `LLMCall` (the TS core's
+ * `instrument()` only structurally detects openai/anthropic clients, so an "ollama-shaped" client
+ * isn't wrapped — this exercises the same clamp-fallback code path).
+ */
+export function _preflightInterceptor(call) {
+    if (!(call instanceof LLMCall))
+        return MISS;
+    const frames = currentFrames();
+    for (let i = frames.length - 1; i >= 0; i--) {
+        const frame = frames[i];
+        if (frame.onExceed === 'downgrade' && frame.downgrade && frame.capUsd !== null) {
+            const cheaper = frame.downgrade[call.model];
+            if (!cheaper)
+                continue;
+            let projected;
+            try {
+                projected = estimateEvent(call, frame.outputReserve, frame.reasoningReserve);
+            }
+            catch (err) {
+                if (err instanceof UnknownModelError) {
+                    warnUnpriced(call.model, 'downgrade'); // can't project — no longer silent
+                    continue; // unknown model price — leave the call as-is
+                }
+                throw err;
+            }
+            if (frame.spentUsd.plus(projected).greaterThan(frame.capUsd)) {
+                downgradeRows.push({ from: call.model, to: cheaper, tags: { ...currentTags() } });
+                return new Reroute({ model: cheaper });
+            }
+        }
+        else if (frame.onExceed === 'clamp') {
+            const reroute = clamp(call, frame);
+            if (reroute !== null)
+                return reroute;
+        }
+        else if (frame.onExceed === 'block') {
+            const projTokens = projectTokens(call, frame.outputReserve, frame.reasoningReserve);
+            if (frame.capTokens !== null && frame.spentTokens + projTokens > frame.capTokens) {
+                throw new BudgetExceeded(`pre-flight block: ~${frame.spentTokens + projTokens} tokens would exceed cap ${frame.capTokens} (model=${call.model})`);
+            }
+            if (frame.capUsd !== null) {
+                let projected;
+                try {
+                    projected = estimateEvent(call, frame.outputReserve, frame.reasoningReserve);
+                }
+                catch (err) {
+                    if (err instanceof UnknownModelError) {
+                        if (onUnpriced === 'raise') {
+                            throw new BudgetExceeded(`pre-flight block: model=${call.model} has no price, so a USD cap cannot be projected; configure(on_unpriced='raise') rejects unpriced calls (set on_unpriced='warn' to let them through as $0).`);
+                        }
+                        warnUnpriced(call.model, 'block');
+                        continue;
+                    }
+                    throw err;
+                }
+                if (frame.spentUsd.plus(projected).greaterThan(frame.capUsd)) {
+                    throw new BudgetExceeded(`pre-flight block: projected $${frame.spentUsd.plus(projected)} would exceed cap $${frame.capUsd} (model=${call.model})`);
+                }
+            }
+        }
+    }
+    return MISS;
+}
+/**
+ * Inject a provider output ceiling so a single call can't exceed the remaining token budget.
+ * Requires a `tokens=` cap; a no-op when comfortably under; falls back to a hard block when it
+ * would breach but no ceiling can be injected safely (unknown provider, or no room even for input).
+ */
+function clamp(call, frame) {
+    if (frame.capTokens === null)
+        return null;
+    const projectedInput = tokens.count(call.messages, call.model);
+    const projOutput = _projectedOutput(call, frame.outputReserve, frame.reasoningReserve);
+    if (frame.spentTokens + projectedInput + projOutput <= frame.capTokens)
+        return null;
+    const allowance = frame.capTokens - frame.spentTokens - projectedInput;
+    const kwarg = CLAMP_KWARG[call.provider];
+    if (kwarg === undefined || allowance <= 0) {
+        throw new BudgetExceeded(`pre-flight clamp: cannot fit call within the remaining token budget (~${frame.capTokens - frame.spentTokens} left, ~${projectedInput} input; provider='${call.provider}', model=${call.model}) — use on_exceed='block' to reject, or raise the cap`);
+    }
+    const existingRaw = call.metadata.request_kwargs?.[kwarg];
+    const existing = existingRaw == null ? null : Math.trunc(Number(existingRaw));
+    if (existing !== null && existing <= allowance)
+        return null; // caller's own cap already fits
+    const target = existing === null ? allowance : Math.min(existing, allowance);
+    clampRows.push({ model: call.model, kwarg, limit: target, tags: { ...currentTags() } });
+    return new Reroute({ [kwarg]: target });
+}
+// --------------------------------------------------------------------------- post-flight subscriber
+/** Bus subscriber: record spend by active tags and enforce active budgets (post-flight). */
+function onCall(call) {
+    if (!(call instanceof LLMCall))
+        return; // tokenguard only accounts for model calls
+    const unpriced = call.cost === null; // no cost -> unknown/unpriced model, a USD blind spot
+    const usd = call.cost !== null ? call.cost.amount : new Dec(0);
+    const inp = call.usage !== null ? call.usage.inputTokens : 0;
+    const out = call.usage !== null ? call.usage.outputTokens : 0;
+    const rsn = call.usage !== null ? call.usage.reasoningTokens : 0;
+    const frames = currentFrames();
+    if (unpriced) {
+        // A USD-cap budget can't enforce against a $0-recorded call. Warn once per model, naming the
+        // innermost USD-cap frame's mode. (block/downgrade already warned pre-flight; this covers the
+        // post-flight modes: raise/truncate/callable.)
+        let usdFrame = null;
+        for (let i = frames.length - 1; i >= 0; i--) {
+            if (frames[i].capUsd !== null) {
+                usdFrame = frames[i];
+                break;
+            }
+        }
+        if (usdFrame !== null) {
+            const mode = usdFrame.onExceed;
+            warnUnpriced(call.model, typeof mode === 'string' ? mode : 'callable');
+        }
+    }
+    const tags = { ...currentTags() };
+    records.push({
+        tags,
+        usd,
+        inputTokens: inp,
+        outputTokens: out,
+        reasoningTokens: rsn,
+        model: call.model,
+        calls: 1,
+        unpriced,
+    });
+    // append + FIFO eviction — a read-modify-write on shared state. JS is single-threaded so no lock
+    // is needed; append/evict is atomic within a single emit.
+    if (maxRecords !== null && records.length > maxRecords) {
+        const overflow = records.length - maxRecords;
+        records.splice(0, overflow); // evict oldest (FIFO); counted, never silently
+        droppedCount += overflow;
+    }
+    if (sink !== null) {
+        sink.write({
+            tags,
+            usd: usd.toString(),
+            input_tokens: inp,
+            output_tokens: out,
+            reasoning_tokens: rsn,
+            model: call.model,
+        });
+    }
+    for (const frame of frames) {
+        frame.spentUsd = frame.spentUsd.plus(usd);
+        frame.spentTokens += inp + out;
+        frame.calls += 1;
+    }
+    for (let i = frames.length - 1; i >= 0; i--) {
+        const frame = frames[i]; // enforce the tightest (innermost) breached cap first
+        if (over(frame)) {
+            if (frame.onExceed === 'downgrade' || frame.onExceed === 'clamp') {
+                continue; // handled pre-flight; a no-op here must not mask an outer cap's action
+            }
+            enforce(frame, call);
+            break;
+        }
+    }
+}
+function over(frame) {
+    if (frame.capUsd !== null && frame.spentUsd.greaterThan(frame.capUsd))
+        return true;
+    if (frame.capTokens !== null && frame.spentTokens > frame.capTokens)
+        return true;
+    return false;
+}
+function enforce(frame, call) {
+    const mode = frame.onExceed;
+    if (typeof mode === 'function') {
+        mode({ frame, call, spentUsd: frame.spentUsd, capUsd: frame.capUsd });
+        return;
+    }
+    if (mode === 'truncate')
+        throw new Truncated();
+    if (mode === 'downgrade' || mode === 'clamp')
+        return; // already handled pre-flight
+    throw new BudgetExceeded(`budget exceeded: spent $${frame.spentUsd} > cap $${frame.capUsd} after ${frame.calls} call(s); last model=${call.model}. on_exceed='raise' is post-flight, so the cap is crossed by this one in-flight call — use on_exceed='block' for a pre-flight hard cap that never overspends.`);
+}
+function validateBudgetConfig(cfg) {
+    const onExceedValue = cfg.onExceed ?? 'raise';
+    if (typeof onExceedValue !== 'function' &&
+        !ON_EXCEED.includes(onExceedValue)) {
+        throw new ValueError(`on_exceed must be a callable or one of ${ON_EXCEED.join(',')}, got '${String(onExceedValue)}'`);
+    }
+    if (cfg.usd == null && cfg.tokens == null) {
+        throw new ValueError('budget requires a cap: pass usd= and/or tokens=');
+    }
+    if (onExceedValue === 'downgrade') {
+        if (!cfg.downgrade || Object.keys(cfg.downgrade).length === 0) {
+            throw new ValueError("on_exceed='downgrade' requires a downgrade={model: cheaper} map");
+        }
+        if (cfg.usd == null) {
+            throw new ValueError("on_exceed='downgrade' requires a usd= cap (the projection is USD-based)");
+        }
+    }
+    if (onExceedValue === 'clamp' && cfg.tokens == null) {
+        throw new ValueError("on_exceed='clamp' requires a tokens= cap (it injects a provider token ceiling)");
+    }
+}
+function makeFrame(cfg) {
+    return new Frame({
+        capUsd: cfg.usd == null ? null : new Dec(String(cfg.usd)),
+        capTokens: cfg.tokens ?? null,
+        onExceed: cfg.onExceed ?? 'raise',
+        scope: cfg.scope ?? null,
+        downgrade: cfg.downgrade ?? null,
+        outputReserve: cfg.outputReserve ?? DEFAULT_OUTPUT_RESERVE,
+        reasoningReserve: cfg.reasoningReserve ?? 0,
+    });
+}
+/** A handle to an open budget scope, passed to the {@link withBudget} callback. */
+export class BudgetHandle {
+    frame;
+    constructor(frame) {
+        this.frame = frame;
+    }
+    /** Spend recorded against this budget so far. Meaningful inside the `withBudget` callback. */
+    get spent() {
+        return new Money(this.frame.spentUsd);
+    }
+}
+/**
+ * Cap spend on a decorated function (the parity of Python's `@budget(...)`). Validates its
+ * configuration eagerly (throws on a missing cap / unknown `onExceed` / bad `downgrade` or
+ * `clamp`), then returns a wrapper that opens a fresh budget frame per invocation.
+ *
+ * The TS core is async-first, so the returned wrapper is always async (model calls are async);
+ * on `onExceed: 'truncate'` it resolves to `undefined` (graceful degradation).
+ */
+export function budget(cfg) {
+    validateBudgetConfig(cfg);
+    return (fn) => async (...args) => {
+        ensureSubscribed();
+        const frame = makeFrame(cfg);
+        const frames = [...currentFrames(), frame];
+        try {
+            return await budgetsStore.run(frames, () => fn(...args));
+        }
+        catch (err) {
+            if (err instanceof Truncated)
+                return undefined; // degraded gracefully instead of crashing
+            throw err;
+        }
+    };
+}
+/**
+ * Cap spend within an async-callback scope (the parity of Python's `with budget(...) as b:`).
+ * Validates eagerly, opens a fresh frame, runs `cb` inside the budget's `AsyncLocalStorage` scope so
+ * enforcement applies to every instrumented call made inside (including across awaits). On
+ * `onExceed: 'truncate'` it resolves to `undefined`; all other errors (including `BudgetExceeded`)
+ * propagate.
+ */
+export async function withBudget(cfg, cb) {
+    validateBudgetConfig(cfg);
+    ensureSubscribed();
+    const frame = makeFrame(cfg);
+    const frames = [...currentFrames(), frame];
+    const handle = new BudgetHandle(frame);
+    try {
+        return await budgetsStore.run(frames, () => cb(handle));
+    }
+    catch (err) {
+        if (err instanceof Truncated)
+            return undefined;
+        throw err;
+    }
+}
+function trackImpl(tags, cb) {
+    ensureSubscribed();
+    const merged = { ...currentTags(), ...tags };
+    return Promise.resolve(tagsStore.run(merged, cb));
+}
+/**
+ * Attribute spend by ambient tags within an async-callback scope (the parity of Python's
+ * `with track(**tags):`). Tags merge with any enclosing `track(...)` and apply to every
+ * instrumented call made inside — including across nested and async calls. `track.report` is the
+ * documented alias for {@link report}.
+ */
+export const track = Object.assign(trackImpl, { report });
+// --------------------------------------------------------------------------- estimate + report
+/**
+ * Pre-flight cost projection without making a call (budget "linting"). Prices the input via
+ * `core.tokens` × `core.prices`, plus `maxOutputTokens` of output (defaults to 0 — input-only).
+ * Throws `UnknownModelError` (the KeyError-equivalent) for an unknown model.
+ */
+export function estimate(model, messages, maxOutputTokens = 0) {
+    const inputTokens = tokens.count(messages, model);
+    return prices.estimate(model, inputTokens, { outputTokens: maxOutputTokens });
+}
+/** Aggregated spend rows. Iterable; cost-as-a-test-assertion via {@link assertUnder}. */
+export class Report {
+    rows;
+    constructor(rows = []) {
+        this.rows = rows;
+    }
+    [Symbol.iterator]() {
+        return this.rows[Symbol.iterator]();
+    }
+    get length() {
+        return this.rows.length;
+    }
+    total() {
+        let sum = new Dec(0);
+        for (const row of this.rows)
+            sum = sum.plus(row.usd.amount);
+        return new Money(sum);
+    }
+    /** Assert spend (optionally filtered by tags) is under `usd`, else throw an AssertionError. */
+    assertUnder(usd, tagFilter = {}) {
+        const cap = new Dec(String(usd));
+        let spent = new Dec(0);
+        const filters = Object.entries(tagFilter);
+        for (const row of this.rows) {
+            if (filters.every(([k, v]) => row.tags[k] === v))
+                spent = spent.plus(row.usd.amount);
+        }
+        if (spent.greaterThan(cap)) {
+            const where = filters.length > 0 ? JSON.stringify(tagFilter) : 'all spend';
+            throw new AssertionError(`$${spent} exceeds cap $${cap} for ${where}`);
+        }
+        return true;
+    }
+}
+/**
+ * Aggregate recorded spend, grouped by the given tag keys. Rows carry `usd` as `Money` and
+ * `tokens` = `input_tokens + output_tokens` (reasoning is a subset of output, NOT double-counted).
+ * Aggregates only the retained window (see {@link dropped}).
+ */
+export function report(groupBy) {
+    const keys = groupBy ?? [];
+    const groups = new Map();
+    const snapshot = [...records]; // snapshot so a concurrent emit can't resize mid-iteration
+    for (const rec of snapshot) {
+        const keyVals = keys.map((k) => (rec.tags[k] === undefined ? null : rec.tags[k]));
+        const gk = JSON.stringify(keyVals);
+        let group = groups.get(gk);
+        if (group === undefined) {
+            const tags = {};
+            for (const k of keys)
+                tags[k] = rec.tags[k] === undefined ? null : rec.tags[k];
+            group = {
+                tags,
+                usd: new Dec(0),
+                inputTokens: 0,
+                outputTokens: 0,
+                reasoningTokens: 0,
+                calls: 0,
+                unpricedCalls: 0,
+            };
+            groups.set(gk, group);
+        }
+        group.usd = group.usd.plus(rec.usd);
+        group.inputTokens += rec.inputTokens;
+        group.outputTokens += rec.outputTokens;
+        group.reasoningTokens += rec.reasoningTokens;
+        group.calls += rec.calls;
+        if (rec.unpriced)
+            group.unpricedCalls += rec.calls;
+    }
+    const rows = [];
+    for (const g of groups.values()) {
+        rows.push({
+            tags: g.tags,
+            usd: new Money(g.usd),
+            tokens: g.inputTokens + g.outputTokens,
+            input_tokens: g.inputTokens,
+            output_tokens: g.outputTokens,
+            reasoning_tokens: g.reasoningTokens,
+            calls: g.calls,
+            unpriced_calls: g.unpricedCalls,
+        });
+    }
+    return new Report(rows);
+}
+// --------------------------------------------------------------------------- introspection + config
+/** The pre-flight model downgrades performed so far (`{ from, to, tags }` rows). Returns a copy. */
+export function downgrades() {
+    return [...downgradeRows];
+}
+/** The pre-flight token clamps applied so far (`{ model, kwarg, limit, tags }` rows). Returns a copy. */
+export function clamps() {
+    return [...clampRows];
+}
+/** Attach a spend sink (e.g. `SQLiteSink`/`OTelSink`); returns the previous one. Pass `null` to
+ * detach. The in-memory aggregation (`report()`) always runs regardless. */
+export function useSink(next) {
+    const previous = sink;
+    sink = next;
+    return previous;
+}
+/** Tune tokenguard's runtime behavior. Each argument is independent — omit one to leave it as is. */
+export function configure(opts = {}) {
+    if (opts.maxRecords !== undefined)
+        maxRecords = opts.maxRecords;
+    if (opts.onUnpriced !== undefined) {
+        if (opts.onUnpriced !== 'warn' && opts.onUnpriced !== 'raise') {
+            throw new ValueError(`on_unpriced must be 'warn' or 'raise', got '${opts.onUnpriced}'`);
+        }
+        onUnpriced = opts.onUnpriced;
+    }
+}
+/** Spend rows evicted by the {@link configure} cap since the last {@link reset} (0 if none). */
+export function dropped() {
+    return droppedCount;
+}
+/** Count of recorded calls whose cost was `null` (unpriced/unknown model) in the retained buffer. */
+export function unpricedCalls() {
+    let total = 0;
+    for (const rec of records)
+        if (rec.unpriced)
+            total += rec.calls;
+    return total;
+}
+/** Clear recorded spend and config (tags/budgets are `AsyncLocalStorage`-scoped), and re-arm the
+ * bus subscription. Useful between tests so spend doesn't leak across cases. */
+export function reset() {
+    records.length = 0;
+    droppedCount = 0;
+    downgradeRows.length = 0;
+    clampRows.length = 0;
+    warnedUnpriced.clear();
+    sink = null;
+    maxRecords = DEFAULT_MAX_RECORDS;
+    onUnpriced = DEFAULT_ON_UNPRICED;
+    ensureSubscribed();
+}
+// Subscribe at import so even a bare instrumented call (no budget/track) is aggregated, and arm the
+// pre-flight interceptor. Idempotent, and re-armed by reset().
+ensureSubscribed();
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;GAmBG;AACH,OAAO,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AACrD,OAAO,EACL,GAAG,EACH,OAAO,EACP,IAAI,EACJ,KAAK,EACL,OAAO,EACP,iBAAiB,EACjB,cAAc,EACd,GAAG,EACH,MAAM,EACN,MAAM,GACP,MAAM,cAAc,CAAC;AAGtB,wFAAwF;AAExF;iEACiE;AACjE,MAAM,sBAAsB,GAAG,GAAG,CAAC;AAEnC,oFAAoF;AACpF,MAAM,SAAS,GAAG,CAAC,OAAO,EAAE,OAAO,EAAE,UAAU,EAAE,WAAW,EAAE,OAAO,CAAU,CAAC;AAEhF;;uBAEuB;AACvB,MAAM,WAAW,GAA2B;IAC1C,MAAM,EAAE,uBAAuB;IAC/B,SAAS,EAAE,YAAY;CACxB,CAAC;AAEF,MAAM,mBAAmB,GAAG,OAAO,CAAC;AACpC,MAAM,mBAAmB,GAAG,MAAM,CAAC;AAEnC,gGAAgG;AAEhG;kCACkC;AAClC,MAAM,OAAO,cAAe,SAAQ,KAAK;IACvC,YAAY,OAAe;QACzB,KAAK,CAAC,OAAO,CAAC,CAAC;QACf,IAAI,CAAC,IAAI,GAAG,gBAAgB,CAAC;IAC/B,CAAC;CACF;AAED;+FAC+F;AAC/F,MAAM,OAAO,oBAAqB,SAAQ,KAAK;IAC7C,YAAY,OAAe;QACzB,KAAK,CAAC,OAAO,CAAC,CAAC;QACf,IAAI,CAAC,IAAI,GAAG,sBAAsB,CAAC;IACrC,CAAC;CACF;AAED,+FAA+F;AAC/F,MAAM,SAAU,SAAQ,KAAK;IAC3B;QACE,KAAK,CAAC,WAAW,CAAC,CAAC;QACnB,IAAI,CAAC,IAAI,GAAG,WAAW,CAAC;IAC1B,CAAC;CACF;AAED,sGAAsG;AACtG,MAAM,UAAW,SAAQ,KAAK;IAC5B,YAAY,OAAe;QACzB,KAAK,CAAC,OAAO,CAAC,CAAC;QACf,IAAI,CAAC,IAAI,GAAG,YAAY,CAAC;IAC3B,CAAC;CACF;AAED,mFAAmF;AACnF,MAAM,cAAe,SAAQ,KAAK;IAChC,YAAY,OAAe;QACzB,KAAK,CAAC,OAAO,CAAC,CAAC;QACf,IAAI,CAAC,IAAI,GAAG,gBAAgB,CAAC;IAC/B,CAAC;CACF;AAGD,MAAM,gBAAgB,GAAG,IAAI,GAAG,EAA2B,CAAC;AAE5D;;;;;GAKG;AACH,MAAM,UAAU,iBAAiB,CAAC,QAAiC;IACjE,gBAAgB,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IAC/B,OAAO,GAAG,EAAE;QACV,gBAAgB,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;IACpC,CAAC,CAAC;AACJ,CAAC;AAkBD,0FAA0F;AAC1F,MAAM,OAAO,KAAK;IAChB,MAAM,CAAiB;IACvB,SAAS,CAAgB;IACzB,QAAQ,CAAW;IACnB,KAAK,CAAgB;IACrB,SAAS,CAAgC;IACzC,aAAa,CAAS;IACtB,gBAAgB,CAAS;IACzB,QAAQ,GAAY,IAAI,GAAG,CAAC,CAAC,CAAC,CAAC;IAC/B,WAAW,GAAG,CAAC,CAAC;IAChB,KAAK,GAAG,CAAC,CAAC;IAEV,YAAY,IAQX;QACC,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;QAC1B,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC;QAChC,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC;QAC9B,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC;QACxB,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC;QAChC,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,aAAa,CAAC;QACxC,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC,gBAAgB,CAAC;IAChD,CAAC;CACF;AAyCD,2FAA2F;AAE3F,MAAM,SAAS,GAAG,IAAI,iBAAiB,EAA2B,CAAC;AACnE,MAAM,YAAY,GAAG,IAAI,iBAAiB,EAAW,CAAC;AACtD,MAAM,OAAO,GAAkB,EAAE,CAAC;AAClC,MAAM,aAAa,GAAmB,EAAE,CAAC;AACzC,MAAM,SAAS,GAAe,EAAE,CAAC;AACjC,IAAI,IAAI,GAAgB,IAAI,CAAC;AAC7B,IAAI,UAAU,GAAkB,mBAAmB,CAAC;AACpD,IAAI,YAAY,GAAG,CAAC,CAAC;AACrB,IAAI,UAAU,GAAG,mBAAmB,CAAC;AACrC,MAAM,cAAc,GAAG,IAAI,GAAG,EAAU,CAAC;AAEzC,SAAS,WAAW;IAClB,OAAO,SAAS,CAAC,QAAQ,EAAE,IAAI,EAAE,CAAC;AACpC,CAAC;AAED,SAAS,aAAa;IACpB,OAAO,YAAY,CAAC,QAAQ,EAAE,IAAI,EAAE,CAAC;AACvC,CAAC;AAED,SAAS,YAAY,CAAC,KAAa,EAAE,IAAY;IAC/C,IAAI,cAAc,CAAC,GAAG,CAAC,KAAK,CAAC;QAAE,OAAO;IACtC,cAAc,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IAC1B,MAAM,OAAO,GAAG,IAAI,oBAAoB,CACtC,mCAAmC,KAAK,2CAA2C,IAAI,+MAA+M,CACvS,CAAC;IACF,IAAI,gBAAgB,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;QAChC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QAC9B,OAAO;IACT,CAAC;IACD,KAAK,MAAM,QAAQ,IAAI,gBAAgB;QAAE,QAAQ,CAAC,OAAO,CAAC,CAAC;AAC7D,CAAC;AAED,SAAS,gBAAgB;IACvB,GAAG,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,6BAA6B;IACpD,cAAc,CAAC,qBAAqB,CAAC,CAAC,CAAC,uDAAuD;AAChG,CAAC;AAED,iGAAiG;AAEjG;;;;;;GAMG;AACH,MAAM,UAAU,gBAAgB,CAAC,IAAa,EAAE,OAAe,EAAE,gBAAgB,GAAG,CAAC;IACnF,MAAM,MAAM,GAAI,IAAI,CAAC,QAAQ,CAAC,cAAsD,IAAI,EAAE,CAAC;IAC3F,IAAI,QAAQ,GAAG,MAAM,CAAC,qBAAqB,CAAC;IAC5C,IAAI,QAAQ,IAAI,IAAI;QAAE,QAAQ,GAAG,MAAM,CAAC,UAAU,CAAC;IACnD,IAAI,QAAQ,IAAI,IAAI;QAAE,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC;IAC1D,OAAO,OAAO,GAAG,gBAAgB,CAAC;AACpC,CAAC;AAED;0EAC0E;AAC1E,SAAS,aAAa,CAAC,IAAa,EAAE,OAAe,EAAE,gBAAgB,GAAG,CAAC;IACzE,MAAM,WAAW,GAAG,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC;IAC5D,MAAM,SAAS,GAAG,gBAAgB,CAAC,IAAI,EAAE,OAAO,EAAE,gBAAgB,CAAC,CAAC;IACpE,OAAO,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,EAAE,WAAW,EAAE,EAAE,YAAY,EAAE,SAAS,EAAE,CAAC,CAAC,MAAM,CAAC;AACtF,CAAC;AAED,8FAA8F;AAC9F,SAAS,aAAa,CAAC,IAAa,EAAE,OAAe,EAAE,gBAAgB,GAAG,CAAC;IACzE,OAAO,CACL,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,gBAAgB,CAAC,IAAI,EAAE,OAAO,EAAE,gBAAgB,CAAC,CAC5F,CAAC;AACJ,CAAC;AAED,qGAAqG;AAErG;;;;;;;;GAQG;AACH,MAAM,UAAU,qBAAqB,CAAC,IAAa;IACjD,IAAI,CAAC,CAAC,IAAI,YAAY,OAAO,CAAC;QAAE,OAAO,IAAI,CAAC;IAC5C,MAAM,MAAM,GAAG,aAAa,EAAE,CAAC;IAC/B,KAAK,IAAI,CAAC,GAAG,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QAC5C,MAAM,KAAK,GAAG,MAAM,CAAC,CAAC,CAAE,CAAC;QACzB,IAAI,KAAK,CAAC,QAAQ,KAAK,WAAW,IAAI,KAAK,CAAC,SAAS,IAAI,KAAK,CAAC,MAAM,KAAK,IAAI,EAAE,CAAC;YAC/E,MAAM,OAAO,GAAG,KAAK,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC5C,IAAI,CAAC,OAAO;gBAAE,SAAS;YACvB,IAAI,SAAkB,CAAC;YACvB,IAAI,CAAC;gBACH,SAAS,GAAG,aAAa,CAAC,IAAI,EAAE,KAAK,CAAC,aAAa,EAAE,KAAK,CAAC,gBAAgB,CAAC,CAAC;YAC/E,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,IAAI,GAAG,YAAY,iBAAiB,EAAE,CAAC;oBACrC,YAAY,CAAC,IAAI,CAAC,KAAK,EAAE,WAAW,CAAC,CAAC,CAAC,mCAAmC;oBAC1E,SAAS,CAAC,6CAA6C;gBACzD,CAAC;gBACD,MAAM,GAAG,CAAC;YACZ,CAAC;YACD,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,WAAW,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC;gBAC7D,aAAa,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,KAAK,EAAE,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE,GAAG,WAAW,EAAE,EAAE,EAAE,CAAC,CAAC;gBAClF,OAAO,IAAI,OAAO,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC,CAAC;YACzC,CAAC;QACH,CAAC;aAAM,IAAI,KAAK,CAAC,QAAQ,KAAK,OAAO,EAAE,CAAC;YACtC,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;YACnC,IAAI,OAAO,KAAK,IAAI;gBAAE,OAAO,OAAO,CAAC;QACvC,CAAC;aAAM,IAAI,KAAK,CAAC,QAAQ,KAAK,OAAO,EAAE,CAAC;YACtC,MAAM,UAAU,GAAG,aAAa,CAAC,IAAI,EAAE,KAAK,CAAC,aAAa,EAAE,KAAK,CAAC,gBAAgB,CAAC,CAAC;YACpF,IAAI,KAAK,CAAC,SAAS,KAAK,IAAI,IAAI,KAAK,CAAC,WAAW,GAAG,UAAU,GAAG,KAAK,CAAC,SAAS,EAAE,CAAC;gBACjF,MAAM,IAAI,cAAc,CACtB,sBAAsB,KAAK,CAAC,WAAW,GAAG,UAAU,4BAA4B,KAAK,CAAC,SAAS,WAAW,IAAI,CAAC,KAAK,GAAG,CACxH,CAAC;YACJ,CAAC;YACD,IAAI,KAAK,CAAC,MAAM,KAAK,IAAI,EAAE,CAAC;gBAC1B,IAAI,SAAkB,CAAC;gBACvB,IAAI,CAAC;oBACH,SAAS,GAAG,aAAa,CAAC,IAAI,EAAE,KAAK,CAAC,aAAa,EAAE,KAAK,CAAC,gBAAgB,CAAC,CAAC;gBAC/E,CAAC;gBAAC,OAAO,GAAG,EAAE,CAAC;oBACb,IAAI,GAAG,YAAY,iBAAiB,EAAE,CAAC;wBACrC,IAAI,UAAU,KAAK,OAAO,EAAE,CAAC;4BAC3B,MAAM,IAAI,cAAc,CACtB,2BAA2B,IAAI,CAAC,KAAK,4JAA4J,CAClM,CAAC;wBACJ,CAAC;wBACD,YAAY,CAAC,IAAI,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;wBAClC,SAAS;oBACX,CAAC;oBACD,MAAM,GAAG,CAAC;gBACZ,CAAC;gBACD,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,WAAW,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC;oBAC7D,MAAM,IAAI,cAAc,CACtB,gCAAgC,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAC,sBAAsB,KAAK,CAAC,MAAM,WAAW,IAAI,CAAC,KAAK,GAAG,CACzH,CAAC;gBACJ,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED;;;;GAIG;AACH,SAAS,KAAK,CAAC,IAAa,EAAE,KAAY;IACxC,IAAI,KAAK,CAAC,SAAS,KAAK,IAAI;QAAE,OAAO,IAAI,CAAC;IAC1C,MAAM,cAAc,GAAG,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC;IAC/D,MAAM,UAAU,GAAG,gBAAgB,CAAC,IAAI,EAAE,KAAK,CAAC,aAAa,EAAE,KAAK,CAAC,gBAAgB,CAAC,CAAC;IACvF,IAAI,KAAK,CAAC,WAAW,GAAG,cAAc,GAAG,UAAU,IAAI,KAAK,CAAC,SAAS;QAAE,OAAO,IAAI,CAAC;IACpF,MAAM,SAAS,GAAG,KAAK,CAAC,SAAS,GAAG,KAAK,CAAC,WAAW,GAAG,cAAc,CAAC;IACvE,MAAM,KAAK,GAAG,WAAW,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IACzC,IAAI,KAAK,KAAK,SAAS,IAAI,SAAS,IAAI,CAAC,EAAE,CAAC;QAC1C,MAAM,IAAI,cAAc,CACtB,yEAAyE,KAAK,CAAC,SAAS,GAAG,KAAK,CAAC,WAAW,WAAW,cAAc,qBAAqB,IAAI,CAAC,QAAQ,YAAY,IAAI,CAAC,KAAK,uDAAuD,CACrP,CAAC;IACJ,CAAC;IACD,MAAM,WAAW,GAAI,IAAI,CAAC,QAAQ,CAAC,cAAsD,EAAE,CACzF,KAAK,CACN,CAAC;IACF,MAAM,QAAQ,GAAG,WAAW,IAAI,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC;IAC9E,IAAI,QAAQ,KAAK,IAAI,IAAI,QAAQ,IAAI,SAAS;QAAE,OAAO,IAAI,CAAC,CAAC,gCAAgC;IAC7F,MAAM,MAAM,GAAG,QAAQ,KAAK,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;IAC7E,SAAS,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,GAAG,WAAW,EAAE,EAAE,EAAE,CAAC,CAAC;IACxF,OAAO,IAAI,OAAO,CAAC,EAAE,CAAC,KAAK,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC;AAC1C,CAAC;AAED,qGAAqG;AAErG,4FAA4F;AAC5F,SAAS,MAAM,CAAC,IAAa;IAC3B,IAAI,CAAC,CAAC,IAAI,YAAY,OAAO,CAAC;QAAE,OAAO,CAAC,2CAA2C;IACnF,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,KAAK,IAAI,CAAC,CAAC,sDAAsD;IAC3F,MAAM,GAAG,GAAG,IAAI,CAAC,IAAI,KAAK,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC,CAAC;IAC/D,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,KAAK,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC;IAC7D,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,KAAK,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC;IAC9D,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,KAAK,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,CAAC;IAEjE,MAAM,MAAM,GAAG,aAAa,EAAE,CAAC;IAC/B,IAAI,QAAQ,EAAE,CAAC;QACb,6FAA6F;QAC7F,8FAA8F;QAC9F,+CAA+C;QAC/C,IAAI,QAAQ,GAAiB,IAAI,CAAC;QAClC,KAAK,IAAI,CAAC,GAAG,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;YAC5C,IAAI,MAAM,CAAC,CAAC,CAAE,CAAC,MAAM,KAAK,IAAI,EAAE,CAAC;gBAC/B,QAAQ,GAAG,MAAM,CAAC,CAAC,CAAE,CAAC;gBACtB,MAAM;YACR,CAAC;QACH,CAAC;QACD,IAAI,QAAQ,KAAK,IAAI,EAAE,CAAC;YACtB,MAAM,IAAI,GAAG,QAAQ,CAAC,QAAQ,CAAC;YAC/B,YAAY,CAAC,IAAI,CAAC,KAAK,EAAE,OAAO,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC;QACzE,CAAC;IACH,CAAC;IAED,MAAM,IAAI,GAAG,EAAE,GAAG,WAAW,EAAE,EAAE,CAAC;IAClC,OAAO,CAAC,IAAI,CAAC;QACX,IAAI;QACJ,GAAG;QACH,WAAW,EAAE,GAAG;QAChB,YAAY,EAAE,GAAG;QACjB,eAAe,EAAE,GAAG;QACpB,KAAK,EAAE,IAAI,CAAC,KAAK;QACjB,KAAK,EAAE,CAAC;QACR,QAAQ;KACT,CAAC,CAAC;IACH,iGAAiG;IACjG,0DAA0D;IAC1D,IAAI,UAAU,KAAK,IAAI,IAAI,OAAO,CAAC,MAAM,GAAG,UAAU,EAAE,CAAC;QACvD,MAAM,QAAQ,GAAG,OAAO,CAAC,MAAM,GAAG,UAAU,CAAC;QAC7C,OAAO,CAAC,MAAM,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAC,CAAC,+CAA+C;QAC5E,YAAY,IAAI,QAAQ,CAAC;IAC3B,CAAC;IACD,IAAI,IAAI,KAAK,IAAI,EAAE,CAAC;QAClB,IAAI,CAAC,KAAK,CAAC;YACT,IAAI;YACJ,GAAG,EAAE,GAAG,CAAC,QAAQ,EAAE;YACnB,YAAY,EAAE,GAAG;YACjB,aAAa,EAAE,GAAG;YAClB,gBAAgB,EAAE,GAAG;YACrB,KAAK,EAAE,IAAI,CAAC,KAAK;SAClB,CAAC,CAAC;IACL,CAAC;IAED,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;QAC3B,KAAK,CAAC,QAAQ,GAAG,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC1C,KAAK,CAAC,WAAW,IAAI,GAAG,GAAG,GAAG,CAAC;QAC/B,KAAK,CAAC,KAAK,IAAI,CAAC,CAAC;IACnB,CAAC;IAED,KAAK,IAAI,CAAC,GAAG,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QAC5C,MAAM,KAAK,GAAG,MAAM,CAAC,CAAC,CAAE,CAAC,CAAC,sDAAsD;QAChF,IAAI,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;YAChB,IAAI,KAAK,CAAC,QAAQ,KAAK,WAAW,IAAI,KAAK,CAAC,QAAQ,KAAK,OAAO,EAAE,CAAC;gBACjE,SAAS,CAAC,uEAAuE;YACnF,CAAC;YACD,OAAO,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;YACrB,MAAM;QACR,CAAC;IACH,CAAC;AACH,CAAC;AAED,SAAS,IAAI,CAAC,KAAY;IACxB,IAAI,KAAK,CAAC,MAAM,KAAK,IAAI,IAAI,KAAK,CAAC,QAAQ,CAAC,WAAW,CAAC,KAAK,CAAC,MAAM,CAAC;QAAE,OAAO,IAAI,CAAC;IACnF,IAAI,KAAK,CAAC,SAAS,KAAK,IAAI,IAAI,KAAK,CAAC,WAAW,GAAG,KAAK,CAAC,SAAS;QAAE,OAAO,IAAI,CAAC;IACjF,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,OAAO,CAAC,KAAY,EAAE,IAAa;IAC1C,MAAM,IAAI,GAAG,KAAK,CAAC,QAAQ,CAAC;IAC5B,IAAI,OAAO,IAAI,KAAK,UAAU,EAAE,CAAC;QAC/B,IAAI,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC;QACtE,OAAO;IACT,CAAC;IACD,IAAI,IAAI,KAAK,UAAU;QAAE,MAAM,IAAI,SAAS,EAAE,CAAC;IAC/C,IAAI,IAAI,KAAK,WAAW,IAAI,IAAI,KAAK,OAAO;QAAE,OAAO,CAAC,6BAA6B;IACnF,MAAM,IAAI,cAAc,CACtB,2BAA2B,KAAK,CAAC,QAAQ,WAAW,KAAK,CAAC,MAAM,UAAU,KAAK,CAAC,KAAK,wBAAwB,IAAI,CAAC,KAAK,+JAA+J,CACvR,CAAC;AACJ,CAAC;AAeD,SAAS,oBAAoB,CAAC,GAAiB;IAC7C,MAAM,aAAa,GAAG,GAAG,CAAC,QAAQ,IAAI,OAAO,CAAC;IAC9C,IACE,OAAO,aAAa,KAAK,UAAU;QACnC,CAAE,SAA+B,CAAC,QAAQ,CAAC,aAAa,CAAC,EACzD,CAAC;QACD,MAAM,IAAI,UAAU,CAClB,0CAA0C,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC,UAAU,MAAM,CAAC,aAAa,CAAC,GAAG,CAChG,CAAC;IACJ,CAAC;IACD,IAAI,GAAG,CAAC,GAAG,IAAI,IAAI,IAAI,GAAG,CAAC,MAAM,IAAI,IAAI,EAAE,CAAC;QAC1C,MAAM,IAAI,UAAU,CAAC,iDAAiD,CAAC,CAAC;IAC1E,CAAC;IACD,IAAI,aAAa,KAAK,WAAW,EAAE,CAAC;QAClC,IAAI,CAAC,GAAG,CAAC,SAAS,IAAI,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC9D,MAAM,IAAI,UAAU,CAAC,iEAAiE,CAAC,CAAC;QAC1F,CAAC;QACD,IAAI,GAAG,CAAC,GAAG,IAAI,IAAI,EAAE,CAAC;YACpB,MAAM,IAAI,UAAU,CAClB,yEAAyE,CAC1E,CAAC;QACJ,CAAC;IACH,CAAC;IACD,IAAI,aAAa,KAAK,OAAO,IAAI,GAAG,CAAC,MAAM,IAAI,IAAI,EAAE,CAAC;QACpD,MAAM,IAAI,UAAU,CAClB,gFAAgF,CACjF,CAAC;IACJ,CAAC;AACH,CAAC;AAED,SAAS,SAAS,CAAC,GAAiB;IAClC,OAAO,IAAI,KAAK,CAAC;QACf,MAAM,EAAE,GAAG,CAAC,GAAG,IAAI,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACzD,SAAS,EAAE,GAAG,CAAC,MAAM,IAAI,IAAI;QAC7B,QAAQ,EAAE,GAAG,CAAC,QAAQ,IAAI,OAAO;QACjC,KAAK,EAAE,GAAG,CAAC,KAAK,IAAI,IAAI;QACxB,SAAS,EAAE,GAAG,CAAC,SAAS,IAAI,IAAI;QAChC,aAAa,EAAE,GAAG,CAAC,aAAa,IAAI,sBAAsB;QAC1D,gBAAgB,EAAE,GAAG,CAAC,gBAAgB,IAAI,CAAC;KAC5C,CAAC,CAAC;AACL,CAAC;AAED,mFAAmF;AACnF,MAAM,OAAO,YAAY;IACM;IAA7B,YAA6B,KAAY;QAAZ,UAAK,GAAL,KAAK,CAAO;IAAG,CAAC;IAE7C,8FAA8F;IAC9F,IAAI,KAAK;QACP,OAAO,IAAI,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;IACxC,CAAC;CACF;AAED;;;;;;;GAOG;AACH,MAAM,UAAU,MAAM,CAAC,GAAiB;IACtC,oBAAoB,CAAC,GAAG,CAAC,CAAC;IAC1B,OAAO,CAAyB,EAAkC,EAAE,EAAE,CACpE,KAAK,EAAE,GAAG,IAAO,EAA0B,EAAE;QAC3C,gBAAgB,EAAE,CAAC;QACnB,MAAM,KAAK,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC;QAC7B,MAAM,MAAM,GAAG,CAAC,GAAG,aAAa,EAAE,EAAE,KAAK,CAAC,CAAC;QAC3C,IAAI,CAAC;YACH,OAAO,MAAM,YAAY,CAAC,GAAG,CAAC,MAAM,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC;QAC3D,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,GAAG,YAAY,SAAS;gBAAE,OAAO,SAAS,CAAC,CAAC,0CAA0C;YAC1F,MAAM,GAAG,CAAC;QACZ,CAAC;IACH,CAAC,CAAC;AACN,CAAC;AAED;;;;;;GAMG;AACH,MAAM,CAAC,KAAK,UAAU,UAAU,CAC9B,GAAiB,EACjB,EAAuC;IAEvC,oBAAoB,CAAC,GAAG,CAAC,CAAC;IAC1B,gBAAgB,EAAE,CAAC;IACnB,MAAM,KAAK,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC;IAC7B,MAAM,MAAM,GAAG,CAAC,GAAG,aAAa,EAAE,EAAE,KAAK,CAAC,CAAC;IAC3C,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,KAAK,CAAC,CAAC;IACvC,IAAI,CAAC;QACH,OAAO,MAAM,YAAY,CAAC,GAAG,CAAC,MAAM,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC;IAC1D,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,IAAI,GAAG,YAAY,SAAS;YAAE,OAAO,SAAS,CAAC;QAC/C,MAAM,GAAG,CAAC;IACZ,CAAC;AACH,CAAC;AAUD,SAAS,SAAS,CAAI,IAA6B,EAAE,EAAwB;IAC3E,gBAAgB,EAAE,CAAC;IACnB,MAAM,MAAM,GAAG,EAAE,GAAG,WAAW,EAAE,EAAE,GAAG,IAAI,EAAE,CAAC;IAC7C,OAAO,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,GAAG,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,CAAC;AACpD,CAAC;AAED;;;;;GAKG;AACH,MAAM,CAAC,MAAM,KAAK,GAAkB,MAAM,CAAC,MAAM,CAAC,SAAS,EAAE,EAAE,MAAM,EAAE,CAAkB,CAAC;AAE1F,gGAAgG;AAEhG;;;;GAIG;AACH,MAAM,UAAU,QAAQ,CAAC,KAAa,EAAE,QAAmB,EAAE,eAAe,GAAG,CAAC;IAC9E,MAAM,WAAW,GAAG,MAAM,CAAC,KAAK,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;IAClD,OAAO,MAAM,CAAC,QAAQ,CAAC,KAAK,EAAE,WAAW,EAAE,EAAE,YAAY,EAAE,eAAe,EAAE,CAAC,CAAC;AAChF,CAAC;AAED,yFAAyF;AACzF,MAAM,OAAO,MAAM;IACE;IAAnB,YAAmB,OAAoB,EAAE;QAAtB,SAAI,GAAJ,IAAI,CAAkB;IAAG,CAAC;IAE7C,CAAC,MAAM,CAAC,QAAQ,CAAC;QACf,OAAO,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,EAAE,CAAC;IACtC,CAAC;IAED,IAAI,MAAM;QACR,OAAO,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC;IAC1B,CAAC;IAED,KAAK;QACH,IAAI,GAAG,GAAG,IAAI,GAAG,CAAC,CAAC,CAAC,CAAC;QACrB,KAAK,MAAM,GAAG,IAAI,IAAI,CAAC,IAAI;YAAE,GAAG,GAAG,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QAC5D,OAAO,IAAI,KAAK,CAAC,GAAG,CAAC,CAAC;IACxB,CAAC;IAED,+FAA+F;IAC/F,WAAW,CAAC,GAAoB,EAAE,YAAqC,EAAE;QACvE,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;QACjC,IAAI,KAAK,GAAG,IAAI,GAAG,CAAC,CAAC,CAAC,CAAC;QACvB,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QAC1C,KAAK,MAAM,GAAG,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;YAC5B,IAAI,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;gBAAE,KAAK,GAAG,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACvF,CAAC;QACD,IAAI,KAAK,CAAC,WAAW,CAAC,GAAG,CAAC,EAAE,CAAC;YAC3B,MAAM,KAAK,GAAG,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC;YAC3E,MAAM,IAAI,cAAc,CAAC,IAAI,KAAK,iBAAiB,GAAG,QAAQ,KAAK,EAAE,CAAC,CAAC;QACzE,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;CACF;AAYD;;;;GAIG;AACH,MAAM,UAAU,MAAM,CAAC,OAAkB;IACvC,MAAM,IAAI,GAAG,OAAO,IAAI,EAAE,CAAC;IAC3B,MAAM,MAAM,GAAG,IAAI,GAAG,EAAiB,CAAC;IACxC,MAAM,QAAQ,GAAG,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,2DAA2D;IAC1F,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;QAC3B,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAClF,MAAM,EAAE,GAAG,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QACnC,IAAI,KAAK,GAAG,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAC3B,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;YACxB,MAAM,IAAI,GAA4B,EAAE,CAAC;YACzC,KAAK,MAAM,CAAC,IAAI,IAAI;gBAAE,IAAI,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAC/E,KAAK,GAAG;gBACN,IAAI;gBACJ,GAAG,EAAE,IAAI,GAAG,CAAC,CAAC,CAAC;gBACf,WAAW,EAAE,CAAC;gBACd,YAAY,EAAE,CAAC;gBACf,eAAe,EAAE,CAAC;gBAClB,KAAK,EAAE,CAAC;gBACR,aAAa,EAAE,CAAC;aACjB,CAAC;YACF,MAAM,CAAC,GAAG,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;QACxB,CAAC;QACD,KAAK,CAAC,GAAG,GAAG,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACpC,KAAK,CAAC,WAAW,IAAI,GAAG,CAAC,WAAW,CAAC;QACrC,KAAK,CAAC,YAAY,IAAI,GAAG,CAAC,YAAY,CAAC;QACvC,KAAK,CAAC,eAAe,IAAI,GAAG,CAAC,eAAe,CAAC;QAC7C,KAAK,CAAC,KAAK,IAAI,GAAG,CAAC,KAAK,CAAC;QACzB,IAAI,GAAG,CAAC,QAAQ;YAAE,KAAK,CAAC,aAAa,IAAI,GAAG,CAAC,KAAK,CAAC;IACrD,CAAC;IACD,MAAM,IAAI,GAAgB,EAAE,CAAC;IAC7B,KAAK,MAAM,CAAC,IAAI,MAAM,CAAC,MAAM,EAAE,EAAE,CAAC;QAChC,IAAI,CAAC,IAAI,CAAC;YACR,IAAI,EAAE,CAAC,CAAC,IAAI;YACZ,GAAG,EAAE,IAAI,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC;YACrB,MAAM,EAAE,CAAC,CAAC,WAAW,GAAG,CAAC,CAAC,YAAY;YACtC,YAAY,EAAE,CAAC,CAAC,WAAW;YAC3B,aAAa,EAAE,CAAC,CAAC,YAAY;YAC7B,gBAAgB,EAAE,CAAC,CAAC,eAAe;YACnC,KAAK,EAAE,CAAC,CAAC,KAAK;YACd,cAAc,EAAE,CAAC,CAAC,aAAa;SAChC,CAAC,CAAC;IACL,CAAC;IACD,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,CAAC;AAC1B,CAAC;AAED,qGAAqG;AAErG,oGAAoG;AACpG,MAAM,UAAU,UAAU;IACxB,OAAO,CAAC,GAAG,aAAa,CAAC,CAAC;AAC5B,CAAC;AAED,yGAAyG;AACzG,MAAM,UAAU,MAAM;IACpB,OAAO,CAAC,GAAG,SAAS,CAAC,CAAC;AACxB,CAAC;AAED;4EAC4E;AAC5E,MAAM,UAAU,OAAO,CAAC,IAAiB;IACvC,MAAM,QAAQ,GAAG,IAAI,CAAC;IACtB,IAAI,GAAG,IAAI,CAAC;IACZ,OAAO,QAAQ,CAAC;AAClB,CAAC;AAQD,qGAAqG;AACrG,MAAM,UAAU,SAAS,CAAC,OAAyB,EAAE;IACnD,IAAI,IAAI,CAAC,UAAU,KAAK,SAAS;QAAE,UAAU,GAAG,IAAI,CAAC,UAAU,CAAC;IAChE,IAAI,IAAI,CAAC,UAAU,KAAK,SAAS,EAAE,CAAC;QAClC,IAAI,IAAI,CAAC,UAAU,KAAK,MAAM,IAAI,IAAI,CAAC,UAAU,KAAK,OAAO,EAAE,CAAC;YAC9D,MAAM,IAAI,UAAU,CAAC,+CAA+C,IAAI,CAAC,UAAU,GAAG,CAAC,CAAC;QAC1F,CAAC;QACD,UAAU,GAAG,IAAI,CAAC,UAAU,CAAC;IAC/B,CAAC;AACH,CAAC;AAED,gGAAgG;AAChG,MAAM,UAAU,OAAO;IACrB,OAAO,YAAY,CAAC;AACtB,CAAC;AAED,qGAAqG;AACrG,MAAM,UAAU,aAAa;IAC3B,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,KAAK,MAAM,GAAG,IAAI,OAAO;QAAE,IAAI,GAAG,CAAC,QAAQ;YAAE,KAAK,IAAI,GAAG,CAAC,KAAK,CAAC;IAChE,OAAO,KAAK,CAAC;AACf,CAAC;AAED;gFACgF;AAChF,MAAM,UAAU,KAAK;IACnB,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC;IACnB,YAAY,GAAG,CAAC,CAAC;IACjB,aAAa,CAAC,MAAM,GAAG,CAAC,CAAC;IACzB,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC;IACrB,cAAc,CAAC,KAAK,EAAE,CAAC;IACvB,IAAI,GAAG,IAAI,CAAC;IACZ,UAAU,GAAG,mBAAmB,CAAC;IACjC,UAAU,GAAG,mBAAmB,CAAC;IACjC,gBAAgB,EAAE,CAAC;AACrB,CAAC;AAED,oGAAoG;AACpG,+DAA+D;AAC/D,gBAAgB,EAAE,CAAC"}