erosolar-cli 1.7.373 → 1.7.377

package/dist/core/costTracker.js

@@ -0,0 +1,285 @@
+/**
+ * Cost Tracker - Claude Code Style
+ *
+ * Tracks token usage and API costs across the session.
+ * Provides real-time cost display and detailed breakdowns.
+ *
+ * Features:
+ * - Real-time token counting (input/output)
+ * - Cost estimation based on model pricing
+ * - Session and lifetime cost tracking
+ * - Detailed breakdown by model and request
+ */
+import { existsSync, readFileSync, writeFileSync, mkdirSync } from 'node:fs';
+import { homedir } from 'node:os';
+import { join } from 'node:path';
+const USAGE_FILE = join(homedir(), '.erosolar', 'usage.json');
+/**
+ * Model pricing (per 1M tokens)
+ */
+const MODEL_PRICING = {
+    // Anthropic
+    'claude-sonnet-4-5-20250929': { input: 3.0, output: 15.0 },
+    'claude-opus-4-20250514': { input: 15.0, output: 75.0 },
+    'claude-haiku-4-5-20251001': { input: 0.25, output: 1.25 },
+    // OpenAI
+    'gpt-5.1-codex-mini': { input: 0.15, output: 0.60 },
+    'gpt-5.1-codex': { input: 2.5, output: 10.0 },
+    'gpt-5.1': { input: 5.0, output: 15.0 },
+    'gpt-5-pro': { input: 15.0, output: 60.0 },
+    'gpt-5-mini': { input: 0.3, output: 1.2 },
+    'gpt-5-nano': { input: 0.075, output: 0.30 },
+    // Google
+    'gemini-2.5-pro': { input: 1.25, output: 5.0 },
+    'gemini-2.5-flash': { input: 0.075, output: 0.30 },
+    // xAI
+    'grok-4': { input: 5.0, output: 15.0 },
+    'grok-4-1-fast-reasoning': { input: 2.0, output: 10.0 },
+    'grok-4-1-fast-non-reasoning': { input: 0.5, output: 2.0 },
+    'grok-code-fast-1': { input: 0.5, output: 2.0 },
+    // DeepSeek
+    'deepseek-reasoner': { input: 0.55, output: 2.19 },
+    'deepseek-chat': { input: 0.27, output: 1.10 },
+    // Default for unknown models
+    'default': { input: 1.0, output: 3.0 },
+};
+/**
+ * Get pricing for a model
+ */
+function getModelPricing(model) {
+    // Try exact match first
+    if (MODEL_PRICING[model]) {
+        return MODEL_PRICING[model];
+    }
+    // Try prefix match (for versioned models)
+    for (const [key, pricing] of Object.entries(MODEL_PRICING)) {
+        if (model.startsWith(key) || key.startsWith(model)) {
+            return pricing;
+        }
+    }
+    return MODEL_PRICING['default'];
+}
+/**
+ * Calculate cost for tokens
+ */
+function calculateCost(model, inputTokens, outputTokens) {
+    const pricing = getModelPricing(model);
+    const inputCost = (inputTokens / 1_000_000) * pricing.input;
+    const outputCost = (outputTokens / 1_000_000) * pricing.output;
+    return {
+        inputCost,
+        outputCost,
+        totalCost: inputCost + outputCost,
+    };
+}
+/**
+ * Cost Tracker class
+ */
+export class CostTracker {
+    sessionId;
+    startTime;
+    requests = [];
+    constructor(sessionId) {
+        this.sessionId = sessionId ?? `session_${Date.now().toString(36)}`;
+        this.startTime = Date.now();
+    }
+    /**
+     * Record a request's token usage
+     */
+    recordUsage(model, inputTokens, outputTokens, requestType = 'chat') {
+        const costs = calculateCost(model, inputTokens, outputTokens);
+        const request = {
+            timestamp: Date.now(),
+            model,
+            inputTokens,
+            outputTokens,
+            inputCost: costs.inputCost,
+            outputCost: costs.outputCost,
+            totalCost: costs.totalCost,
+            requestType,
+        };
+        this.requests.push(request);
+        this.saveToLifetime(request);
+        return request;
+    }
+    /**
+     * Get session usage summary
+     */
+    getSessionUsage() {
+        let totalInputTokens = 0;
+        let totalOutputTokens = 0;
+        let totalCost = 0;
+        for (const req of this.requests) {
+            totalInputTokens += req.inputTokens;
+            totalOutputTokens += req.outputTokens;
+            totalCost += req.totalCost;
+        }
+        return {
+            sessionId: this.sessionId,
+            startTime: this.startTime,
+            requests: [...this.requests],
+            totalInputTokens,
+            totalOutputTokens,
+            totalCost,
+        };
+    }
+    /**
+     * Get lifetime usage
+     */
+    getLifetimeUsage() {
+        try {
+            if (!existsSync(USAGE_FILE)) {
+                return {
+                    totalInputTokens: 0,
+                    totalOutputTokens: 0,
+                    totalCost: 0,
+                    sessions: 0,
+                    firstUse: Date.now(),
+                    lastUse: Date.now(),
+                };
+            }
+            const data = JSON.parse(readFileSync(USAGE_FILE, 'utf8'));
+            return data;
+        }
+        catch {
+            return {
+                totalInputTokens: 0,
+                totalOutputTokens: 0,
+                totalCost: 0,
+                sessions: 0,
+                firstUse: Date.now(),
+                lastUse: Date.now(),
+            };
+        }
+    }
+    /**
+     * Save request to lifetime usage
+     */
+    saveToLifetime(request) {
+        try {
+            const usage = this.getLifetimeUsage();
+            usage.totalInputTokens += request.inputTokens;
+            usage.totalOutputTokens += request.outputTokens;
+            usage.totalCost += request.totalCost;
+            usage.lastUse = Date.now();
+            mkdirSync(join(homedir(), '.erosolar'), { recursive: true });
+            writeFileSync(USAGE_FILE, JSON.stringify(usage, null, 2));
+        }
+        catch {
+            // Ignore save errors
+        }
+    }
+    /**
+     * Format session usage for display
+     */
+    formatSessionUsage() {
+        const usage = this.getSessionUsage();
+        const duration = Math.floor((Date.now() - this.startTime) / 1000);
+        const mins = Math.floor(duration / 60);
+        const secs = duration % 60;
+        const lines = [
+            'Session Usage:',
+            '',
+            `  Duration: ${mins}m ${secs}s`,
+            `  Requests: ${usage.requests.length}`,
+            '',
+            '  Tokens:',
+            `    Input:  ${formatNumber(usage.totalInputTokens)}`,
+            `    Output: ${formatNumber(usage.totalOutputTokens)}`,
+            `    Total:  ${formatNumber(usage.totalInputTokens + usage.totalOutputTokens)}`,
+            '',
+            '  Cost:',
+            `    Input:  $${usage.requests.reduce((sum, r) => sum + r.inputCost, 0).toFixed(4)}`,
+            `    Output: $${usage.requests.reduce((sum, r) => sum + r.outputCost, 0).toFixed(4)}`,
+            `    Total:  $${usage.totalCost.toFixed(4)}`,
+        ];
+        return lines.join('\n');
+    }
+    /**
+     * Format cost breakdown for display
+     */
+    formatCostBreakdown() {
+        const usage = this.getSessionUsage();
+        const lifetime = this.getLifetimeUsage();
+        // Group by model
+        const byModel = new Map();
+        for (const req of usage.requests) {
+            const existing = byModel.get(req.model) ?? { input: 0, output: 0, cost: 0, count: 0 };
+            existing.input += req.inputTokens;
+            existing.output += req.outputTokens;
+            existing.cost += req.totalCost;
+            existing.count += 1;
+            byModel.set(req.model, existing);
+        }
+        const lines = [
+            'Cost Breakdown:',
+            '',
+            '┌─ Session ─────────────────────────────────────┐',
+            `│ Total Cost: $${usage.totalCost.toFixed(4)}`,
+            `│ Tokens: ${formatNumber(usage.totalInputTokens + usage.totalOutputTokens)} (${formatNumber(usage.totalInputTokens)} in / ${formatNumber(usage.totalOutputTokens)} out)`,
+            '│',
+        ];
+        if (byModel.size > 0) {
+            lines.push('│ By Model:');
+            for (const [model, data] of byModel) {
+                const shortModel = model.length > 25 ? model.slice(0, 22) + '...' : model;
+                lines.push(`│   ${shortModel}: $${data.cost.toFixed(4)} (${data.count} req)`);
+            }
+        }
+        lines.push('└───────────────────────────────────────────────┘');
+        lines.push('');
+        lines.push('┌─ Lifetime ────────────────────────────────────┐');
+        lines.push(`│ Total Cost: $${lifetime.totalCost.toFixed(4)}`);
+        lines.push(`│ Total Tokens: ${formatNumber(lifetime.totalInputTokens + lifetime.totalOutputTokens)}`);
+        lines.push(`│ Sessions: ${lifetime.sessions}`);
+        lines.push('└───────────────────────────────────────────────┘');
+        return lines.join('\n');
+    }
+    /**
+     * Format compact cost display for status bar
+     */
+    formatCompactCost() {
+        const usage = this.getSessionUsage();
+        const totalTokens = usage.totalInputTokens + usage.totalOutputTokens;
+        return `$${usage.totalCost.toFixed(3)} · ${formatCompactNumber(totalTokens)} tok`;
+    }
+    /**
+     * Reset session tracking
+     */
+    reset() {
+        this.requests = [];
+        this.startTime = Date.now();
+    }
+}
+/**
+ * Format number with commas
+ */
+function formatNumber(n) {
+    return n.toLocaleString('en-US');
+}
+/**
+ * Format number compactly (1.2k, 3.5M)
+ */
+function formatCompactNumber(n) {
+    if (n >= 1_000_000) {
+        return `${(n / 1_000_000).toFixed(1)}M`;
+    }
+    if (n >= 1_000) {
+        return `${(n / 1_000).toFixed(1)}k`;
+    }
+    return n.toString();
+}
+/**
+ * Global cost tracker instance
+ */
+let globalCostTracker = null;
+export function getCostTracker(sessionId) {
+    if (!globalCostTracker) {
+        globalCostTracker = new CostTracker(sessionId);
+    }
+    return globalCostTracker;
+}
+export function resetCostTracker() {
+    globalCostTracker = null;
+}
+//# sourceMappingURL=costTracker.js.map

