pi-harness-runtime 0.9.2 → 0.9.5

package/packages/quota-manager/dist/quota-manager.d.ts

@@ -0,0 +1,70 @@
+/**
+ * Quota Manager — RFC-0003
+ *
+ * Collects quota signals from API responses, provider status,
+ * Playwright, and local estimates. Produces provider availability state.
+ */
+import type { QuotaSignal, QuotaState } from "../../packages/types/src/runtime-types.js";
+export interface QuotaSignalInput {
+    provider: string;
+    source: "api_response" | "provider_status" | "playwright" | "local_estimate" | "tui_message";
+    windowType: "5h" | "daily" | "weekly" | "monthly";
+    usedPct?: number;
+    exhausted?: boolean;
+    resetsAt?: string;
+    retryAfterMs?: number;
+}
+export declare class QuotaManager {
+    private signals;
+    private readonly staleThresholdMs;
+    constructor(staleThresholdMs?: number);
+    /**
+     * Record a quota signal
+     */
+    recordSignal(input: QuotaSignalInput): void;
+    /**
+     * Get the latest signal for a provider
+     */
+    getLatestSignal(provider: string, windowType?: "5h" | "daily" | "weekly" | "monthly"): QuotaSignal | null;
+    /**
+     * Get quota state for a provider
+     */
+    getProviderState(provider: string): QuotaState;
+    /**
+     * Get signals for a provider
+     */
+    getSignals(provider: string): QuotaSignal[];
+    /**
+     * Check if provider is available
+     */
+    isAvailable(provider: string): boolean;
+    /**
+     * Check if provider is exhausted
+     */
+    isExhausted(provider: string): boolean;
+    /**
+     * Get time until provider is available again
+     */
+    getWaitTime(provider: string): number | null;
+    /**
+     * Get best available provider from a list
+     */
+    selectBestProvider(providers: string[]): string | null;
+    /**
+     * Clear stale signals
+     */
+    clearStale(): void;
+    /**
+     * Generate quota report
+     */
+    generateReport(providers: string[]): string;
+}
+/**
+ * Parse quota signal from MiniMax error
+ */
+export declare function parseMiniMaxError(error: unknown): QuotaSignal | null;
+/**
+ * Parse quota signal from OpenAI error
+ */
+export declare function parseOpenAIError(error: unknown): QuotaSignal | null;
+//# sourceMappingURL=quota-manager.d.ts.map

package/packages/quota-manager/dist/quota-manager.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"quota-manager.d.ts","sourceRoot":"","sources":["../src/quota-manager.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,KAAK,EACX,WAAW,EACX,UAAU,EACV,MAAM,2CAA2C,CAAC;AAEnD,MAAM,WAAW,gBAAgB;IAChC,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EACH,cAAc,GACd,iBAAiB,GACjB,YAAY,GACZ,gBAAgB,GAChB,aAAa,CAAC;IACjB,UAAU,EAAE,IAAI,GAAG,OAAO,GAAG,QAAQ,GAAG,SAAS,CAAC;IAClD,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,YAAY,CAAC,EAAE,MAAM,CAAC;CACtB;AAED,qBAAa,YAAY;IACxB,OAAO,CAAC,OAAO,CAAyC;IACxD,OAAO,CAAC,QAAQ,CAAC,gBAAgB,CAAS;IAE1C,YAAY,gBAAgB,GAAE,MAAuB,EAEpD;IAED;;OAEG;IACH,YAAY,CAAC,KAAK,EAAE,gBAAgB,GAAG,IAAI,CAe1C;IAED;;OAEG;IACH,eAAe,CACd,QAAQ,EAAE,MAAM,EAChB,UAAU,CAAC,EAAE,IAAI,GAAG,OAAO,GAAG,QAAQ,GAAG,SAAS,GAChD,WAAW,GAAG,IAAI,CAkCpB;IAED;;OAEG;IACH,gBAAgB,CAAC,QAAQ,EAAE,MAAM,GAAG,UAAU,CA8B7C;IAED;;OAEG;IACH,UAAU,CAAC,QAAQ,EAAE,MAAM,GAAG,WAAW,EAAE,CAgB1C;IAED;;OAEG;IACH,WAAW,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAErC;IAED;;OAEG;IACH,WAAW,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAErC;IAED;;OAEG;IACH,WAAW,CAAC,QAAQ,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI,CAM3C;IAED;;OAEG;IACH,kBAAkB,CAAC,SAAS,EAAE,MAAM,EAAE,GAAG,MAAM,GAAG,IAAI,CAerD;IAED;;OAEG;IACH,UAAU,IAAI,IAAI,CAcjB;IAED;;OAEG;IACH,cAAc,CAAC,SAAS,EAAE,MAAM,EAAE,GAAG,MAAM,CA8B1C;CACD;AAED;;GAEG;AACH,wBAAgB,iBAAiB,CAAC,KAAK,EAAE,OAAO,GAAG,WAAW,GAAG,IAAI,CAmCpE;AAED;;GAEG;AACH,wBAAgB,gBAAgB,CAAC,KAAK,EAAE,OAAO,GAAG,WAAW,GAAG,IAAI,CA0CnE"}

