mcp-cost-tracker 1.0.0

package/dist/tools/index.js

@@ -0,0 +1,660 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.registerAllTools = registerAllTools;
+const zod_1 = require("zod");
+const models_js_1 = require("../pricing/models.js");
+const generator_js_1 = require("../dashboard/generator.js");
+const helpers_js_1 = require("../utils/helpers.js");
+const fs_1 = __importDefault(require("fs"));
+const path_1 = __importDefault(require("path"));
+const os_1 = __importDefault(require("os"));
+function registerAllTools(server, db) {
+    // ============================================================
+    // 1. track_usage — Record an API call's token usage and cost
+    // ============================================================
+    server.registerTool('track_usage', {
+        title: 'Track API Usage',
+        description: 'Record an LLM API call with token counts. Automatically calculates cost based on built-in pricing for 50+ models across OpenAI, Anthropic, Google, DeepSeek, Mistral, Meta, and more.',
+        inputSchema: zod_1.z.object({
+            provider: zod_1.z.string().describe('LLM provider name (e.g., openai, anthropic, google, deepseek, mistral)'),
+            model: zod_1.z.string().describe('Model identifier (e.g., gpt-4o, claude-sonnet-4, gemini-2.5-flash)'),
+            input_tokens: zod_1.z.number().int().min(0).describe('Number of input/prompt tokens'),
+            output_tokens: zod_1.z.number().int().min(0).describe('Number of output/completion tokens'),
+            cached_input_tokens: zod_1.z.number().int().min(0).optional().default(0).describe('Number of cached input tokens (if applicable)'),
+            session_id: zod_1.z.string().optional().default('').describe('Session or conversation ID for grouping'),
+            project: zod_1.z.string().optional().default('default').describe('Project name for categorization'),
+            description: zod_1.z.string().optional().default('').describe('Optional description of the API call'),
+            metadata: zod_1.z.string().optional().default('{}').describe('Optional JSON metadata'),
+        }),
+        annotations: {
+            title: 'Track API Usage',
+            readOnlyHint: false,
+            destructiveHint: false,
+            idempotentHint: false,
+            openWorldHint: false,
+        },
+    }, async (args) => {
+        const { provider, model, input_tokens, output_tokens, cached_input_tokens, session_id, project, description, metadata } = args;
+        const costResult = (0, models_js_1.calculateCost)(model, input_tokens, output_tokens, provider, cached_input_tokens);
+        const record = db.recordUsage({
+            timestamp: new Date().toISOString(),
+            provider: provider.toLowerCase(),
+            model,
+            input_tokens,
+            output_tokens,
+            cached_input_tokens: cached_input_tokens || 0,
+            total_tokens: input_tokens + output_tokens,
+            input_cost: costResult.inputCost,
+            output_cost: costResult.outputCost,
+            cached_input_cost: costResult.cachedInputCost,
+            total_cost: costResult.cost,
+            session_id: session_id || '',
+            project: project || 'default',
+            description: description || '',
+            metadata: metadata || '{}',
+        });
+        const pricingInfo = costResult.pricing
+            ? `Pricing: ${costResult.pricing.model} @ $${costResult.pricing.inputPricePerMTok}/$${costResult.pricing.outputPricePerMTok} per MTok`
+            : '⚠️ No pricing found for this model. Cost recorded as $0. Use set_custom_pricing to add pricing.';
+        return {
+            content: [{
+                    type: 'text',
+                    text: [
+                        `✅ Usage tracked successfully (ID: ${record.id})`,
+                        ``,
+                        `Provider: ${provider}`,
+                        `Model: ${model}`,
+                        `Input tokens: ${(0, helpers_js_1.formatTokens)(input_tokens)}`,
+                        `Output tokens: ${(0, helpers_js_1.formatTokens)(output_tokens)}`,
+                        cached_input_tokens ? `Cached tokens: ${(0, helpers_js_1.formatTokens)(cached_input_tokens)}` : null,
+                        ``,
+                        `💰 Cost breakdown:`,
+                        `  Input cost:  ${(0, helpers_js_1.formatCost)(costResult.inputCost)}`,
+                        `  Output cost: ${(0, helpers_js_1.formatCost)(costResult.outputCost)}`,
+                        costResult.cachedInputCost > 0 ? `  Cache cost:  ${(0, helpers_js_1.formatCost)(costResult.cachedInputCost)}` : null,
+                        `  Total cost:  ${(0, helpers_js_1.formatCost)(costResult.cost)}`,
+                        ``,
+                        pricingInfo,
+                    ].filter(Boolean).join('\n'),
+                }],
+        };
+    });
+    // ============================================================
+    // 2. get_summary — Get cost summary with filters
+    // ============================================================
+    server.registerTool('get_summary', {
+        title: 'Get Cost Summary',
+        description: 'Get a summary of total costs, tokens, and request counts. Filter by date range, provider, or project.',
+        inputSchema: zod_1.z.object({
+            start_date: zod_1.z.string().optional().describe('Start date (ISO 8601 format, e.g., 2025-01-01)'),
+            end_date: zod_1.z.string().optional().describe('End date (ISO 8601 format)'),
+            provider: zod_1.z.string().optional().describe('Filter by provider'),
+            project: zod_1.z.string().optional().describe('Filter by project'),
+        }),
+        annotations: { readOnlyHint: true, destructiveHint: false, idempotentHint: true, openWorldHint: false },
+    }, async (args) => {
+        const summary = db.getSummary({
+            startDate: args.start_date,
+            endDate: args.end_date,
+            provider: args.provider,
+            project: args.project,
+        });
+        return {
+            content: [{
+                    type: 'text',
+                    text: [
+                        `📊 Cost Summary`,
+                        args.start_date || args.end_date ? `Period: ${args.start_date || '∞'} → ${args.end_date || 'now'}` : 'Period: All time',
+                        args.provider ? `Provider: ${args.provider}` : null,
+                        args.project ? `Project: ${args.project}` : null,
+                        ``,
+                        `| Metric | Value |`,
+                        `|--------|-------|`,
+                        `| Total Spend | ${(0, helpers_js_1.formatCost)(summary.total_cost)} |`,
+                        `| Total Requests | ${summary.total_requests.toLocaleString()} |`,
+                        `| Input Tokens | ${(0, helpers_js_1.formatTokens)(summary.total_input_tokens)} |`,
+                        `| Output Tokens | ${(0, helpers_js_1.formatTokens)(summary.total_output_tokens)} |`,
+                        `| Cached Tokens | ${(0, helpers_js_1.formatTokens)(summary.total_cached_tokens)} |`,
+                        `| Avg Cost/Request | ${(0, helpers_js_1.formatCost)(summary.avg_cost_per_request)} |`,
+                    ].filter(Boolean).join('\n'),
+                }],
+        };
+    });
+    // ============================================================
+    // 3. get_cost_by_model — Breakdown by model
+    // ============================================================
+    server.registerTool('get_cost_by_model', {
+        title: 'Cost by Model',
+        description: 'Get cost breakdown grouped by model. See which models are costing you the most.',
+        inputSchema: zod_1.z.object({
+            start_date: zod_1.z.string().optional().describe('Start date filter'),
+            end_date: zod_1.z.string().optional().describe('End date filter'),
+            project: zod_1.z.string().optional().describe('Filter by project'),
+        }),
+        annotations: { readOnlyHint: true, destructiveHint: false, idempotentHint: true, openWorldHint: false },
+    }, async (args) => {
+        const data = db.getCostByModel({
+            startDate: args.start_date,
+            endDate: args.end_date,
+            project: args.project,
+        });
+        if (data.length === 0) {
+            return { content: [{ type: 'text', text: 'No usage data found for the specified filters.' }] };
+        }
+        let table = `📊 Cost by Model\n\n`;
+        table += `| Provider | Model | Cost | Requests | Input Tokens | Output Tokens |\n`;
+        table += `|----------|-------|------|----------|--------------|---------------|\n`;
+        for (const m of data) {
+            table += `| ${m.provider} | ${m.model} | ${(0, helpers_js_1.formatCost)(m.total_cost)} | ${m.request_count} | ${(0, helpers_js_1.formatTokens)(m.total_input_tokens)} | ${(0, helpers_js_1.formatTokens)(m.total_output_tokens)} |\n`;
+        }
+        return { content: [{ type: 'text', text: table }] };
+    });
+    // ============================================================
+    // 4. get_cost_by_provider — Breakdown by provider
+    // ============================================================
+    server.registerTool('get_cost_by_provider', {
+        title: 'Cost by Provider',
+        description: 'Get cost breakdown grouped by provider (OpenAI, Anthropic, Google, etc.).',
+        inputSchema: zod_1.z.object({
+            start_date: zod_1.z.string().optional().describe('Start date filter'),
+            end_date: zod_1.z.string().optional().describe('End date filter'),
+        }),
+        annotations: { readOnlyHint: true, destructiveHint: false, idempotentHint: true, openWorldHint: false },
+    }, async (args) => {
+        const data = db.getCostByProvider({
+            startDate: args.start_date,
+            endDate: args.end_date,
+        });
+        if (data.length === 0) {
+            return { content: [{ type: 'text', text: 'No usage data found.' }] };
+        }
+        let table = `📊 Cost by Provider\n\n`;
+        table += `| Provider | Cost | Requests | Input Tokens | Output Tokens |\n`;
+        table += `|----------|------|----------|--------------|---------------|\n`;
+        for (const p of data) {
+            table += `| ${p.provider} | ${(0, helpers_js_1.formatCost)(p.total_cost)} | ${p.request_count} | ${(0, helpers_js_1.formatTokens)(p.total_input_tokens)} | ${(0, helpers_js_1.formatTokens)(p.total_output_tokens)} |\n`;
+        }
+        return { content: [{ type: 'text', text: table }] };
+    });
+    // ============================================================
+    // 5. get_daily_trend — Daily cost trend
+    // ============================================================
+    server.registerTool('get_daily_trend', {
+        title: 'Daily Cost Trend',
+        description: 'Get daily cost trend over a specified number of days. Useful for spotting spending patterns.',
+        inputSchema: zod_1.z.object({
+            days: zod_1.z.number().int().min(1).max(365).optional().default(30).describe('Number of days to look back (default: 30)'),
+            provider: zod_1.z.string().optional().describe('Filter by provider'),
+            project: zod_1.z.string().optional().describe('Filter by project'),
+        }),
+        annotations: { readOnlyHint: true, destructiveHint: false, idempotentHint: true, openWorldHint: false },
+    }, async (args) => {
+        const data = db.getCostByDay({
+            days: args.days,
+            provider: args.provider,
+            project: args.project,
+        });
+        if (data.length === 0) {
+            return { content: [{ type: 'text', text: 'No usage data found for the specified period.' }] };
+        }
+        let table = `📈 Daily Cost Trend (Last ${args.days} days)\n\n`;
+        table += `| Date | Cost | Requests | Tokens |\n`;
+        table += `|------|------|----------|--------|\n`;
+        for (const d of data) {
+            table += `| ${d.date} | ${(0, helpers_js_1.formatCost)(d.total_cost)} | ${d.request_count} | ${(0, helpers_js_1.formatTokens)(d.total_tokens)} |\n`;
+        }
+        const totalCost = data.reduce((sum, d) => sum + d.total_cost, 0);
+        const avgDaily = totalCost / data.length;
+        table += `\nTotal: ${(0, helpers_js_1.formatCost)(totalCost)} | Avg/day: ${(0, helpers_js_1.formatCost)(avgDaily)}`;
+        return { content: [{ type: 'text', text: table }] };
+    });
+    // ============================================================
+    // 6. set_budget — Create or update a budget
+    // ============================================================
+    server.registerTool('set_budget', {
+        title: 'Set Budget',
+        description: 'Create or update a spending budget with alerts. Get notified when approaching or exceeding limits.',
+        inputSchema: zod_1.z.object({
+            name: zod_1.z.string().describe('Budget name (unique identifier)'),
+            limit: zod_1.z.number().positive().describe('Budget limit in USD'),
+            period: zod_1.z.enum(['daily', 'weekly', 'monthly', 'total']).optional().default('monthly').describe('Budget period'),
+            provider: zod_1.z.string().optional().default('').describe('Filter budget to specific provider'),
+            model: zod_1.z.string().optional().default('').describe('Filter budget to specific model'),
+            project: zod_1.z.string().optional().default('').describe('Filter budget to specific project'),
+        }),
+        annotations: { readOnlyHint: false, destructiveHint: false, idempotentHint: true, openWorldHint: false },
+    }, async (args) => {
+        const budget = db.setBudget({
+            name: args.name,
+            limit_amount: args.limit,
+            period: args.period || 'monthly',
+            provider_filter: args.provider || '',
+            model_filter: args.model || '',
+            project_filter: args.project || '',
+        });
+        const status = db.checkBudget(args.name);
+        return {
+            content: [{
+                    type: 'text',
+                    text: [
+                        `✅ Budget "${args.name}" set successfully`,
+                        ``,
+                        `Limit: ${(0, helpers_js_1.formatCost)(args.limit)} / ${args.period}`,
+                        args.provider ? `Provider filter: ${args.provider}` : null,
+                        args.model ? `Model filter: ${args.model}` : null,
+                        args.project ? `Project filter: ${args.project}` : null,
+                        ``,
+                        status ? `Current spend: ${(0, helpers_js_1.formatCost)(status.spent)} (${status.percentage.toFixed(1)}%)` : null,
+                        status ? `Remaining: ${(0, helpers_js_1.formatCost)(status.remaining)}` : null,
+                        status?.exceeded ? `⚠️ BUDGET EXCEEDED!` : null,
+                    ].filter(Boolean).join('\n'),
+                }],
+        };
+    });
+    // ============================================================
+    // 7. check_budget — Check budget status
+    // ============================================================
+    server.registerTool('check_budget', {
+        title: 'Check Budget',
+        description: 'Check the current status of all budgets or a specific budget. Shows spend vs limit.',
+        inputSchema: zod_1.z.object({
+            name: zod_1.z.string().optional().describe('Budget name to check (leave empty for all budgets)'),
+        }),
+        annotations: { readOnlyHint: true, destructiveHint: false, idempotentHint: true, openWorldHint: false },
+    }, async (args) => {
+        if (args.name) {
+            const status = db.checkBudget(args.name);
+            if (!status) {
+                return { content: [{ type: 'text', text: `Budget "${args.name}" not found.` }] };
+            }
+            const bar = generateProgressBar(status.percentage);
+            return {
+                content: [{
+                        type: 'text',
+                        text: [
+                            `💰 Budget: ${status.budget.name}`,
+                            `Period: ${status.budget.period}`,
+                            ``,
+                            `${bar} ${status.percentage.toFixed(1)}%`,
+                            `Spent: ${(0, helpers_js_1.formatCost)(status.spent)} / ${(0, helpers_js_1.formatCost)(status.budget.limit_amount)}`,
+                            `Remaining: ${(0, helpers_js_1.formatCost)(status.remaining)}`,
+                            ``,
+                            status.exceeded ? `🚨 BUDGET EXCEEDED!` : status.percentage >= 80 ? `⚠️ Warning: Approaching budget limit!` : `✅ Within budget`,
+                        ].join('\n'),
+                    }],
+            };
+        }
+        // Show all budgets
+        const budgets = db.getBudgets();
+        if (budgets.length === 0) {
+            return { content: [{ type: 'text', text: 'No budgets configured. Use set_budget to create one.' }] };
+        }
+        let result = `💰 Budget Status Overview\n\n`;
+        result += `| Budget | Period | Spent | Limit | % | Status |\n`;
+        result += `|--------|--------|-------|-------|---|--------|\n`;
+        for (const b of budgets) {
+            const status = db.checkBudget(b.name);
+            if (status) {
+                const statusIcon = status.exceeded ? '🚨' : status.percentage >= 80 ? '⚠️' : '✅';
+                result += `| ${b.name} | ${b.period} | ${(0, helpers_js_1.formatCost)(status.spent)} | ${(0, helpers_js_1.formatCost)(b.limit_amount)} | ${status.percentage.toFixed(0)}% | ${statusIcon} |\n`;
+            }
+        }
+        return { content: [{ type: 'text', text: result }] };
+    });
+    // ============================================================
+    // 8. generate_dashboard — Generate HTML dashboard
+    // ============================================================
+    server.registerTool('generate_dashboard', {
+        title: 'Generate Dashboard',
+        description: 'Generate a beautiful HTML dashboard with charts showing cost trends, breakdowns by model/provider/project, and top expensive requests.',
+        inputSchema: zod_1.z.object({
+            days: zod_1.z.number().int().min(1).max(365).optional().default(30).describe('Number of days to include in the dashboard'),
+            output_path: zod_1.z.string().optional().describe('Custom output path for the HTML file'),
+            format: zod_1.z.enum(['html', 'text']).optional().default('html').describe('Output format: html for visual dashboard, text for markdown report'),
+        }),
+        annotations: { readOnlyHint: true, destructiveHint: false, idempotentHint: false, openWorldHint: false },
+    }, async (args) => {
+        const data = (0, generator_js_1.collectDashboardData)(db, args.days);
+        if (args.format === 'text') {
+            const report = (0, generator_js_1.generateTextReport)(data);
+            return { content: [{ type: 'text', text: report }] };
+        }
+        const html = (0, generator_js_1.generateDashboardHTML)(data);
+        const outputDir = path_1.default.join(os_1.default.homedir(), '.mcp-cost-tracker');
+        if (!fs_1.default.existsSync(outputDir)) {
+            fs_1.default.mkdirSync(outputDir, { recursive: true });
+        }
+        const outputPath = args.output_path || path_1.default.join(outputDir, 'dashboard.html');
+        fs_1.default.writeFileSync(outputPath, html, 'utf-8');
+        return {
+            content: [{
+                    type: 'text',
+                    text: [
+                        `📊 Dashboard generated successfully!`,
+                        ``,
+                        `File: ${outputPath}`,
+                        `Period: Last ${args.days} days`,
+                        `Total records: ${data.summary.total_requests}`,
+                        `Total spend: ${(0, helpers_js_1.formatCost)(data.summary.total_cost)}`,
+                        ``,
+                        `Open the HTML file in a browser to view interactive charts.`,
+                    ].join('\n'),
+                }],
+        };
+    });
+    // ============================================================
+    // 9. lookup_pricing — Look up model pricing
+    // ============================================================
+    server.registerTool('lookup_pricing', {
+        title: 'Lookup Model Pricing',
+        description: 'Look up pricing for a specific model or list all supported models and their prices.',
+        inputSchema: zod_1.z.object({
+            model: zod_1.z.string().optional().describe('Model name to look up (leave empty to list all)'),
+            provider: zod_1.z.string().optional().describe('Provider name to filter by'),
+        }),
+        annotations: { readOnlyHint: true, destructiveHint: false, idempotentHint: true, openWorldHint: false },
+    }, async (args) => {
+        if (args.model) {
+            const pricing = (0, models_js_1.lookupPricing)(args.model, args.provider);
+            if (!pricing) {
+                return {
+                    content: [{
+                            type: 'text',
+                            text: `No pricing found for model "${args.model}". Use set_custom_pricing to add it, or check supported models with lookup_pricing (no arguments).`,
+                        }],
+                };
+            }
+            return {
+                content: [{
+                        type: 'text',
+                        text: [
+                            `💲 Pricing: ${pricing.provider} / ${pricing.model}`,
+                            ``,
+                            `| Metric | Price |`,
+                            `|--------|-------|`,
+                            `| Input | $${pricing.inputPricePerMTok.toFixed(4)} / MTok |`,
+                            `| Output | $${pricing.outputPricePerMTok.toFixed(4)} / MTok |`,
+                            pricing.cachedInputPricePerMTok ? `| Cached Input | $${pricing.cachedInputPricePerMTok.toFixed(4)} / MTok |` : null,
+                            pricing.contextWindow ? `| Context Window | ${(0, helpers_js_1.formatTokens)(pricing.contextWindow)} |` : null,
+                            `| Category | ${pricing.category} |`,
+                            pricing.notes ? `\nNote: ${pricing.notes}` : null,
+                        ].filter(Boolean).join('\n'),
+                    }],
+            };
+        }
+        // List all models
+        if (args.provider) {
+            const models = (0, models_js_1.getModelsForProvider)(args.provider);
+            if (models.length === 0) {
+                return {
+                    content: [{
+                            type: 'text',
+                            text: `No models found for provider "${args.provider}". Supported providers: ${(0, models_js_1.getSupportedProviders)().join(', ')}`,
+                        }],
+                };
+            }
+            let table = `💲 ${args.provider} Models\n\n`;
+            table += `| Model | Input $/MTok | Output $/MTok | Category |\n`;
+            table += `|-------|-------------|---------------|----------|\n`;
+            for (const m of models) {
+                table += `| ${m.model} | $${m.inputPricePerMTok.toFixed(4)} | $${m.outputPricePerMTok.toFixed(4)} | ${m.category} |\n`;
+            }
+            return { content: [{ type: 'text', text: table }] };
+        }
+        // List all providers
+        const providers = (0, models_js_1.getSupportedProviders)();
+        let result = `💲 Supported Providers & Models\n\n`;
+        result += `Providers: ${providers.join(', ')}\n`;
+        result += `Total models: ${(0, models_js_1.getAllModels)().length}\n\n`;
+        result += `Use \`lookup_pricing\` with a provider name to see all models for that provider.\n`;
+        result += `Use \`lookup_pricing\` with a model name to see detailed pricing.\n`;
+        return { content: [{ type: 'text', text: result }] };
+    });
+    // ============================================================
+    // 10. set_custom_pricing — Add custom model pricing
+    // ============================================================
+    server.registerTool('set_custom_pricing', {
+        title: 'Set Custom Pricing',
+        description: 'Add or update custom pricing for a model not in the built-in database, or override existing pricing.',
+        inputSchema: zod_1.z.object({
+            provider: zod_1.z.string().describe('Provider name'),
+            model: zod_1.z.string().describe('Model identifier'),
+            input_price: zod_1.z.number().min(0).describe('Input price per million tokens (USD)'),
+            output_price: zod_1.z.number().min(0).describe('Output price per million tokens (USD)'),
+            cached_input_price: zod_1.z.number().min(0).optional().default(0).describe('Cached input price per million tokens (USD)'),
+        }),
+        annotations: { readOnlyHint: false, destructiveHint: false, idempotentHint: true, openWorldHint: false },
+    }, async (args) => {
+        db.setCustomPricing({
+            provider: args.provider.toLowerCase(),
+            model: args.model,
+            input_price_per_mtok: args.input_price,
+            output_price_per_mtok: args.output_price,
+            cached_input_price_per_mtok: args.cached_input_price || 0,
+        });
+        return {
+            content: [{
+                    type: 'text',
+                    text: [
+                        `✅ Custom pricing set for ${args.provider}/${args.model}`,
+                        ``,
+                        `Input:  $${args.input_price.toFixed(4)} / MTok`,
+                        `Output: $${args.output_price.toFixed(4)} / MTok`,
+                        args.cached_input_price ? `Cached: $${args.cached_input_price.toFixed(4)} / MTok` : null,
+                    ].filter(Boolean).join('\n'),
+                }],
+        };
+    });
+    // ============================================================
+    // 11. estimate_cost — Estimate cost before making an API call
+    // ============================================================
+    server.registerTool('estimate_cost', {
+        title: 'Estimate Cost',
+        description: 'Estimate the cost of an API call before making it. Useful for pre-flight cost checks.',
+        inputSchema: zod_1.z.object({
+            model: zod_1.z.string().describe('Model identifier'),
+            input_tokens: zod_1.z.number().int().min(0).describe('Estimated input tokens'),
+            output_tokens: zod_1.z.number().int().min(0).describe('Estimated output tokens'),
+            provider: zod_1.z.string().optional().describe('Provider name'),
+            requests: zod_1.z.number().int().min(1).optional().default(1).describe('Number of requests to estimate'),
+        }),
+        annotations: { readOnlyHint: true, destructiveHint: false, idempotentHint: true, openWorldHint: false },
+    }, async (args) => {
+        const costResult = (0, models_js_1.calculateCost)(args.model, args.input_tokens, args.output_tokens, args.provider);
+        if (!costResult.pricing) {
+            return {
+                content: [{
+                        type: 'text',
+                        text: `No pricing found for model "${args.model}". Use lookup_pricing to see available models.`,
+                    }],
+            };
+        }
+        const totalForRequests = costResult.cost * (args.requests || 1);
+        return {
+            content: [{
+                    type: 'text',
+                    text: [
+                        `🔮 Cost Estimate: ${costResult.pricing.provider}/${costResult.pricing.model}`,
+                        ``,
+                        `Per request:`,
+                        `  Input (${(0, helpers_js_1.formatTokens)(args.input_tokens)} tokens):  ${(0, helpers_js_1.formatCost)(costResult.inputCost)}`,
+                        `  Output (${(0, helpers_js_1.formatTokens)(args.output_tokens)} tokens): ${(0, helpers_js_1.formatCost)(costResult.outputCost)}`,
+                        `  Total: ${(0, helpers_js_1.formatCost)(costResult.cost)}`,
+                        args.requests && args.requests > 1 ? `\nFor ${args.requests} requests: ${(0, helpers_js_1.formatCost)(totalForRequests)}` : null,
+                        ``,
+                        `Pricing: $${costResult.pricing.inputPricePerMTok}/$${costResult.pricing.outputPricePerMTok} per MTok`,
+                    ].filter(Boolean).join('\n'),
+                }],
+        };
+    });
+    // ============================================================
+    // 12. compare_models — Compare costs between models
+    // ============================================================
+    server.registerTool('compare_models', {
+        title: 'Compare Model Costs',
+        description: 'Compare the cost of the same workload across different models. Helps find the most cost-effective option.',
+        inputSchema: zod_1.z.object({
+            models: zod_1.z.array(zod_1.z.string()).min(2).max(10).describe('List of model names to compare'),
+            input_tokens: zod_1.z.number().int().min(0).describe('Input tokens for comparison'),
+            output_tokens: zod_1.z.number().int().min(0).describe('Output tokens for comparison'),
+        }),
+        annotations: { readOnlyHint: true, destructiveHint: false, idempotentHint: true, openWorldHint: false },
+    }, async (args) => {
+        const results = args.models.map(model => {
+            const cost = (0, models_js_1.calculateCost)(model, args.input_tokens, args.output_tokens);
+            return {
+                model,
+                provider: cost.pricing?.provider || 'unknown',
+                cost: cost.cost,
+                pricing: cost.pricing,
+            };
+        }).sort((a, b) => a.cost - b.cost);
+        let table = `⚖️ Model Cost Comparison\n`;
+        table += `Workload: ${(0, helpers_js_1.formatTokens)(args.input_tokens)} input + ${(0, helpers_js_1.formatTokens)(args.output_tokens)} output tokens\n\n`;
+        table += `| # | Provider | Model | Cost | vs Cheapest |\n`;
+        table += `|---|----------|-------|------|-------------|\n`;
+        const cheapest = results[0]?.cost || 0;
+        results.forEach((r, i) => {
+            const diff = cheapest > 0 ? ((r.cost - cheapest) / cheapest * 100).toFixed(0) : '0';
+            const marker = i === 0 ? '🏆' : '';
+            table += `| ${i + 1} | ${r.provider} | ${r.model} | ${(0, helpers_js_1.formatCost)(r.cost)} | ${i === 0 ? 'baseline' : `+${diff}%`} ${marker} |\n`;
+        });
+        if (results.length >= 2) {
+            const savings = results[results.length - 1].cost - results[0].cost;
+            table += `\n💡 Switching from ${results[results.length - 1].model} to ${results[0].model} saves ${(0, helpers_js_1.formatCost)(savings)} per request.`;
+        }
+        return { content: [{ type: 'text', text: table }] };
+    });
+    // ============================================================
+    // 13. get_recent_usage — Get recent usage records
+    // ============================================================
+    server.registerTool('get_recent_usage', {
+        title: 'Recent Usage',
+        description: 'Get the most recent API usage records with details.',
+        inputSchema: zod_1.z.object({
+            limit: zod_1.z.number().int().min(1).max(100).optional().default(10).describe('Number of records to return'),
+            provider: zod_1.z.string().optional().describe('Filter by provider'),
+            model: zod_1.z.string().optional().describe('Filter by model'),
+            project: zod_1.z.string().optional().describe('Filter by project'),
+        }),
+        annotations: { readOnlyHint: true, destructiveHint: false, idempotentHint: true, openWorldHint: false },
+    }, async (args) => {
+        const records = db.getUsageRecords({
+            provider: args.provider,
+            model: args.model,
+            project: args.project,
+            limit: args.limit,
+        });
+        if (records.length === 0) {
+            return { content: [{ type: 'text', text: 'No usage records found.' }] };
+        }
+        let table = `📋 Recent Usage (${records.length} records)\n\n`;
+        table += `| Time | Provider | Model | Tokens | Cost | Project |\n`;
+        table += `|------|----------|-------|--------|------|---------|\n`;
+        for (const r of records) {
+            table += `| ${r.timestamp} | ${r.provider} | ${r.model} | ${(0, helpers_js_1.formatTokens)(r.total_tokens)} | ${(0, helpers_js_1.formatCost)(r.total_cost)} | ${r.project} |\n`;
+        }
+        return { content: [{ type: 'text', text: table }] };
+    });
+    // ============================================================
+    // 14. export_data — Export usage data
+    // ============================================================
+    server.registerTool('export_data', {
+        title: 'Export Data',
+        description: 'Export usage data as CSV or JSON for external analysis.',
+        inputSchema: zod_1.z.object({
+            format: zod_1.z.enum(['csv', 'json']).optional().default('csv').describe('Export format'),
+            start_date: zod_1.z.string().optional().describe('Start date filter'),
+            end_date: zod_1.z.string().optional().describe('End date filter'),
+            output_path: zod_1.z.string().optional().describe('Custom output file path'),
+        }),
+        annotations: { readOnlyHint: true, destructiveHint: false, idempotentHint: false, openWorldHint: false },
+    }, async (args) => {
+        const records = db.getUsageRecords({
+            startDate: args.start_date,
+            endDate: args.end_date,
+        });
+        const outputDir = path_1.default.join(os_1.default.homedir(), '.mcp-cost-tracker');
+        if (!fs_1.default.existsSync(outputDir)) {
+            fs_1.default.mkdirSync(outputDir, { recursive: true });
+        }
+        let outputPath;
+        let content;
+        if (args.format === 'json') {
+            outputPath = args.output_path || path_1.default.join(outputDir, `export_${Date.now()}.json`);
+            content = JSON.stringify(records, null, 2);
+        }
+        else {
+            outputPath = args.output_path || path_1.default.join(outputDir, `export_${Date.now()}.csv`);
+            const headers = ['timestamp', 'provider', 'model', 'input_tokens', 'output_tokens', 'cached_input_tokens', 'total_tokens', 'input_cost', 'output_cost', 'total_cost', 'session_id', 'project', 'description'];
+            const rows = records.map(r => headers.map(h => {
+                const val = r[h];
+                return typeof val === 'string' && val.includes(',') ? `"${val}"` : val;
+            }).join(','));
+            content = [headers.join(','), ...rows].join('\n');
+        }
+        fs_1.default.writeFileSync(outputPath, content, 'utf-8');
+        return {
+            content: [{
+                    type: 'text',
+                    text: [
+                        `📁 Data exported successfully!`,
+                        ``,
+                        `File: ${outputPath}`,
+                        `Format: ${args.format?.toUpperCase()}`,
+                        `Records: ${records.length}`,
+                    ].join('\n'),
+                }],
+        };
+    });
+    // ============================================================
+    // 15. get_stats — Get database statistics
+    // ============================================================
+    server.registerTool('get_stats', {
+        title: 'Database Stats',
+        description: 'Get statistics about the cost tracking database.',
+        inputSchema: zod_1.z.object({}),
+        annotations: { readOnlyHint: true, destructiveHint: false, idempotentHint: true, openWorldHint: false },
+    }, async () => {
+        const stats = db.getStats();
+        const budgets = db.getBudgets();
+        const customPricing = db.getCustomPricing();
+        const providers = (0, models_js_1.getSupportedProviders)();
+        const formatBytes = (bytes) => {
+            if (bytes >= 1_000_000)
+                return `${(bytes / 1_000_000).toFixed(2)} MB`;
+            if (bytes >= 1_000)
+                return `${(bytes / 1_000).toFixed(1)} KB`;
+            return `${bytes} bytes`;
+        };
+        return {
+            content: [{
+                    type: 'text',
+                    text: [
+                        `📊 MCP Cost Tracker Stats`,
+                        ``,
+                        `| Metric | Value |`,
+                        `|--------|-------|`,
+                        `| Total Records | ${stats.total_records.toLocaleString()} |`,
+                        `| First Record | ${stats.first_record || 'N/A'} |`,
+                        `| Last Record | ${stats.last_record || 'N/A'} |`,
+                        `| Database Size | ${formatBytes(stats.db_size_bytes)} |`,
+                        `| Budgets Configured | ${budgets.length} |`,
+                        `| Custom Pricing Rules | ${customPricing.length} |`,
+                        `| Built-in Models | ${(0, models_js_1.getAllModels)().length} |`,
+                        `| Supported Providers | ${providers.join(', ')} |`,
+                    ].join('\n'),
+                }],
+        };
+    });
+}
+function generateProgressBar(percentage) {
+    const filled = Math.min(20, Math.round(percentage / 5));
+    const empty = 20 - filled;
+    const color = percentage >= 100 ? '🟥' : percentage >= 80 ? '🟧' : '🟩';
+    return `[${color.repeat(filled)}${'⬜'.repeat(empty)}]`;
+}
+//# sourceMappingURL=index.js.map