cost-tracker-pro
技能简介
企业级 AI 模型成本管控工具,实时统计模型消耗、计算使用费用、实现成本管控,支持预算管理和优化建议。核心功能
1. 成本统计:实时统计 AI 模型调用成本 2. 预算管理:设置预算和告警阈值 3. 优化建议:基于使用模式提供优化建议 4. 报告生成:自动生成成本分析报告 5. 多模型支持:支持主流 AI 模型成本跟踪支持的 AI 模型
安装方法
```bash npm install -g cost-tracker-pro ```使用示例
```javascript const { CostTracker } = require('cost-tracker-pro');// 初始化成本跟踪器 const tracker = new CostTracker({ models: { 'gpt-4o': { provider: 'openai', pricePerToken: 0.000005 }, 'claude-3-opus': { provider: 'anthropic', pricePerToken: 0.000015 }, 'gemini-pro': { provider: 'google', pricePerToken: 0.0000015 } }, budget: { monthly: 1000, // 每月预算 alertThreshold: 0.8, // 告警阈值 80% dailyLimit: 50 // 每日限制 } });
// 跟踪单个调用 async function trackAPICall(model, tokens, metadata = {}) { const cost = await tracker.record({ model: model, inputTokens: tokens.input, outputTokens: tokens.output, timestamp: new Date(), userId: metadata.userId, projectId: metadata.projectId });
// 检查预算 const budgetStatus = await tracker.checkBudget(); if (budgetStatus.exceeded) { console.warn(`预算超出: ${budgetStatus.current}/${budgetStatus.limit}`); }
return cost; }
// 生成成本报告 async function generateCostReport(period = 'monthly') { const report = await tracker.generateReport({ period: period, breakdownBy: ['model', 'user', 'project'], includeRecommendations: true, includeTrends: true });
// 优化建议 const recommendations = report.recommendations; // 成本趋势 const trends = report.trends; // 详细数据 const detailedData = report.data;
return report; } ```