package/packages/quota-manager/dist/quota-manager.js

@@ -0,0 +1,260 @@
+/**
+ * Quota Manager — RFC-0003
+ *
+ * Collects quota signals from API responses, provider status,
+ * Playwright, and local estimates. Produces provider availability state.
+ */
+export class QuotaManager {
+    signals = new Map();
+    staleThresholdMs;
+    constructor(staleThresholdMs = 30 * 60 * 1000) {
+        this.staleThresholdMs = staleThresholdMs;
+    }
+    /**
+     * Record a quota signal
+     */
+    recordSignal(input) {
+        const signal = {
+            provider: input.provider,
+            windowType: input.windowType,
+            usedPct: input.usedPct ?? 0,
+            remainingPct: 100 - (input.usedPct ?? 0),
+            exhausted: input.exhausted ?? false,
+            source: input.source,
+            capturedAt: new Date().toISOString(),
+            resetsAt: input.resetsAt,
+        };
+        const key = `${input.provider}:${input.windowType}`;
+        const existing = this.signals.get(key) ?? [];
+        this.signals.set(key, [...existing, signal]);
+    }
+    /**
+     * Get the latest signal for a provider
+     */
+    getLatestSignal(provider, windowType) {
+        const key = windowType ? `${provider}:${windowType}` : provider;
+        if (windowType) {
+            const signals = this.signals.get(key) ?? [];
+            return (signals.sort((a, b) => Date.parse(b.capturedAt) - Date.parse(a.capturedAt))[0] ?? null);
+        }
+        // Return most recent across all window types
+        let latest = null;
+        let latestTs = 0;
+        for (const entry of Array.from(this.signals.entries())) {
+            const [k, sigs] = entry;
+            if (k.startsWith(`${provider}:`)) {
+                const mostRecent = sigs.sort((a, b) => Date.parse(b.capturedAt) - Date.parse(a.capturedAt))[0];
+                if (mostRecent) {
+                    const ts = Date.parse(mostRecent.capturedAt);
+                    if (ts > latestTs) {
+                        latestTs = ts;
+                        latest = mostRecent;
+                    }
+                }
+            }
+        }
+        return latest;
+    }
+    /**
+     * Get quota state for a provider
+     */
+    getProviderState(provider) {
+        const signals = this.getSignals(provider);
+        const exhausted = signals.some((s) => s.exhausted || (s.usedPct !== undefined && s.usedPct >= 100));
+        const limited = signals.some((s) => s.usedPct !== undefined && s.usedPct >= 90);
+        // Find next reset time
+        let nextAvailableAt;
+        for (const signal of signals) {
+            if (signal.resetsAt) {
+                if (!nextAvailableAt ||
+                    Date.parse(signal.resetsAt) < Date.parse(nextAvailableAt)) {
+                    nextAvailableAt = signal.resetsAt;
+                }
+            }
+        }
+        return {
+            provider,
+            available: !exhausted && !limited,
+            limited,
+            exhausted,
+            signals,
+            nextAvailableAt,
+        };
+    }
+    /**
+     * Get signals for a provider
+     */
+    getSignals(provider) {
+        const result = [];
+        const cutoff = Date.now() - this.staleThresholdMs;
+        for (const entry of Array.from(this.signals.entries())) {
+            const [key, sigs] = entry;
+            if (key.startsWith(`${provider}:`)) {
+                for (const signal of sigs) {
+                    if (Date.parse(signal.capturedAt) >= cutoff) {
+                        result.push(signal);
+                    }
+                }
+            }
+        }
+        return result;
+    }
+    /**
+     * Check if provider is available
+     */
+    isAvailable(provider) {
+        return this.getProviderState(provider).available;
+    }
+    /**
+     * Check if provider is exhausted
+     */
+    isExhausted(provider) {
+        return this.getProviderState(provider).exhausted;
+    }
+    /**
+     * Get time until provider is available again
+     */
+    getWaitTime(provider) {
+        const state = this.getProviderState(provider);
+        if (!state.nextAvailableAt)
+            return null;
+        const waitMs = Date.parse(state.nextAvailableAt) - Date.now();
+        return waitMs > 0 ? waitMs : null;
+    }
+    /**
+     * Get best available provider from a list
+     */
+    selectBestProvider(providers) {
+        const available = providers.filter((p) => this.isAvailable(p));
+        if (available.length === 0)
+            return null;
+        // Sort by remaining quota (prefer those with more remaining)
+        available.sort((a, b) => {
+            const signalA = this.getLatestSignal(a);
+            const signalB = this.getLatestSignal(b);
+            const remA = signalA?.remainingPct ?? 100;
+            const remB = signalB?.remainingPct ?? 100;
+            return remB - remA;
+        });
+        return available[0];
+    }
+    /**
+     * Clear stale signals
+     */
+    clearStale() {
+        const cutoff = Date.now() - this.staleThresholdMs;
+        for (const entry of Array.from(this.signals.entries())) {
+            const [key, sigs] = entry;
+            const fresh = sigs.filter((s) => Date.parse(s.capturedAt) >= cutoff);
+            if (fresh.length === 0) {
+                this.signals.delete(key);
+            }
+            else {
+                this.signals.set(key, fresh);
+            }
+        }
+    }
+    /**
+     * Generate quota report
+     */
+    generateReport(providers) {
+        const lines = ["Quota Status Report", "=".repeat(40), ""];
+        for (const provider of providers) {
+            const state = this.getProviderState(provider);
+            const latest = this.getLatestSignal(provider);
+            const status = state.exhausted
+                ? "EXHAUSTED"
+                : state.limited
+                    ? "LIMITED"
+                    : "AVAILABLE";
+            lines.push(`${provider}: ${status}`);
+            if (latest) {
+                lines.push(`  Latest: ${latest.usedPct ?? 0}% used`);
+                if (latest.resetsAt) {
+                    const wait = Date.parse(latest.resetsAt) - Date.now();
+                    if (wait > 0) {
+                        lines.push(`  Resets: in ${Math.ceil(wait / 60000)} minutes`);
+                    }
+                }
+                lines.push(`  Source: ${latest.source}`);
+            }
+            lines.push("");
+        }
+        return lines.join("\n");
+    }
+}
+/**
+ * Parse quota signal from MiniMax error
+ */
+export function parseMiniMaxError(error) {
+    const e = error;
+    const msg = String(e.message ?? e.error ?? "");
+    // MiniMax quota error: "Error code: 2056 - Rate limit exceeded. Retry after..."
+    if (msg.includes("2056") ||
+        msg.includes("quota") ||
+        msg.includes("rate limit")) {
+        // Try to extract retry time
+        const retryMatch = msg.match(/retry after (\d+) (seconds?|minutes?)/i);
+        let retryAfterMs;
+        if (retryMatch) {
+            const value = parseInt(retryMatch[1], 10);
+            const unit = retryMatch[2].toLowerCase();
+            retryAfterMs = unit.startsWith("minute")
+                ? value * 60 * 1000
+                : value * 1000;
+        }
+        return {
+            provider: "minimax",
+            windowType: "5h",
+            usedPct: 100,
+            remainingPct: 0,
+            exhausted: true,
+            source: "api_response",
+            capturedAt: new Date().toISOString(),
+            retryAfterMs,
+        };
+    }
+    return null;
+}
+/**
+ * Parse quota signal from OpenAI error
+ */
+export function parseOpenAIError(error) {
+    const e = error;
+    const code = String(e.code ?? "");
+    const msg = String(e.message ?? e.error ?? "");
+    if (code === "insufficient_quota" || code === "context_length_exceeded") {
+        return {
+            provider: "openai",
+            windowType: "daily",
+            usedPct: 100,
+            remainingPct: 0,
+            exhausted: true,
+            source: "api_response",
+            capturedAt: new Date().toISOString(),
+        };
+    }
+    if (msg.includes("429")) {
+        const retryMatch = msg.match(/retry after (\d+) (seconds?|minutes?)/i);
+        let retryAfterMs;
+        if (retryMatch) {
+            const value = parseInt(retryMatch[1], 10);
+            const unit = retryMatch[2].toLowerCase();
+            retryAfterMs = unit.startsWith("minute")
+                ? value * 60 * 1000
+                : value * 1000;
+        }
+        return {
+            provider: "openai",
+            windowType: "5h",
+            usedPct: 100,
+            remainingPct: 0,
+            exhausted: true,
+            source: "api_response",
+            capturedAt: new Date().toISOString(),
+            retryAfterMs,
+        };
+    }
+    return null;
+}
+//# sourceMappingURL=quota-manager.js.map