package/dist/core/costTracker.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"costTracker.js","sourceRoot":"","sources":["../../src/core/costTracker.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;GAWG;AAEH,OAAO,EAAE,UAAU,EAAE,YAAY,EAAE,aAAa,EAAE,SAAS,EAAE,MAAM,SAAS,CAAC;AAC7E,OAAO,EAAE,OAAO,EAAE,MAAM,SAAS,CAAC;AAClC,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AAEjC,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,EAAE,EAAE,WAAW,EAAE,YAAY,CAAC,CAAC;AAE9D;;GAEG;AACH,MAAM,aAAa,GAAsD;IACvE,YAAY;IACZ,4BAA4B,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE;IAC1D,wBAAwB,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE;IACvD,2BAA2B,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE;IAE1D,SAAS;IACT,oBAAoB,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE;IACnD,eAAe,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE;IAC7C,SAAS,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE;IACvC,WAAW,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE;IAC1C,YAAY,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE;IACzC,YAAY,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE;IAE5C,SAAS;IACT,gBAAgB,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,EAAE;IAC9C,kBAAkB,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE;IAElD,MAAM;IACN,QAAQ,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE;IACtC,yBAAyB,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE;IACvD,6BAA6B,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE;IAC1D,kBAAkB,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE;IAE/C,WAAW;IACX,mBAAmB,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE;IAClD,eAAe,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE;IAE9C,6BAA6B;IAC7B,SAAS,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE;CACvC,CAAC;AAqCF;;GAEG;AACH,SAAS,eAAe,CAAC,KAAa;IACpC,wBAAwB;IACxB,IAAI,aAAa,CAAC,KAAK,CAAC,EAAE,CAAC;QACzB,OAAO,aAAa,CAAC,KAAK,CAAE,CAAC;IAC/B,CAAC;IAED,0CAA0C;IAC1C,KAAK,MAAM,CAAC,GAAG,EAAE,OAAO,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,EAAE,CAAC;QAC3D,IAAI,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,GAAG,CAAC,UAAU,CAAC,KAAK,CAAC,EAAE,CAAC;YACnD,OAAO,OAAO,CAAC;QACjB,CAAC;IACH,CAAC;IAED,OAAO,aAAa,CAAC,SAAS,CAAE,CAAC;AACnC,CAAC;AAED;;GAEG;AACH,SAAS,aAAa,CAAC,KAAa,EAAE,WAAmB,EAAE,YAAoB;IAK7E,MAAM,OAAO,GAAG,eAAe,CAAC,KAAK,CAAC,CAAC;IACvC,MAAM,SAAS,GAAG,CAAC,WAAW,GAAG,SAAS,CAAC,GAAG,OAAO,CAAC,KAAK,CAAC;IAC5D,MAAM,UAAU,GAAG,CAAC,YAAY,GAAG,SAAS,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC;IAC/D,OAAO;QACL,SAAS;QACT,UAAU;QACV,SAAS,EAAE,SAAS,GAAG,UAAU;KAClC,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,OAAO,WAAW;IACd,SAAS,CAAS;IAClB,SAAS,CAAS;IAClB,QAAQ,GAAkB,EAAE,CAAC;IAErC,YAAY,SAAkB;QAC5B,IAAI,CAAC,SAAS,GAAG,SAAS,IAAI,WAAW,IAAI,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,EAAE,CAAC;QACnE,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAC9B,CAAC;IAED;;OAEG;IACH,WAAW,CACT,KAAa,EACb,WAAmB,EACnB,YAAoB,EACpB,cAA6C,MAAM;QAEnD,MAAM,KAAK,GAAG,aAAa,CAAC,KAAK,EAAE,WAAW,EAAE,YAAY,CAAC,CAAC;QAE9D,MAAM,OAAO,GAAgB;YAC3B,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;YACrB,KAAK;YACL,WAAW;YACX,YAAY;YACZ,SAAS,EAAE,KAAK,CAAC,SAAS;YAC1B,UAAU,EAAE,KAAK,CAAC,UAAU;YAC5B,SAAS,EAAE,KAAK,CAAC,SAAS;YAC1B,WAAW;SACZ,CAAC;QAEF,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC5B,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;QAE7B,OAAO,OAAO,CAAC;IACjB,CAAC;IAED;;OAEG;IACH,eAAe;QACb,IAAI,gBAAgB,GAAG,CAAC,CAAC;QACzB,IAAI,iBAAiB,GAAG,CAAC,CAAC;QAC1B,IAAI,SAAS,GAAG,CAAC,CAAC;QAElB,KAAK,MAAM,GAAG,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAChC,gBAAgB,IAAI,GAAG,CAAC,WAAW,CAAC;YACpC,iBAAiB,IAAI,GAAG,CAAC,YAAY,CAAC;YACtC,SAAS,IAAI,GAAG,CAAC,SAAS,CAAC;QAC7B,CAAC;QAED,OAAO;YACL,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,QAAQ,EAAE,CAAC,GAAG,IAAI,CAAC,QAAQ,CAAC;YAC5B,gBAAgB;YAChB,iBAAiB;YACjB,SAAS;SACV,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,gBAAgB;QACd,IAAI,CAAC;YACH,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC;gBAC5B,OAAO;oBACL,gBAAgB,EAAE,CAAC;oBACnB,iBAAiB,EAAE,CAAC;oBACpB,SAAS,EAAE,CAAC;oBACZ,QAAQ,EAAE,CAAC;oBACX,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE;oBACpB,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE;iBACpB,CAAC;YACJ,CAAC;YAED,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC,CAAC;YAC1D,OAAO,IAAqB,CAAC;QAC/B,CAAC;QAAC,MAAM,CAAC;YACP,OAAO;gBACL,gBAAgB,EAAE,CAAC;gBACnB,iBAAiB,EAAE,CAAC;gBACpB,SAAS,EAAE,CAAC;gBACZ,QAAQ,EAAE,CAAC;gBACX,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE;gBACpB,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE;aACpB,CAAC;QACJ,CAAC;IACH,CAAC;IAED;;OAEG;IACK,cAAc,CAAC,OAAoB;QACzC,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAEtC,KAAK,CAAC,gBAAgB,IAAI,OAAO,CAAC,WAAW,CAAC;YAC9C,KAAK,CAAC,iBAAiB,IAAI,OAAO,CAAC,YAAY,CAAC;YAChD,KAAK,CAAC,SAAS,IAAI,OAAO,CAAC,SAAS,CAAC;YACrC,KAAK,CAAC,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAE3B,SAAS,CAAC,IAAI,CAAC,OAAO,EAAE,EAAE,WAAW,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YAC7D,aAAa,CAAC,UAAU,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;QAC5D,CAAC;QAAC,MAAM,CAAC;YACP,qBAAqB;QACvB,CAAC;IACH,CAAC;IAED;;OAEG;IACH,kBAAkB;QAChB,MAAM,KAAK,GAAG,IAAI,CAAC,eAAe,EAAE,CAAC;QACrC,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,IAAI,CAAC,CAAC;QAClE,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,GAAG,EAAE,CAAC,CAAC;QACvC,MAAM,IAAI,GAAG,QAAQ,GAAG,EAAE,CAAC;QAE3B,MAAM,KAAK,GAAa;YACtB,gBAAgB;YAChB,EAAE;YACF,eAAe,IAAI,KAAK,IAAI,GAAG;YAC/B,eAAe,KAAK,CAAC,QAAQ,CAAC,MAAM,EAAE;YACtC,EAAE;YACF,WAAW;YACX,eAAe,YAAY,CAAC,KAAK,CAAC,gBAAgB,CAAC,EAAE;YACrD,eAAe,YAAY,CAAC,KAAK,CAAC,iBAAiB,CAAC,EAAE;YACtD,eAAe,YAAY,CAAC,KAAK,CAAC,gBAAgB,GAAG,KAAK,CAAC,iBAAiB,CAAC,EAAE;YAC/E,EAAE;YACF,SAAS;YACT,gBAAgB,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE;YACpF,gBAAgB,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE;YACrF,gBAAgB,KAAK,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE;SAC7C,CAAC;QAEF,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC1B,CAAC;IAED;;OAEG;IACH,mBAAmB;QACjB,MAAM,KAAK,GAAG,IAAI,CAAC,eAAe,EAAE,CAAC;QACrC,MAAM,QAAQ,GAAG,IAAI,CAAC,gBAAgB,EAAE,CAAC;QAEzC,iBAAiB;QACjB,MAAM,OAAO,GAAG,IAAI,GAAG,EAA0E,CAAC;QAClG,KAAK,MAAM,GAAG,IAAI,KAAK,CAAC,QAAQ,EAAE,CAAC;YACjC,MAAM,QAAQ,GAAG,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC;YACtF,QAAQ,CAAC,KAAK,IAAI,GAAG,CAAC,WAAW,CAAC;YAClC,QAAQ,CAAC,MAAM,IAAI,GAAG,CAAC,YAAY,CAAC;YACpC,QAAQ,CAAC,IAAI,IAAI,GAAG,CAAC,SAAS,CAAC;YAC/B,QAAQ,CAAC,KAAK,IAAI,CAAC,CAAC;YACpB,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC;QACnC,CAAC;QAED,MAAM,KAAK,GAAa;YACtB,iBAAiB;YACjB,EAAE;YACF,mDAAmD;YACnD,kBAAkB,KAAK,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE;YAC9C,aAAa,YAAY,CAAC,KAAK,CAAC,gBAAgB,GAAG,KAAK,CAAC,iBAAiB,CAAC,KAAK,YAAY,CAAC,KAAK,CAAC,gBAAgB,CAAC,SAAS,YAAY,CAAC,KAAK,CAAC,iBAAiB,CAAC,OAAO;YACzK,GAAG;SACJ,CAAC;QAEF,IAAI,OAAO,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;YACrB,KAAK,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;YAC1B,KAAK,MAAM,CAAC,KAAK,EAAE,IAAI,CAAC,IAAI,OAAO,EAAE,CAAC;gBACpC,MAAM,UAAU,GAAG,KAAK,CAAC,MAAM,GAAG,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC;gBAC1E,KAAK,CAAC,IAAI,CAAC,OAAO,UAAU,MAAM,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,IAAI,CAAC,KAAK,OAAO,CAAC,CAAC;YAChF,CAAC;QACH,CAAC;QAED,KAAK,CAAC,IAAI,CAAC,mDAAmD,CAAC,CAAC;QAChE,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACf,KAAK,CAAC,IAAI,CAAC,mDAAmD,CAAC,CAAC;QAChE,KAAK,CAAC,IAAI,CAAC,kBAAkB,QAAQ,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QAC9D,KAAK,CAAC,IAAI,CAAC,mBAAmB,YAAY,CAAC,QAAQ,CAAC,gBAAgB,GAAG,QAAQ,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;QACtG,KAAK,CAAC,IAAI,CAAC,eAAe,QAAQ,CAAC,QAAQ,EAAE,CAAC,CAAC;QAC/C,KAAK,CAAC,IAAI,CAAC,mDAAmD,CAAC,CAAC;QAEhE,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC1B,CAAC;IAED;;OAEG;IACH,iBAAiB;QACf,MAAM,KAAK,GAAG,IAAI,CAAC,eAAe,EAAE,CAAC;QACrC,MAAM,WAAW,GAAG,KAAK,CAAC,gBAAgB,GAAG,KAAK,CAAC,iBAAiB,CAAC;QACrE,OAAO,IAAI,KAAK,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,mBAAmB,CAAC,WAAW,CAAC,MAAM,CAAC;IACpF,CAAC;IAED;;OAEG;IACH,KAAK;QACH,IAAI,CAAC,QAAQ,GAAG,EAAE,CAAC;QACnB,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAC9B,CAAC;CACF;AAED;;GAEG;AACH,SAAS,YAAY,CAAC,CAAS;IAC7B,OAAO,CAAC,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;AACnC,CAAC;AAED;;GAEG;AACH,SAAS,mBAAmB,CAAC,CAAS;IACpC,IAAI,CAAC,IAAI,SAAS,EAAE,CAAC;QACnB,OAAO,GAAG,CAAC,CAAC,GAAG,SAAS,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC;IAC1C,CAAC;IACD,IAAI,CAAC,IAAI,KAAK,EAAE,CAAC;QACf,OAAO,GAAG,CAAC,CAAC,GAAG,KAAK,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC;IACtC,CAAC;IACD,OAAO,CAAC,CAAC,QAAQ,EAAE,CAAC;AACtB,CAAC;AAED;;GAEG;AACH,IAAI,iBAAiB,GAAuB,IAAI,CAAC;AAEjD,MAAM,UAAU,cAAc,CAAC,SAAkB;IAC/C,IAAI,CAAC,iBAAiB,EAAE,CAAC;QACvB,iBAAiB,GAAG,IAAI,WAAW,CAAC,SAAS,CAAC,CAAC;IACjD,CAAC;IACD,OAAO,iBAAiB,CAAC;AAC3B,CAAC;AAED,MAAM,UAAU,gBAAgB;IAC9B,iBAAiB,GAAG,IAAI,CAAC;AAC3B,CAAC"}

