axon-code 2.4.0

package/dist/models/quota.js

@@ -0,0 +1,397 @@
+/**
+ * 配额管理和成本预警系统
+ * 提供使用限制、成本预警和事件通知功能
+ */
+import { modelStats } from './stats.js';
+/**
+ * 配额管理器
+ */
+export class QuotaManager {
+    limits = {};
+    alerts = [];
+    triggeredAlerts = new Set();
+    thresholdCallbacks = [];
+    limitExceededCallbacks = [];
+    enabled = false;
+    /**
+     * 设置配额限制
+     */
+    setLimit(type, limit) {
+        if (limit < 0) {
+            throw new Error(`Limit must be non-negative, got ${limit}`);
+        }
+        switch (type) {
+            case 'cost':
+                this.limits.maxCost = limit;
+                break;
+            case 'tokens':
+                this.limits.maxTokens = limit;
+                break;
+            case 'requests':
+                this.limits.maxRequests = limit;
+                break;
+        }
+        // 设置限制后自动启用配额管理
+        this.enabled = true;
+        // 清除已触发的预警（允许重新触发）
+        this.triggeredAlerts.clear();
+        // 自动设置默认预警
+        this.setupDefaultAlerts(type, limit);
+    }
+    /**
+     * 设置自动预警阈值
+     */
+    setupDefaultAlerts(type, limit) {
+        // 清除该类型的现有预警
+        this.alerts = this.alerts.filter((alert) => !alert.message.includes(this.getTypeName(type)));
+        const typeName = this.getTypeName(type);
+        const formatter = type === 'cost' ? (v) => `$${v.toFixed(2)}` : (v) => v.toLocaleString();
+        // 50% - 信息提示
+        this.alerts.push({
+            threshold: limit * 0.5,
+            action: 'warn',
+            level: 'info',
+            message: `${typeName} usage reached 50% (${formatter(limit * 0.5)} of ${formatter(limit)})`,
+        });
+        // 80% - 警告
+        this.alerts.push({
+            threshold: limit * 0.8,
+            action: 'warn',
+            level: 'warning',
+            message: `${typeName} usage reached 80% (${formatter(limit * 0.8)} of ${formatter(limit)})`,
+        });
+        // 95% - 严重警告
+        this.alerts.push({
+            threshold: limit * 0.95,
+            action: 'pause',
+            level: 'critical',
+            message: `${typeName} usage reached 95% (${formatter(limit * 0.95)} of ${formatter(limit)}). Consider pausing.`,
+        });
+        // 100% - 超出限制
+        this.alerts.push({
+            threshold: limit,
+            action: 'stop',
+            level: 'exceeded',
+            message: `${typeName} limit exceeded! Reached ${formatter(limit)}.`,
+        });
+    }
+    /**
+     * 获取类型名称
+     */
+    getTypeName(type) {
+        switch (type) {
+            case 'cost':
+                return 'Cost';
+            case 'tokens':
+                return 'Token';
+            case 'requests':
+                return 'Request';
+        }
+    }
+    /**
+     * 获取当前使用情况
+     */
+    getCurrentUsage() {
+        const stats = modelStats.getGlobal();
+        return {
+            currentCost: stats.costUSD,
+            currentTokens: modelStats.getTotalTokens(),
+            currentRequests: stats.apiCalls,
+            maxCost: this.limits.maxCost,
+            maxTokens: this.limits.maxTokens,
+            maxRequests: this.limits.maxRequests,
+        };
+    }
+    /**
+     * 检查是否超出限制
+     */
+    checkLimit() {
+        if (!this.enabled) {
+            return { withinLimit: true, exceeded: [], warnings: [] };
+        }
+        const usage = this.getCurrentUsage();
+        const exceeded = [];
+        const warnings = [];
+        // 检查成本限制
+        if (this.limits.maxCost !== undefined) {
+            const status = this.getBudgetStatus('cost');
+            this.checkThresholds('cost', usage.currentCost, this.limits.maxCost, warnings);
+            if (status.exceeded) {
+                exceeded.push('cost');
+                this.notifyLimitExceeded('cost', status);
+            }
+        }
+        // 检查 Token 限制
+        if (this.limits.maxTokens !== undefined) {
+            const status = this.getBudgetStatus('tokens');
+            this.checkThresholds('tokens', usage.currentTokens, this.limits.maxTokens, warnings);
+            if (status.exceeded) {
+                exceeded.push('tokens');
+                this.notifyLimitExceeded('tokens', status);
+            }
+        }
+        // 检查请求次数限制
+        if (this.limits.maxRequests !== undefined) {
+            const status = this.getBudgetStatus('requests');
+            this.checkThresholds('requests', usage.currentRequests, this.limits.maxRequests, warnings);
+            if (status.exceeded) {
+                exceeded.push('requests');
+                this.notifyLimitExceeded('requests', status);
+            }
+        }
+        return {
+            withinLimit: exceeded.length === 0,
+            exceeded,
+            warnings,
+        };
+    }
+    /**
+     * 检查阈值并触发预警
+     */
+    checkThresholds(type, current, limit, warnings) {
+        for (const alert of this.alerts) {
+            const alertKey = `${type}:${alert.threshold}`;
+            // 如果当前值超过阈值且未触发过
+            if (current >= alert.threshold && !this.triggeredAlerts.has(alertKey)) {
+                warnings.push(alert);
+                this.triggeredAlerts.add(alertKey);
+                // 触发阈值回调
+                const status = this.getBudgetStatus(type);
+                this.notifyThresholdReached(alert, status);
+            }
+        }
+    }
+    /**
+     * 获取使用百分比
+     */
+    getUsagePercentage(type) {
+        if (type) {
+            const status = this.getBudgetStatus(type);
+            return status.percentage;
+        }
+        // 返回所有类型的百分比
+        return {
+            cost: this.getBudgetStatus('cost').percentage,
+            tokens: this.getBudgetStatus('tokens').percentage,
+            requests: this.getBudgetStatus('requests').percentage,
+        };
+    }
+    /**
+     * 获取剩余预算
+     */
+    getRemainingBudget(type) {
+        if (type) {
+            const status = this.getBudgetStatus(type);
+            return status.remaining;
+        }
+        // 返回所有类型的剩余预算
+        return {
+            cost: this.getBudgetStatus('cost').remaining,
+            tokens: this.getBudgetStatus('tokens').remaining,
+            requests: this.getBudgetStatus('requests').remaining,
+        };
+    }
+    /**
+     * 获取预算状态
+     */
+    getBudgetStatus(type) {
+        const usage = this.getCurrentUsage();
+        let current = 0;
+        let limit = 0;
+        switch (type) {
+            case 'cost':
+                current = usage.currentCost;
+                limit = this.limits.maxCost || 0;
+                break;
+            case 'tokens':
+                current = usage.currentTokens;
+                limit = this.limits.maxTokens || 0;
+                break;
+            case 'requests':
+                current = usage.currentRequests;
+                limit = this.limits.maxRequests || 0;
+                break;
+        }
+        // 如果未设置限制，返回无限制状态
+        if (limit === 0) {
+            return {
+                type,
+                current,
+                limit: 0,
+                percentage: 0,
+                remaining: Infinity,
+                alertLevel: 'info',
+                exceeded: false,
+            };
+        }
+        const percentage = Math.min((current / limit) * 100, 100);
+        const remaining = Math.max(limit - current, 0);
+        const exceeded = current >= limit;
+        // 确定预警级别
+        let alertLevel = 'info';
+        if (exceeded) {
+            alertLevel = 'exceeded';
+        }
+        else if (percentage >= 95) {
+            alertLevel = 'critical';
+        }
+        else if (percentage >= 80) {
+            alertLevel = 'warning';
+        }
+        return {
+            type,
+            current,
+            limit,
+            percentage,
+            remaining,
+            alertLevel,
+            exceeded,
+        };
+    }
+    /**
+     * 添加自定义预警
+     */
+    addAlert(alert) {
+        this.alerts.push(alert);
+    }
+    /**
+     * 清除所有预警
+     */
+    clearAlerts() {
+        this.alerts = [];
+        this.triggeredAlerts.clear();
+    }
+    /**
+     * 注册阈值达到回调
+     */
+    onThresholdReached(callback) {
+        this.thresholdCallbacks.push(callback);
+    }
+    /**
+     * 注册限制超出回调
+     */
+    onLimitExceeded(callback) {
+        this.limitExceededCallbacks.push(callback);
+    }
+    /**
+     * 触发阈值达到通知
+     */
+    notifyThresholdReached(alert, status) {
+        for (const callback of this.thresholdCallbacks) {
+            try {
+                callback(alert, status);
+            }
+            catch (error) {
+                console.error('Error in threshold callback:', error);
+            }
+        }
+    }
+    /**
+     * 触发限制超出通知
+     */
+    notifyLimitExceeded(type, status) {
+        for (const callback of this.limitExceededCallbacks) {
+            try {
+                callback(type, status);
+            }
+            catch (error) {
+                console.error('Error in limit exceeded callback:', error);
+            }
+        }
+    }
+    /**
+     * 获取配额摘要
+     */
+    getSummary() {
+        if (!this.enabled) {
+            return 'Quota management disabled';
+        }
+        const lines = ['=== Quota Status ==='];
+        const usage = this.getCurrentUsage();
+        // 成本状态
+        if (this.limits.maxCost !== undefined) {
+            const status = this.getBudgetStatus('cost');
+            const icon = this.getAlertIcon(status.alertLevel);
+            lines.push(`${icon} Cost: $${status.current.toFixed(4)} / $${status.limit.toFixed(2)} (${status.percentage.toFixed(1)}%)`);
+        }
+        // Token 状态
+        if (this.limits.maxTokens !== undefined) {
+            const status = this.getBudgetStatus('tokens');
+            const icon = this.getAlertIcon(status.alertLevel);
+            lines.push(`${icon} Tokens: ${status.current.toLocaleString()} / ${status.limit.toLocaleString()} (${status.percentage.toFixed(1)}%)`);
+        }
+        // 请求状态
+        if (this.limits.maxRequests !== undefined) {
+            const status = this.getBudgetStatus('requests');
+            const icon = this.getAlertIcon(status.alertLevel);
+            lines.push(`${icon} Requests: ${status.current} / ${status.limit} (${status.percentage.toFixed(1)}%)`);
+        }
+        lines.push('====================');
+        return lines.join('\n');
+    }
+    /**
+     * 获取预警图标
+     */
+    getAlertIcon(level) {
+        switch (level) {
+            case 'info':
+                return '✓';
+            case 'warning':
+                return '⚠';
+            case 'critical':
+                return '🔴';
+            case 'exceeded':
+                return '❌';
+        }
+    }
+    /**
+     * 重置配额管理器
+     */
+    reset() {
+        this.limits = {};
+        this.alerts = [];
+        this.triggeredAlerts.clear();
+        this.enabled = false;
+    }
+    /**
+     * 启用/禁用配额管理
+     */
+    setEnabled(enabled) {
+        this.enabled = enabled;
+    }
+    /**
+     * 检查是否启用
+     */
+    isEnabled() {
+        return this.enabled;
+    }
+    /**
+     * 导出配置
+     */
+    exportConfig() {
+        return {
+            limits: { ...this.limits },
+            alerts: [...this.alerts],
+            enabled: this.enabled,
+        };
+    }
+    /**
+     * 导入配置
+     */
+    importConfig(config) {
+        if (config.limits) {
+            this.limits = { ...config.limits };
+        }
+        if (config.alerts) {
+            this.alerts = [...config.alerts];
+        }
+        if (config.enabled !== undefined) {
+            this.enabled = config.enabled;
+        }
+    }
+}
+/**
+ * 全局配额管理器实例
+ */
+export const quotaManager = new QuotaManager();
+//# sourceMappingURL=quota.js.map