package/packages/quota-manager/dist/quota-manager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"quota-manager.js","sourceRoot":"","sources":["../src/quota-manager.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAsBH,MAAM,OAAO,YAAY;IAChB,OAAO,GAA+B,IAAI,GAAG,EAAE,CAAC;IACvC,gBAAgB,CAAS;IAE1C,YAAY,gBAAgB,GAAW,EAAE,GAAG,EAAE,GAAG,IAAI;QACpD,IAAI,CAAC,gBAAgB,GAAG,gBAAgB,CAAC;IAC1C,CAAC;IAED;;OAEG;IACH,YAAY,CAAC,KAAuB;QACnC,MAAM,MAAM,GAAgB;YAC3B,QAAQ,EAAE,KAAK,CAAC,QAAQ;YACxB,UAAU,EAAE,KAAK,CAAC,UAAU;YAC5B,OAAO,EAAE,KAAK,CAAC,OAAO,IAAI,CAAC;YAC3B,YAAY,EAAE,GAAG,GAAG,CAAC,KAAK,CAAC,OAAO,IAAI,CAAC,CAAC;YACxC,SAAS,EAAE,KAAK,CAAC,SAAS,IAAI,KAAK;YACnC,MAAM,EAAE,KAAK,CAAC,MAAM;YACpB,UAAU,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACpC,QAAQ,EAAE,KAAK,CAAC,QAAQ;SACxB,CAAC;QAEF,MAAM,GAAG,GAAG,GAAG,KAAK,CAAC,QAAQ,IAAI,KAAK,CAAC,UAAU,EAAE,CAAC;QACpD,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC;QAC7C,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,GAAG,QAAQ,EAAE,MAAM,CAAC,CAAC,CAAC;IAC9C,CAAC;IAED;;OAEG;IACH,eAAe,CACd,QAAgB,EAChB,UAAkD;QAElD,MAAM,GAAG,GAAG,UAAU,CAAC,CAAC,CAAC,GAAG,QAAQ,IAAI,UAAU,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC;QAEhE,IAAI,UAAU,EAAE,CAAC;YAChB,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC;YAC5C,OAAO,CACN,OAAO,CAAC,IAAI,CACX,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,UAAU,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,UAAU,CAAC,CAC7D,CAAC,CAAC,CAAC,IAAI,IAAI,CACZ,CAAC;QACH,CAAC;QAED,6CAA6C;QAC7C,IAAI,MAAM,GAAuB,IAAI,CAAC;QACtC,IAAI,QAAQ,GAAG,CAAC,CAAC;QAEjB,KAAK,MAAM,KAAK,IAAI,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC;YACxD,MAAM,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,KAAK,CAAC;YACxB,IAAI,CAAC,CAAC,UAAU,CAAC,GAAG,QAAQ,GAAG,CAAC,EAAE,CAAC;gBAClC,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAC3B,CAAC,CAAc,EAAE,CAAc,EAAE,EAAE,CAClC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,UAAU,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,UAAU,CAAC,CACpD,CAAC,CAAC,CAAC,CAAC;gBACL,IAAI,UAAU,EAAE,CAAC;oBAChB,MAAM,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC;oBAC7C,IAAI,EAAE,GAAG,QAAQ,EAAE,CAAC;wBACnB,QAAQ,GAAG,EAAE,CAAC;wBACd,MAAM,GAAG,UAAU,CAAC;oBACrB,CAAC;gBACF,CAAC;YACF,CAAC;QACF,CAAC;QAED,OAAO,MAAM,CAAC;IACf,CAAC;IAED;;OAEG;IACH,gBAAgB,CAAC,QAAgB;QAChC,MAAM,OAAO,GAAG,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;QAC1C,MAAM,SAAS,GAAG,OAAO,CAAC,IAAI,CAC7B,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,IAAI,CAAC,CAAC,CAAC,OAAO,KAAK,SAAS,IAAI,CAAC,CAAC,OAAO,IAAI,GAAG,CAAC,CACnE,CAAC;QACF,MAAM,OAAO,GAAG,OAAO,CAAC,IAAI,CAC3B,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,KAAK,SAAS,IAAI,CAAC,CAAC,OAAO,IAAI,EAAE,CACjD,CAAC;QAEF,uBAAuB;QACvB,IAAI,eAAmC,CAAC;QACxC,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;YAC9B,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;gBACrB,IACC,CAAC,eAAe;oBAChB,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,QAAQ,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,eAAe,CAAC,EACxD,CAAC;oBACF,eAAe,GAAG,MAAM,CAAC,QAAQ,CAAC;gBACnC,CAAC;YACF,CAAC;QACF,CAAC;QAED,OAAO;YACN,QAAQ;YACR,SAAS,EAAE,CAAC,SAAS,IAAI,CAAC,OAAO;YACjC,OAAO;YACP,SAAS;YACT,OAAO;YACP,eAAe;SACf,CAAC;IACH,CAAC;IAED;;OAEG;IACH,UAAU,CAAC,QAAgB;QAC1B,MAAM,MAAM,GAAkB,EAAE,CAAC;QACjC,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,gBAAgB,CAAC;QAElD,KAAK,MAAM,KAAK,IAAI,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC;YACxD,MAAM,CAAC,GAAG,EAAE,IAAI,CAAC,GAAG,KAAK,CAAC;YAC1B,IAAI,GAAG,CAAC,UAAU,CAAC,GAAG,QAAQ,GAAG,CAAC,EAAE,CAAC;gBACpC,KAAK,MAAM,MAAM,IAAI,IAAI,EAAE,CAAC;oBAC3B,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI,MAAM,EAAE,CAAC;wBAC7C,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;oBACrB,CAAC;gBACF,CAAC;YACF,CAAC;QACF,CAAC;QAED,OAAO,MAAM,CAAC;IACf,CAAC;IAED;;OAEG;IACH,WAAW,CAAC,QAAgB;QAC3B,OAAO,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC,SAAS,CAAC;IAClD,CAAC;IAED;;OAEG;IACH,WAAW,CAAC,QAAgB;QAC3B,OAAO,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC,SAAS,CAAC;IAClD,CAAC;IAED;;OAEG;IACH,WAAW,CAAC,QAAgB;QAC3B,MAAM,KAAK,GAAG,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;QAC9C,IAAI,CAAC,KAAK,CAAC,eAAe;YAAE,OAAO,IAAI,CAAC;QAExC,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,eAAe,CAAC,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC9D,OAAO,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC;IACnC,CAAC;IAED;;OAEG;IACH,kBAAkB,CAAC,SAAmB;QACrC,MAAM,SAAS,GAAG,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC;QAE/D,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,IAAI,CAAC;QAExC,6DAA6D;QAC7D,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;YACvB,MAAM,OAAO,GAAG,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YACxC,MAAM,OAAO,GAAG,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YACxC,MAAM,IAAI,GAAG,OAAO,EAAE,YAAY,IAAI,GAAG,CAAC;YAC1C,MAAM,IAAI,GAAG,OAAO,EAAE,YAAY,IAAI,GAAG,CAAC;YAC1C,OAAO,IAAI,GAAG,IAAI,CAAC;QACpB,CAAC,CAAC,CAAC;QAEH,OAAO,SAAS,CAAC,CAAC,CAAC,CAAC;IACrB,CAAC;IAED;;OAEG;IACH,UAAU;QACT,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,gBAAgB,CAAC;QAElD,KAAK,MAAM,KAAK,IAAI,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC;YACxD,MAAM,CAAC,GAAG,EAAE,IAAI,CAAC,GAAG,KAAK,CAAC;YAC1B,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CACxB,CAAC,CAAc,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,UAAU,CAAC,IAAI,MAAM,CACtD,CAAC;YACF,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACxB,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YAC1B,CAAC;iBAAM,CAAC;gBACP,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;YAC9B,CAAC;QACF,CAAC;IACF,CAAC;IAED;;OAEG;IACH,cAAc,CAAC,SAAmB;QACjC,MAAM,KAAK,GAAG,CAAC,qBAAqB,EAAE,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;QAE1D,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;YAClC,MAAM,KAAK,GAAG,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;YAC9C,MAAM,MAAM,GAAG,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC;YAE9C,MAAM,MAAM,GAAG,KAAK,CAAC,SAAS;gBAC7B,CAAC,CAAC,WAAW;gBACb,CAAC,CAAC,KAAK,CAAC,OAAO;oBACd,CAAC,CAAC,SAAS;oBACX,CAAC,CAAC,WAAW,CAAC;YAEhB,KAAK,CAAC,IAAI,CAAC,GAAG,QAAQ,KAAK,MAAM,EAAE,CAAC,CAAC;YAErC,IAAI,MAAM,EAAE,CAAC;gBACZ,KAAK,CAAC,IAAI,CAAC,aAAa,MAAM,CAAC,OAAO,IAAI,CAAC,QAAQ,CAAC,CAAC;gBACrD,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;oBACrB,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,QAAQ,CAAC,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;oBACtD,IAAI,IAAI,GAAG,CAAC,EAAE,CAAC;wBACd,KAAK,CAAC,IAAI,CAAC,gBAAgB,IAAI,CAAC,IAAI,CAAC,IAAI,GAAG,KAAK,CAAC,UAAU,CAAC,CAAC;oBAC/D,CAAC;gBACF,CAAC;gBACD,KAAK,CAAC,IAAI,CAAC,aAAa,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;YAC1C,CAAC;YAED,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAChB,CAAC;QAED,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACzB,CAAC;CACD;AAED;;GAEG;AACH,MAAM,UAAU,iBAAiB,CAAC,KAAc;IAC/C,MAAM,CAAC,GAAG,KAAgC,CAAC;IAC3C,MAAM,GAAG,GAAG,MAAM,CAAC,CAAC,CAAC,OAAO,IAAI,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;IAE/C,gFAAgF;IAChF,IACC,GAAG,CAAC,QAAQ,CAAC,MAAM,CAAC;QACpB,GAAG,CAAC,QAAQ,CAAC,OAAO,CAAC;QACrB,GAAG,CAAC,QAAQ,CAAC,YAAY,CAAC,EACzB,CAAC;QACF,4BAA4B;QAC5B,MAAM,UAAU,GAAG,GAAG,CAAC,KAAK,CAAC,wCAAwC,CAAC,CAAC;QACvE,IAAI,YAAgC,CAAC;QAErC,IAAI,UAAU,EAAE,CAAC;YAChB,MAAM,KAAK,GAAG,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YAC1C,MAAM,IAAI,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;YACzC,YAAY,GAAG,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC;gBACvC,CAAC,CAAC,KAAK,GAAG,EAAE,GAAG,IAAI;gBACnB,CAAC,CAAC,KAAK,GAAG,IAAI,CAAC;QACjB,CAAC;QAED,OAAO;YACN,QAAQ,EAAE,SAAS;YACnB,UAAU,EAAE,IAAI;YAChB,OAAO,EAAE,GAAG;YACZ,YAAY,EAAE,CAAC;YACf,SAAS,EAAE,IAAI;YACf,MAAM,EAAE,cAAc;YACtB,UAAU,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACpC,YAAY;SACZ,CAAC;IACH,CAAC;IAED,OAAO,IAAI,CAAC;AACb,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,gBAAgB,CAAC,KAAc;IAC9C,MAAM,CAAC,GAAG,KAAgC,CAAC;IAC3C,MAAM,IAAI,GAAG,MAAM,CAAC,CAAC,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC;IAClC,MAAM,GAAG,GAAG,MAAM,CAAC,CAAC,CAAC,OAAO,IAAI,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;IAE/C,IAAI,IAAI,KAAK,oBAAoB,IAAI,IAAI,KAAK,yBAAyB,EAAE,CAAC;QACzE,OAAO;YACN,QAAQ,EAAE,QAAQ;YAClB,UAAU,EAAE,OAAO;YACnB,OAAO,EAAE,GAAG;YACZ,YAAY,EAAE,CAAC;YACf,SAAS,EAAE,IAAI;YACf,MAAM,EAAE,cAAc;YACtB,UAAU,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;SACpC,CAAC;IACH,CAAC;IAED,IAAI,GAAG,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;QACzB,MAAM,UAAU,GAAG,GAAG,CAAC,KAAK,CAAC,wCAAwC,CAAC,CAAC;QACvE,IAAI,YAAgC,CAAC;QAErC,IAAI,UAAU,EAAE,CAAC;YAChB,MAAM,KAAK,GAAG,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YAC1C,MAAM,IAAI,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;YACzC,YAAY,GAAG,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC;gBACvC,CAAC,CAAC,KAAK,GAAG,EAAE,GAAG,IAAI;gBACnB,CAAC,CAAC,KAAK,GAAG,IAAI,CAAC;QACjB,CAAC;QAED,OAAO;YACN,QAAQ,EAAE,QAAQ;YAClB,UAAU,EAAE,IAAI;YAChB,OAAO,EAAE,GAAG;YACZ,YAAY,EAAE,CAAC;YACf,SAAS,EAAE,IAAI;YACf,MAAM,EAAE,cAAc;YACtB,UAAU,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACpC,YAAY;SACZ,CAAC;IACH,CAAC;IAED,OAAO,IAAI,CAAC;AACb,CAAC"}