package/dist/core/failureRecovery.d.ts

@@ -0,0 +1,122 @@
+/**
+ * Intelligent Failure Detection and Recovery System
+ *
+ * Detects non-working solutions and applies AlphaZero-style learning to:
+ * 1. Identify failure patterns
+ * 2. Generate alternative approaches
+ * 3. Store successful recoveries for reuse
+ * 4. Optionally update source code when patterns are reliable
+ *
+ * Principal Investigator: Bo Shang
+ */
+import { type ToolCallRecord } from './alphaZeroEngine.js';
+export interface FailureSignal {
+    type: 'build' | 'test' | 'tool' | 'user-feedback' | 'verification' | 'loop-detected';
+    severity: 'critical' | 'major' | 'minor';
+    message: string;
+    context?: Record<string, unknown>;
+    timestamp: string;
+}
+export interface RecoveryAttempt {
+    id: string;
+    failureSignal: FailureSignal;
+    originalApproach: string;
+    alternativeApproach: string;
+    success: boolean;
+    qualityBefore: number;
+    qualityAfter: number;
+    timestamp: string;
+}
+export interface RecoveryStrategy {
+    id: string;
+    failureType: string;
+    errorPattern: string;
+    recoverySteps: string[];
+    successRate: number;
+    timesUsed: number;
+    lastSuccess: string;
+}
+export interface SourceCodeFix {
+    id: string;
+    filePath: string;
+    description: string;
+    originalCode: string;
+    fixedCode: string;
+    failureType: string;
+    validatedCount: number;
+    appliedToSource: boolean;
+}
+/**
+ * Detect failure from output/response
+ */
+export declare function detectFailure(output: string, context: {
+    toolCalls?: ToolCallRecord[];
+    userMessage?: string;
+}): FailureSignal | null;
+/**
+ * Record an action and detect if we're in a loop
+ */
+export declare function recordActionAndDetectLoop(action: string): FailureSignal | null;
+/**
+ * Clear action history (call after successful task completion)
+ */
+export declare function clearActionHistory(): void;
+/**
+ * Find a recovery strategy for a failure type
+ */
+export declare function findRecoveryStrategy(failure: FailureSignal): RecoveryStrategy | null;
+/**
+ * Record a successful recovery strategy
+ */
+export declare function recordRecoveryStrategy(failure: FailureSignal, recoverySteps: string[], success: boolean): void;
+/**
+ * Generate a prompt for recovering from failure
+ */
+export declare function generateRecoveryPrompt(failure: FailureSignal, originalApproach: string, taskType: string): string;
+/**
+ * Record a potential source code fix
+ */
+export declare function recordPotentialSourceFix(filePath: string, description: string, originalCode: string, fixedCode: string, failureType: string): void;
+/**
+ * Get all source code fixes
+ */
+export declare function getSourceCodeFixes(): SourceCodeFix[];
+/**
+ * Get validated source fixes that should be applied
+ */
+export declare function getValidatedSourceFixes(minValidations?: number): SourceCodeFix[];
+/**
+ * Mark a fix as applied to source
+ */
+export declare function markFixApplied(fixId: string): void;
+/**
+ * Check if we should suggest applying fixes to erosolar source
+ */
+export declare function suggestSourceUpdates(workingDir: string): {
+    shouldSuggest: boolean;
+    fixes: SourceCodeFix[];
+    reason: string;
+};
+/**
+ * Generate commit message for learning-based updates
+ */
+export declare function generateLearningCommitMessage(fixes: SourceCodeFix[]): string;
+export interface RecoveryResult {
+    recovered: boolean;
+    approach: string;
+    strategy?: RecoveryStrategy;
+    qualityImprovement: number;
+}
+/**
+ * Main entry point for failure recovery
+ * Call this when a failure is detected
+ */
+export declare function initiateRecovery(failure: FailureSignal, originalApproach: string, toolCalls: ToolCallRecord[], userQuery: string): {
+    recoveryPrompt: string;
+    strategy: RecoveryStrategy | null;
+};
+/**
+ * Record successful recovery for future use
+ */
+export declare function recordSuccessfulRecovery(failure: FailureSignal, recoveryApproach: string, toolCalls: ToolCallRecord[], userQuery: string, qualityScore: number): void;
+//# sourceMappingURL=failureRecovery.d.ts.map