package/dist/models/quota.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"quota.js","sourceRoot":"","sources":["../../src/models/quota.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AAaxC;;GAEG;AACH,MAAM,OAAO,YAAY;IACf,MAAM,GAAe,EAAE,CAAC;IACxB,MAAM,GAAgB,EAAE,CAAC;IACzB,eAAe,GAAgB,IAAI,GAAG,EAAE,CAAC;IACzC,kBAAkB,GAAwB,EAAE,CAAC;IAC7C,sBAAsB,GAA4B,EAAE,CAAC;IACrD,OAAO,GAAY,KAAK,CAAC;IAEjC;;OAEG;IACH,QAAQ,CAAC,IAAe,EAAE,KAAa;QACrC,IAAI,KAAK,GAAG,CAAC,EAAE,CAAC;YACd,MAAM,IAAI,KAAK,CAAC,mCAAmC,KAAK,EAAE,CAAC,CAAC;QAC9D,CAAC;QAED,QAAQ,IAAI,EAAE,CAAC;YACb,KAAK,MAAM;gBACT,IAAI,CAAC,MAAM,CAAC,OAAO,GAAG,KAAK,CAAC;gBAC5B,MAAM;YACR,KAAK,QAAQ;gBACX,IAAI,CAAC,MAAM,CAAC,SAAS,GAAG,KAAK,CAAC;gBAC9B,MAAM;YACR,KAAK,UAAU;gBACb,IAAI,CAAC,MAAM,CAAC,WAAW,GAAG,KAAK,CAAC;gBAChC,MAAM;QACV,CAAC;QAED,gBAAgB;QAChB,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;QAEpB,mBAAmB;QACnB,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,CAAC;QAE7B,WAAW;QACX,IAAI,CAAC,kBAAkB,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;IACvC,CAAC;IAED;;OAEG;IACK,kBAAkB,CAAC,IAAe,EAAE,KAAa;QACvD,aAAa;QACb,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,CAC9B,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,CAC3D,CAAC;QAEF,MAAM,QAAQ,GAAG,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;QACxC,MAAM,SAAS,GAAG,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,CAAC,CAAS,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAS,EAAE,EAAE,CAAC,CAAC,CAAC,cAAc,EAAE,CAAC;QAE1G,aAAa;QACb,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC;YACf,SAAS,EAAE,KAAK,GAAG,GAAG;YACtB,MAAM,EAAE,MAAM;YACd,KAAK,EAAE,MAAM;YACb,OAAO,EAAE,GAAG,QAAQ,uBAAuB,SAAS,CAAC,KAAK,GAAG,GAAG,CAAC,OAAO,SAAS,CAAC,KAAK,CAAC,GAAG;SAC5F,CAAC,CAAC;QAEH,WAAW;QACX,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC;YACf,SAAS,EAAE,KAAK,GAAG,GAAG;YACtB,MAAM,EAAE,MAAM;YACd,KAAK,EAAE,SAAS;YAChB,OAAO,EAAE,GAAG,QAAQ,uBAAuB,SAAS,CAAC,KAAK,GAAG,GAAG,CAAC,OAAO,SAAS,CAAC,KAAK,CAAC,GAAG;SAC5F,CAAC,CAAC;QAEH,aAAa;QACb,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC;YACf,SAAS,EAAE,KAAK,GAAG,IAAI;YACvB,MAAM,EAAE,OAAO;YACf,KAAK,EAAE,UAAU;YACjB,OAAO,EAAE,GAAG,QAAQ,uBAAuB,SAAS,CAAC,KAAK,GAAG,IAAI,CAAC,OAAO,SAAS,CAAC,KAAK,CAAC,sBAAsB;SAChH,CAAC,CAAC;QAEH,cAAc;QACd,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC;YACf,SAAS,EAAE,KAAK;YAChB,MAAM,EAAE,MAAM;YACd,KAAK,EAAE,UAAU;YACjB,OAAO,EAAE,GAAG,QAAQ,4BAA4B,SAAS,CAAC,KAAK,CAAC,GAAG;SACpE,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACK,WAAW,CAAC,IAAe;QACjC,QAAQ,IAAI,EAAE,CAAC;YACb,KAAK,MAAM;gBACT,OAAO,MAAM,CAAC;YAChB,KAAK,QAAQ;gBACX,OAAO,OAAO,CAAC;YACjB,KAAK,UAAU;gBACb,OAAO,SAAS,CAAC;QACrB,CAAC;IACH,CAAC;IAED;;OAEG;IACH,eAAe;QACb,MAAM,KAAK,GAAG,UAAU,CAAC,SAAS,EAAE,CAAC;QAErC,OAAO;YACL,WAAW,EAAE,KAAK,CAAC,OAAO;YAC1B,aAAa,EAAE,UAAU,CAAC,cAAc,EAAE;YAC1C,eAAe,EAAE,KAAK,CAAC,QAAQ;YAC/B,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,OAAO;YAC5B,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS;YAChC,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW;SACrC,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,UAAU;QAKR,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YAClB,OAAO,EAAE,WAAW,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC;QAC3D,CAAC;QAED,MAAM,KAAK,GAAG,IAAI,CAAC,eAAe,EAAE,CAAC;QACrC,MAAM,QAAQ,GAAgB,EAAE,CAAC;QACjC,MAAM,QAAQ,GAAgB,EAAE,CAAC;QAEjC,SAAS;QACT,IAAI,IAAI,CAAC,MAAM,CAAC,OAAO,KAAK,SAAS,EAAE,CAAC;YACtC,MAAM,MAAM,GAAG,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC;YAC5C,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,KAAK,CAAC,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;YAE/E,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;gBACpB,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;gBACtB,IAAI,CAAC,mBAAmB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;YAC3C,CAAC;QACH,CAAC;QAED,cAAc;QACd,IAAI,IAAI,CAAC,MAAM,CAAC,SAAS,KAAK,SAAS,EAAE,CAAC;YACxC,MAAM,MAAM,GAAG,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC;YAC9C,IAAI,CAAC,eAAe,CAAC,QAAQ,EAAE,KAAK,CAAC,aAAa,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;YAErF,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;gBACpB,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;gBACxB,IAAI,CAAC,mBAAmB,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;YAC7C,CAAC;QACH,CAAC;QAED,WAAW;QACX,IAAI,IAAI,CAAC,MAAM,CAAC,WAAW,KAAK,SAAS,EAAE,CAAC;YAC1C,MAAM,MAAM,GAAG,IAAI,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC;YAChD,IAAI,CAAC,eAAe,CAAC,UAAU,EAAE,KAAK,CAAC,eAAe,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAC;YAE3F,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;gBACpB,QAAQ,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;gBAC1B,IAAI,CAAC,mBAAmB,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;YAC/C,CAAC;QACH,CAAC;QAED,OAAO;YACL,WAAW,EAAE,QAAQ,CAAC,MAAM,KAAK,CAAC;YAClC,QAAQ;YACR,QAAQ;SACT,CAAC;IACJ,CAAC;IAED;;OAEG;IACK,eAAe,CACrB,IAAe,EACf,OAAe,EACf,KAAa,EACb,QAAqB;QAErB,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChC,MAAM,QAAQ,GAAG,GAAG,IAAI,IAAI,KAAK,CAAC,SAAS,EAAE,CAAC;YAE9C,iBAAiB;YACjB,IAAI,OAAO,IAAI,KAAK,CAAC,SAAS,IAAI,CAAC,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;gBACtE,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;gBACrB,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;gBAEnC,SAAS;gBACT,MAAM,MAAM,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;gBAC1C,IAAI,CAAC,sBAAsB,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;YAC7C,CAAC;QACH,CAAC;IACH,CAAC;IAED;;OAEG;IACH,kBAAkB,CAAC,IAAgB;QACjC,IAAI,IAAI,EAAE,CAAC;YACT,MAAM,MAAM,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;YAC1C,OAAO,MAAM,CAAC,UAAU,CAAC;QAC3B,CAAC;QAED,aAAa;QACb,OAAO;YACL,IAAI,EAAE,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC,UAAU;YAC7C,MAAM,EAAE,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC,UAAU;YACjD,QAAQ,EAAE,IAAI,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC,UAAU;SACtD,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,kBAAkB,CAAC,IAAgB;QACjC,IAAI,IAAI,EAAE,CAAC;YACT,MAAM,MAAM,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;YAC1C,OAAO,MAAM,CAAC,SAAS,CAAC;QAC1B,CAAC;QAED,cAAc;QACd,OAAO;YACL,IAAI,EAAE,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC,SAAS;YAC5C,MAAM,EAAE,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC,SAAS;YAChD,QAAQ,EAAE,IAAI,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC,SAAS;SACrD,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,eAAe,CAAC,IAAe;QAC7B,MAAM,KAAK,GAAG,IAAI,CAAC,eAAe,EAAE,CAAC;QACrC,IAAI,OAAO,GAAG,CAAC,CAAC;QAChB,IAAI,KAAK,GAAG,CAAC,CAAC;QAEd,QAAQ,IAAI,EAAE,CAAC;YACb,KAAK,MAAM;gBACT,OAAO,GAAG,KAAK,CAAC,WAAW,CAAC;gBAC5B,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,OAAO,IAAI,CAAC,CAAC;gBACjC,MAAM;YACR,KAAK,QAAQ;gBACX,OAAO,GAAG,KAAK,CAAC,aAAa,CAAC;gBAC9B,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,IAAI,CAAC,CAAC;gBACnC,MAAM;YACR,KAAK,UAAU;gBACb,OAAO,GAAG,KAAK,CAAC,eAAe,CAAC;gBAChC,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,WAAW,IAAI,CAAC,CAAC;gBACrC,MAAM;QACV,CAAC;QAED,kBAAkB;QAClB,IAAI,KAAK,KAAK,CAAC,EAAE,CAAC;YAChB,OAAO;gBACL,IAAI;gBACJ,OAAO;gBACP,KAAK,EAAE,CAAC;gBACR,UAAU,EAAE,CAAC;gBACb,SAAS,EAAE,QAAQ;gBACnB,UAAU,EAAE,MAAM;gBAClB,QAAQ,EAAE,KAAK;aAChB,CAAC;QACJ,CAAC;QAED,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,OAAO,GAAG,KAAK,CAAC,GAAG,GAAG,EAAE,GAAG,CAAC,CAAC;QAC1D,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,KAAK,GAAG,OAAO,EAAE,CAAC,CAAC,CAAC;QAC/C,MAAM,QAAQ,GAAG,OAAO,IAAI,KAAK,CAAC;QAElC,SAAS;QACT,IAAI,UAAU,GAAe,MAAM,CAAC;QACpC,IAAI,QAAQ,EAAE,CAAC;YACb,UAAU,GAAG,UAAU,CAAC;QAC1B,CAAC;aAAM,IAAI,UAAU,IAAI,EAAE,EAAE,CAAC;YAC5B,UAAU,GAAG,UAAU,CAAC;QAC1B,CAAC;aAAM,IAAI,UAAU,IAAI,EAAE,EAAE,CAAC;YAC5B,UAAU,GAAG,SAAS,CAAC;QACzB,CAAC;QAED,OAAO;YACL,IAAI;YACJ,OAAO;YACP,KAAK;YACL,UAAU;YACV,SAAS;YACT,UAAU;YACV,QAAQ;SACT,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,QAAQ,CAAC,KAAgB;QACvB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC1B,CAAC;IAED;;OAEG;IACH,WAAW;QACT,IAAI,CAAC,MAAM,GAAG,EAAE,CAAC;QACjB,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,CAAC;IAC/B,CAAC;IAED;;OAEG;IACH,kBAAkB,CAAC,QAA2B;QAC5C,IAAI,CAAC,kBAAkB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IACzC,CAAC;IAED;;OAEG;IACH,eAAe,CAAC,QAA+B;QAC7C,IAAI,CAAC,sBAAsB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC7C,CAAC;IAED;;OAEG;IACK,sBAAsB,CAAC,KAAgB,EAAE,MAAoB;QACnE,KAAK,MAAM,QAAQ,IAAI,IAAI,CAAC,kBAAkB,EAAE,CAAC;YAC/C,IAAI,CAAC;gBACH,QAAQ,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;YAC1B,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,KAAK,CAAC,8BAA8B,EAAE,KAAK,CAAC,CAAC;YACvD,CAAC;QACH,CAAC;IACH,CAAC;IAED;;OAEG;IACK,mBAAmB,CAAC,IAAe,EAAE,MAAoB;QAC/D,KAAK,MAAM,QAAQ,IAAI,IAAI,CAAC,sBAAsB,EAAE,CAAC;YACnD,IAAI,CAAC;gBACH,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;YACzB,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,KAAK,CAAC,mCAAmC,EAAE,KAAK,CAAC,CAAC;YAC5D,CAAC;QACH,CAAC;IACH,CAAC;IAED;;OAEG;IACH,UAAU;QACR,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YAClB,OAAO,2BAA2B,CAAC;QACrC,CAAC;QAED,MAAM,KAAK,GAAa,CAAC,sBAAsB,CAAC,CAAC;QACjD,MAAM,KAAK,GAAG,IAAI,CAAC,eAAe,EAAE,CAAC;QAErC,OAAO;QACP,IAAI,IAAI,CAAC,MAAM,CAAC,OAAO,KAAK,SAAS,EAAE,CAAC;YACtC,MAAM,MAAM,GAAG,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC;YAC5C,MAAM,IAAI,GAAG,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;YAClD,KAAK,CAAC,IAAI,CACR,GAAG,IAAI,WAAW,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,MAAM,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAC/G,CAAC;QACJ,CAAC;QAED,WAAW;QACX,IAAI,IAAI,CAAC,MAAM,CAAC,SAAS,KAAK,SAAS,EAAE,CAAC;YACxC,MAAM,MAAM,GAAG,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC;YAC9C,MAAM,IAAI,GAAG,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;YAClD,KAAK,CAAC,IAAI,CACR,GAAG,IAAI,YAAY,MAAM,CAAC,OAAO,CAAC,cAAc,EAAE,MAAM,MAAM,CAAC,KAAK,CAAC,cAAc,EAAE,KAAK,MAAM,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAC3H,CAAC;QACJ,CAAC;QAED,OAAO;QACP,IAAI,IAAI,CAAC,MAAM,CAAC,WAAW,KAAK,SAAS,EAAE,CAAC;YAC1C,MAAM,MAAM,GAAG,IAAI,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC;YAChD,MAAM,IAAI,GAAG,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;YAClD,KAAK,CAAC,IAAI,CACR,GAAG,IAAI,cAAc,MAAM,CAAC,OAAO,MAAM,MAAM,CAAC,KAAK,KAAK,MAAM,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAC3F,CAAC;QACJ,CAAC;QAED,KAAK,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;QACnC,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC1B,CAAC;IAED;;OAEG;IACK,YAAY,CAAC,KAAiB;QACpC,QAAQ,KAAK,EAAE,CAAC;YACd,KAAK,MAAM;gBACT,OAAO,GAAG,CAAC;YACb,KAAK,SAAS;gBACZ,OAAO,GAAG,CAAC;YACb,KAAK,UAAU;gBACb,OAAO,IAAI,CAAC;YACd,KAAK,UAAU;gBACb,OAAO,GAAG,CAAC;QACf,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK;QACH,IAAI,CAAC,MAAM,GAAG,EAAE,CAAC;QACjB,IAAI,CAAC,MAAM,GAAG,EAAE,CAAC;QACjB,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,CAAC;QAC7B,IAAI,CAAC,OAAO,GAAG,KAAK,CAAC;IACvB,CAAC;IAED;;OAEG;IACH,UAAU,CAAC,OAAgB;QACzB,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IACzB,CAAC;IAED;;OAEG;IACH,SAAS;QACP,OAAO,IAAI,CAAC,OAAO,CAAC;IACtB,CAAC;IAED;;OAEG;IACH,YAAY;QAKV,OAAO;YACL,MAAM,EAAE,EAAE,GAAG,IAAI,CAAC,MAAM,EAAE;YAC1B,MAAM,EAAE,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC;YACxB,OAAO,EAAE,IAAI,CAAC,OAAO;SACtB,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,YAAY,CAAC,MAIZ;QACC,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;YAClB,IAAI,CAAC,MAAM,GAAG,EAAE,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC;QACrC,CAAC;QACD,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;YAClB,IAAI,CAAC,MAAM,GAAG,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC;QACnC,CAAC;QACD,IAAI,MAAM,CAAC,OAAO,KAAK,SAAS,EAAE,CAAC;YACjC,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC;QAChC,CAAC;IACH,CAAC;CACF;AAED;;GAEG;AACH,MAAM,CAAC,MAAM,YAAY,GAAG,IAAI,YAAY,EAAE,CAAC"}