package/packages/quota-manager/dist/tui-usage-monitor.d.ts

@@ -0,0 +1,96 @@
+/**
+ * TUI Usage Monitor — Unified TUI Message Parser
+ *
+ * Hooks into pi's TUI messages to capture quota signals from all providers.
+ * Handles OpenAI, GLM, and any other provider that reports via TUI.
+ *
+ * Usage:
+ * ```typescript
+ * const monitor = new TUIUsageMonitor(quotaManager);
+ *
+ * // Hook into pi
+ * pi.on("error", (event) => monitor.processMessage(event.message));
+ * pi.on("message", (event) => monitor.processMessage(event.message));
+ * ```
+ */
+import { EventEmitter } from "node:events";
+import type { QuotaManager } from "./quota-manager.js";
+export interface TUIUsageSignal {
+    provider: "openai" | "glm" | "anthropic" | "openrouter" | "unknown";
+    timestamp: string;
+    /** Usage percentage (0-100), if available */
+    usedPct?: number;
+    /** Remaining percentage (0-100) */
+    remainingPct: number;
+    /** If exhausted */
+    exhausted: boolean;
+    /** Reset time */
+    resetsAt?: string;
+    /** Which limit was hit */
+    limitType: "tokens" | "context_window" | "rate_limit" | "unknown";
+    /** Original message */
+    originalMessage: string;
+}
+export interface TUIUsageMonitorConfig {
+    /** QuotaManager instance to feed signals into */
+    quotaManager: QuotaManager;
+    /** Enable debug logging */
+    debug?: boolean;
+}
+export declare class TUIUsageMonitor extends EventEmitter {
+    private readonly quotaManager;
+    private readonly debug;
+    private lastSignals;
+    constructor(config: TUIUsageMonitorConfig);
+    /**
+     * Process a TUI message and extract quota signals
+     */
+    processMessage(message: string): TUIUsageSignal | null;
+    /**
+     * Detect which provider the message is about
+     */
+    detectProvider(message: string): "openai" | "glm" | "anthropic" | "openrouter" | null;
+    /**
+     * Get the last signal for a provider
+     */
+    getLastSignal(provider: "openai" | "glm" | "anthropic" | "openrouter"): TUIUsageSignal | null;
+    /**
+     * Get all last signals
+     */
+    getAllLastSignals(): Map<string, TUIUsageSignal>;
+    /**
+     * Clear stored signals
+     */
+    clear(): void;
+    /**
+     * Parse the quota signal from message
+     */
+    private parseSignal;
+    /**
+     * Detect the type of limit that was hit
+     */
+    private detectLimitType;
+    /**
+     * Check if message contains exhausted keywords
+     */
+    private containsExhaustedKeywords;
+    /**
+     * Parse reset time from message
+     */
+    private parseResetTime;
+    /**
+     * Record signal to QuotaManager
+     */
+    private recordToQuotaManager;
+}
+/**
+ * Create a TUI usage monitor and automatically hook into pi
+ */
+export declare function createTUIUsageMonitor(quotaManager: QuotaManager, pi: {
+    on: (event: string, handler: (event: {
+        message: string;
+    }) => void) => void;
+}, config?: {
+    debug?: boolean;
+}): TUIUsageMonitor;
+//# sourceMappingURL=tui-usage-monitor.d.ts.map