package/dist/core/failureRecovery.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"failureRecovery.d.ts","sourceRoot":"","sources":["../../src/core/failureRecovery.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;GAUG;AAaH,OAAO,EAKL,KAAK,cAAc,EACpB,MAAM,sBAAsB,CAAC;AAM9B,MAAM,WAAW,aAAa;IAC5B,IAAI,EAAE,OAAO,GAAG,MAAM,GAAG,MAAM,GAAG,eAAe,GAAG,cAAc,GAAG,eAAe,CAAC;IACrF,QAAQ,EAAE,UAAU,GAAG,OAAO,GAAG,OAAO,CAAC;IACzC,OAAO,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAClC,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,eAAe;IAC9B,EAAE,EAAE,MAAM,CAAC;IACX,aAAa,EAAE,aAAa,CAAC;IAC7B,gBAAgB,EAAE,MAAM,CAAC;IACzB,mBAAmB,EAAE,MAAM,CAAC;IAC5B,OAAO,EAAE,OAAO,CAAC;IACjB,aAAa,EAAE,MAAM,CAAC;IACtB,YAAY,EAAE,MAAM,CAAC;IACrB,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,gBAAgB;IAC/B,EAAE,EAAE,MAAM,CAAC;IACX,WAAW,EAAE,MAAM,CAAC;IACpB,YAAY,EAAE,MAAM,CAAC;IACrB,aAAa,EAAE,MAAM,EAAE,CAAC;IACxB,WAAW,EAAE,MAAM,CAAC;IACpB,SAAS,EAAE,MAAM,CAAC;IAClB,WAAW,EAAE,MAAM,CAAC;CACrB;AAED,MAAM,WAAW,aAAa;IAC5B,EAAE,EAAE,MAAM,CAAC;IACX,QAAQ,EAAE,MAAM,CAAC;IACjB,WAAW,EAAE,MAAM,CAAC;IACpB,YAAY,EAAE,MAAM,CAAC;IACrB,SAAS,EAAE,MAAM,CAAC;IAClB,WAAW,EAAE,MAAM,CAAC;IACpB,cAAc,EAAE,MAAM,CAAC;IACvB,eAAe,EAAE,OAAO,CAAC;CAC1B;AA4CD;;GAEG;AACH,wBAAgB,aAAa,CAC3B,MAAM,EAAE,MAAM,EACd,OAAO,EAAE;IAAE,SAAS,CAAC,EAAE,cAAc,EAAE,CAAC;IAAC,WAAW,CAAC,EAAE,MAAM,CAAA;CAAE,GAC9D,aAAa,GAAG,IAAI,CAqEtB;AA8BD;;GAEG;AACH,wBAAgB,yBAAyB,CAAC,MAAM,EAAE,MAAM,GAAG,aAAa,GAAG,IAAI,CA+B9E;AAED;;GAEG;AACH,wBAAgB,kBAAkB,IAAI,IAAI,CAEzC;AAsCD;;GAEG;AACH,wBAAgB,oBAAoB,CAAC,OAAO,EAAE,aAAa,GAAG,gBAAgB,GAAG,IAAI,CAwBpF;AAED;;GAEG;AACH,wBAAgB,sBAAsB,CACpC,OAAO,EAAE,aAAa,EACtB,aAAa,EAAE,MAAM,EAAE,EACvB,OAAO,EAAE,OAAO,GACf,IAAI,CA0CN;AAeD;;GAEG;AACH,wBAAgB,sBAAsB,CACpC,OAAO,EAAE,aAAa,EACtB,gBAAgB,EAAE,MAAM,EACxB,QAAQ,EAAE,MAAM,GACf,MAAM,CA2CR;AA6BD;;GAEG;AACH,wBAAgB,wBAAwB,CACtC,QAAQ,EAAE,MAAM,EAChB,WAAW,EAAE,MAAM,EACnB,YAAY,EAAE,MAAM,EACpB,SAAS,EAAE,MAAM,EACjB,WAAW,EAAE,MAAM,GAClB,IAAI,CAwBN;AAED;;GAEG;AACH,wBAAgB,kBAAkB,IAAI,aAAa,EAAE,CAGpD;AAED;;GAEG;AACH,wBAAgB,uBAAuB,CAAC,cAAc,GAAE,MAAU,GAAG,aAAa,EAAE,CAGnF;AAED;;GAEG;AACH,wBAAgB,cAAc,CAAC,KAAK,EAAE,MAAM,GAAG,IAAI,CAOlD;AAMD;;GAEG;AACH,wBAAgB,oBAAoB,CAAC,UAAU,EAAE,MAAM,GAAG;IACxD,aAAa,EAAE,OAAO,CAAC;IACvB,KAAK,EAAE,aAAa,EAAE,CAAC;IACvB,MAAM,EAAE,MAAM,CAAC;CAChB,CA2BA;AAED;;GAEG;AACH,wBAAgB,6BAA6B,CAAC,KAAK,EAAE,aAAa,EAAE,GAAG,MAAM,CAc5E;AAMD,MAAM,WAAW,cAAc;IAC7B,SAAS,EAAE,OAAO,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,CAAC,EAAE,gBAAgB,CAAC;IAC5B,kBAAkB,EAAE,MAAM,CAAC;CAC5B;AAED;;;GAGG;AACH,wBAAgB,gBAAgB,CAC9B,OAAO,EAAE,aAAa,EACtB,gBAAgB,EAAE,MAAM,EACxB,SAAS,EAAE,cAAc,EAAE,EAC3B,SAAS,EAAE,MAAM,GAChB;IAAE,cAAc,EAAE,MAAM,CAAC;IAAC,QAAQ,EAAE,gBAAgB,GAAG,IAAI,CAAA;CAAE,CAqB/D;AAED;;GAEG;AACH,wBAAgB,wBAAwB,CACtC,OAAO,EAAE,aAAa,EACtB,gBAAgB,EAAE,MAAM,EACxB,SAAS,EAAE,cAAc,EAAE,EAC3B,SAAS,EAAE,MAAM,EACjB,YAAY,EAAE,MAAM,GACnB,IAAI,CAoBN"}