package/dist/models/stats.d.ts

@@ -0,0 +1,96 @@
+/**
+ * 模型使用统计
+ * 按模型分类记录使用情况
+ */
+import type { ModelUsageStats } from './types.js';
+/**
+ * 模型统计管理器
+ */
+export declare class ModelStats {
+    private statsPerModel;
+    private globalStats;
+    private startTime;
+    /**
+     * 记录 API 调用
+     */
+    record(modelId: string, usage: {
+        inputTokens: number;
+        outputTokens: number;
+        cacheReadTokens?: number;
+        cacheCreationTokens?: number;
+        thinkingTokens?: number;
+        webSearchRequests?: number;
+        apiDurationMs?: number;
+        toolDurationMs?: number;
+    }): void;
+    /**
+     * 获取特定模型的统计
+     */
+    getByModel(modelId: string): ModelUsageStats;
+    /**
+     * 获取全局统计
+     */
+    getGlobal(): ModelUsageStats;
+    /**
+     * 获取所有模型的统计
+     */
+    getAllByModel(): Map<string, ModelUsageStats>;
+    /**
+     * 获取总成本
+     */
+    getTotalCost(): number;
+    /**
+     * 获取格式化的成本字符串
+     */
+    getFormattedCost(): string;
+    /**
+     * 获取总 tokens
+     */
+    getTotalTokens(): number;
+    /**
+     * 获取会话持续时间
+     */
+    getSessionDuration(): number;
+    /**
+     * 获取性能指标
+     */
+    getPerformanceMetrics(): {
+        averageApiLatency: number;
+        averageToolLatency: number;
+        tokensPerSecond: number;
+        cacheHitRate: number;
+    };
+    /**
+     * 获取模型使用分布
+     */
+    getModelDistribution(): Array<{
+        model: string;
+        displayName: string;
+        percentage: number;
+        tokens: number;
+        cost: number;
+        calls: number;
+    }>;
+    /**
+     * 获取摘要
+     */
+    getSummary(): string;
+    /**
+     * 重置统计
+     */
+    reset(): void;
+    /**
+     * 导出统计数据
+     */
+    export(): {
+        global: ModelUsageStats;
+        byModel: Record<string, ModelUsageStats>;
+        startTime: number;
+        duration: number;
+    };
+}
+/**
+ * 全局模型统计实例
+ */
+export declare const modelStats: ModelStats;
+//# sourceMappingURL=stats.d.ts.map