package/packages/quota-manager/dist/tui-usage-monitor.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tui-usage-monitor.d.ts","sourceRoot":"","sources":["../src/tui-usage-monitor.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAC3C,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAEvD,MAAM,WAAW,cAAc;IAC9B,QAAQ,EAAE,QAAQ,GAAG,KAAK,GAAG,WAAW,GAAG,YAAY,GAAG,SAAS,CAAC;IACpE,SAAS,EAAE,MAAM,CAAC;IAClB,6CAA6C;IAC7C,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,mCAAmC;IACnC,YAAY,EAAE,MAAM,CAAC;IACrB,mBAAmB;IACnB,SAAS,EAAE,OAAO,CAAC;IACnB,iBAAiB;IACjB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,0BAA0B;IAC1B,SAAS,EAAE,QAAQ,GAAG,gBAAgB,GAAG,YAAY,GAAG,SAAS,CAAC;IAClE,uBAAuB;IACvB,eAAe,EAAE,MAAM,CAAC;CACxB;AAED,MAAM,WAAW,qBAAqB;IACrC,iDAAiD;IACjD,YAAY,EAAE,YAAY,CAAC;IAC3B,2BAA2B;IAC3B,KAAK,CAAC,EAAE,OAAO,CAAC;CAChB;AAgDD,qBAAa,eAAgB,SAAQ,YAAY;IAChD,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAe;IAC5C,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAU;IAChC,OAAO,CAAC,WAAW,CAA0C;IAE7D,YAAY,MAAM,EAAE,qBAAqB,EAIxC;IAED;;OAEG;IACH,cAAc,CAAC,OAAO,EAAE,MAAM,GAAG,cAAc,GAAG,IAAI,CA4BrD;IAED;;OAEG;IACH,cAAc,CACb,OAAO,EAAE,MAAM,GACb,QAAQ,GAAG,KAAK,GAAG,WAAW,GAAG,YAAY,GAAG,IAAI,CAStD;IAED;;OAEG;IACH,aAAa,CACZ,QAAQ,EAAE,QAAQ,GAAG,KAAK,GAAG,WAAW,GAAG,YAAY,GACrD,cAAc,GAAG,IAAI,CAEvB;IAED;;OAEG;IACH,iBAAiB,IAAI,GAAG,CAAC,MAAM,EAAE,cAAc,CAAC,CAE/C;IAED;;OAEG;IACH,KAAK,IAAI,IAAI,CAEZ;IAED;;OAEG;IACH,OAAO,CAAC,WAAW;IAqCnB;;OAEG;IACH,OAAO,CAAC,eAAe;IAsCvB;;OAEG;IACH,OAAO,CAAC,yBAAyB;IAcjC;;OAEG;IACH,OAAO,CAAC,cAAc;IAgCtB;;OAEG;IACH,OAAO,CAAC,oBAAoB;CAmB5B;AAED;;GAEG;AACH,wBAAgB,qBAAqB,CACpC,YAAY,EAAE,YAAY,EAC1B,EAAE,EAAE;IACH,EAAE,EAAE,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,KAAK,EAAE;QAAE,OAAO,EAAE,MAAM,CAAA;KAAE,KAAK,IAAI,KAAK,IAAI,CAAC;CAC3E,EACD,MAAM,CAAC,EAAE;IAAE,KAAK,CAAC,EAAE,OAAO,CAAA;CAAE,GAC1B,eAAe,CAgBjB"}