package/dist/models/stats.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"stats.d.ts","sourceRoot":"","sources":["../../src/models/stats.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,KAAK,EAAE,eAAe,EAAgB,MAAM,YAAY,CAAC;AAsBhE;;GAEG;AACH,qBAAa,UAAU;IACrB,OAAO,CAAC,aAAa,CAA2C;IAChE,OAAO,CAAC,WAAW,CAAuC;IAC1D,OAAO,CAAC,SAAS,CAAsB;IAEvC;;OAEG;IACH,MAAM,CACJ,OAAO,EAAE,MAAM,EACf,KAAK,EAAE;QACL,WAAW,EAAE,MAAM,CAAC;QACpB,YAAY,EAAE,MAAM,CAAC;QACrB,eAAe,CAAC,EAAE,MAAM,CAAC;QACzB,mBAAmB,CAAC,EAAE,MAAM,CAAC;QAC7B,cAAc,CAAC,EAAE,MAAM,CAAC;QACxB,iBAAiB,CAAC,EAAE,MAAM,CAAC;QAC3B,aAAa,CAAC,EAAE,MAAM,CAAC;QACvB,cAAc,CAAC,EAAE,MAAM,CAAC;KACzB,GACA,IAAI;IAmDP;;OAEG;IACH,UAAU,CAAC,OAAO,EAAE,MAAM,GAAG,eAAe;IAK5C;;OAEG;IACH,SAAS,IAAI,eAAe;IAI5B;;OAEG;IACH,aAAa,IAAI,GAAG,CAAC,MAAM,EAAE,eAAe,CAAC;IAI7C;;OAEG;IACH,YAAY,IAAI,MAAM;IAItB;;OAEG;IACH,gBAAgB,IAAI,MAAM;IAQ1B;;OAEG;IACH,cAAc,IAAI,MAAM;IASxB;;OAEG;IACH,kBAAkB,IAAI,MAAM;IAI5B;;OAEG;IACH,qBAAqB,IAAI;QACvB,iBAAiB,EAAE,MAAM,CAAC;QAC1B,kBAAkB,EAAE,MAAM,CAAC;QAC3B,eAAe,EAAE,MAAM,CAAC;QACxB,YAAY,EAAE,MAAM,CAAC;KACtB;IA0BD;;OAEG;IACH,oBAAoB,IAAI,KAAK,CAAC;QAC5B,KAAK,EAAE,MAAM,CAAC;QACd,WAAW,EAAE,MAAM,CAAC;QACpB,UAAU,EAAE,MAAM,CAAC;QACnB,MAAM,EAAE,MAAM,CAAC;QACf,IAAI,EAAE,MAAM,CAAC;QACb,KAAK,EAAE,MAAM,CAAC;KACf,CAAC;IA2BF;;OAEG;IACH,UAAU,IAAI,MAAM;IA2BpB;;OAEG;IACH,KAAK,IAAI,IAAI;IAMb;;OAEG;IACH,MAAM,IAAI;QACR,MAAM,EAAE,eAAe,CAAC;QACxB,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,eAAe,CAAC,CAAC;QACzC,SAAS,EAAE,MAAM,CAAC;QAClB,QAAQ,EAAE,MAAM,CAAC;KAClB;CAaF;AAED;;GAEG;AACH,eAAO,MAAM,UAAU,YAAmB,CAAC"}