mcp-cost-tracker 1.0.0

package/src/tools/prompts.ts

@@ -0,0 +1,124 @@
+import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
+import { z } from 'zod';
+
+export function registerPrompts(server: McpServer): void {
+
+  // ============================================================
+  // Prompt: Cost analysis
+  // ============================================================
+  server.registerPrompt(
+    'analyze-costs',
+    {
+      title: 'Analyze LLM Costs',
+      description: 'Analyze your LLM API costs and provide optimization recommendations',
+      argsSchema: {
+        focus: z.string().optional().describe('Specific area to focus on (e.g., "reduce costs", "compare models", "budget planning")'),
+      },
+    },
+    ({ focus }) => ({
+      messages: [
+        {
+          role: 'user' as const,
+          content: {
+            type: 'text' as const,
+            text: [
+              `Please analyze my LLM API costs using the mcp-cost-tracker tools.`,
+              ``,
+              `Steps:`,
+              `1. First, use get_summary to get the overall cost summary`,
+              `2. Use get_cost_by_model to see which models cost the most`,
+              `3. Use get_cost_by_provider to see provider-level breakdown`,
+              `4. Use get_daily_trend to identify spending patterns`,
+              `5. Check check_budget for any budget alerts`,
+              ``,
+              focus ? `Focus area: ${focus}` : '',
+              ``,
+              `Please provide:`,
+              `- A clear summary of total spending`,
+              `- The top cost drivers (models and providers)`,
+              `- Spending trends over time`,
+              `- Specific recommendations to reduce costs`,
+              `- Suggestions for more cost-effective model alternatives`,
+            ].filter(Boolean).join('\n'),
+          },
+        },
+      ],
+    })
+  );
+
+  // ============================================================
+  // Prompt: Cost comparison
+  // ============================================================
+  server.registerPrompt(
+    'compare-options',
+    {
+      title: 'Compare Model Options',
+      description: 'Compare different models for a specific use case to find the most cost-effective option',
+      argsSchema: {
+        use_case: z.string().describe('Description of the use case (e.g., "code generation", "summarization", "chat")'),
+        input_tokens: z.string().optional().describe('Estimated input tokens per request'),
+        output_tokens: z.string().optional().describe('Estimated output tokens per request'),
+      },
+    },
+    ({ use_case, input_tokens, output_tokens }) => ({
+      messages: [
+        {
+          role: 'user' as const,
+          content: {
+            type: 'text' as const,
+            text: [
+              `I need help choosing the most cost-effective LLM for: ${use_case}`,
+              ``,
+              input_tokens ? `Estimated input tokens per request: ${input_tokens}` : '',
+              output_tokens ? `Estimated output tokens per request: ${output_tokens}` : '',
+              ``,
+              `Please:`,
+              `1. Use lookup_pricing to find relevant models`,
+              `2. Use compare_models to compare costs`,
+              `3. Use estimate_cost for each option`,
+              `4. Consider quality vs cost tradeoffs`,
+              `5. Recommend the best option with reasoning`,
+            ].filter(Boolean).join('\n'),
+          },
+        },
+      ],
+    })
+  );
+
+  // ============================================================
+  // Prompt: Budget setup
+  // ============================================================
+  server.registerPrompt(
+    'setup-budgets',
+    {
+      title: 'Setup Cost Budgets',
+      description: 'Help set up appropriate budgets based on current usage patterns',
+      argsSchema: {
+        monthly_target: z.string().optional().describe('Target monthly budget in USD'),
+      },
+    },
+    ({ monthly_target }) => ({
+      messages: [
+        {
+          role: 'user' as const,
+          content: {
+            type: 'text' as const,
+            text: [
+              `Help me set up cost budgets for my LLM API usage.`,
+              ``,
+              monthly_target ? `My target monthly budget is: $${monthly_target}` : '',
+              ``,
+              `Please:`,
+              `1. Use get_summary to understand current spending`,
+              `2. Use get_cost_by_provider and get_cost_by_model for breakdown`,
+              `3. Use get_daily_trend to understand patterns`,
+              `4. Recommend appropriate budget limits`,
+              `5. Use set_budget to create the budgets`,
+              `6. Set up both overall and per-provider budgets if appropriate`,
+            ].filter(Boolean).join('\n'),
+          },
+        },
+      ],
+    })
+  );
+}

package/src/tools/resources.ts

@@ -0,0 +1,171 @@
+import { McpServer, ResourceTemplate } from '@modelcontextprotocol/sdk/server/mcp.js';
+import { CostDatabase } from '../storage/database.js';
+import { collectDashboardData, generateTextReport } from '../dashboard/generator.js';
+import { getAllModels, getSupportedProviders } from '../pricing/models.js';
+
+export function registerResources(server: McpServer, db: CostDatabase): void {
+
+  // ============================================================
+  // Static Resource: Current cost summary
+  // ============================================================
+  server.registerResource(
+    'cost-summary',
+    'cost://summary',
+    {
+      title: 'Cost Summary',
+      description: 'Current cost summary across all providers and models',
+      mimeType: 'application/json',
+    },
+    async (uri) => {
+      const summary = db.getSummary();
+      const byProvider = db.getCostByProvider();
+      const byModel = db.getCostByModel();
+
+      return {
+        contents: [{
+          uri: uri.href,
+          text: JSON.stringify({
+            summary,
+            by_provider: byProvider,
+            by_model: byModel,
+            generated_at: new Date().toISOString(),
+          }, null, 2),
+        }],
+      };
+    }
+  );
+
+  // ============================================================
+  // Static Resource: Supported models pricing
+  // ============================================================
+  server.registerResource(
+    'pricing-catalog',
+    'cost://pricing',
+    {
+      title: 'Pricing Catalog',
+      description: 'Complete pricing catalog for all supported LLM models',
+      mimeType: 'application/json',
+    },
+    async (uri) => {
+      const models = getAllModels();
+      const providers = getSupportedProviders();
+
+      return {
+        contents: [{
+          uri: uri.href,
+          text: JSON.stringify({
+            providers,
+            total_models: models.length,
+            models: models.map(m => ({
+              provider: m.provider,
+              model: m.model,
+              input_price_per_mtok: m.inputPricePerMTok,
+              output_price_per_mtok: m.outputPricePerMTok,
+              cached_input_price_per_mtok: m.cachedInputPricePerMTok,
+              context_window: m.contextWindow,
+              category: m.category,
+            })),
+          }, null, 2),
+        }],
+      };
+    }
+  );
+
+  // ============================================================
+  // Static Resource: Budget status
+  // ============================================================
+  server.registerResource(
+    'budget-status',
+    'cost://budgets',
+    {
+      title: 'Budget Status',
+      description: 'Current status of all configured budgets',
+      mimeType: 'application/json',
+    },
+    async (uri) => {
+      const budgets = db.getBudgets();
+      const statuses = budgets.map(b => {
+        const status = db.checkBudget(b.name);
+        return {
+          name: b.name,
+          period: b.period,
+          limit: b.limit_amount,
+          spent: status?.spent || 0,
+          remaining: status?.remaining || b.limit_amount,
+          percentage: status?.percentage || 0,
+          exceeded: status?.exceeded || false,
+        };
+      });
+
+      return {
+        contents: [{
+          uri: uri.href,
+          text: JSON.stringify({ budgets: statuses }, null, 2),
+        }],
+      };
+    }
+  );
+
+  // ============================================================
+  // Static Resource: Full text report
+  // ============================================================
+  server.registerResource(
+    'full-report',
+    'cost://report',
+    {
+      title: 'Full Cost Report',
+      description: 'Complete cost report in Markdown format',
+      mimeType: 'text/markdown',
+    },
+    async (uri) => {
+      const data = collectDashboardData(db, 30);
+      const report = generateTextReport(data);
+
+      return {
+        contents: [{
+          uri: uri.href,
+          text: report,
+        }],
+      };
+    }
+  );
+
+  // ============================================================
+  // Dynamic Resource: Provider-specific data
+  // ============================================================
+  server.registerResource(
+    'provider-cost',
+    new ResourceTemplate('cost://provider/{provider}', {
+      list: async () => {
+        const providers = getSupportedProviders();
+        return {
+          resources: providers.map(p => ({
+            uri: `cost://provider/${p}`,
+            name: `${p} costs`,
+          })),
+        };
+      },
+    }),
+    {
+      title: 'Provider Cost Data',
+      description: 'Cost data for a specific provider',
+      mimeType: 'application/json',
+    },
+    async (uri, { provider }) => {
+      const providerStr = Array.isArray(provider) ? provider[0] : provider;
+      const records = db.getUsageRecords({ provider: providerStr, limit: 100 });
+      const summary = db.getSummary({ provider: providerStr });
+
+      return {
+        contents: [{
+          uri: uri.href,
+          text: JSON.stringify({
+            provider: providerStr,
+            summary,
+            recent_records: records.slice(0, 20),
+          }, null, 2),
+        }],
+      };
+    }
+  );
+}

package/src/utils/helpers.ts

@@ -0,0 +1,71 @@
+/**
+ * Format a cost value as USD string.
+ */
+export function formatCost(cost: number): string {
+  if (cost >= 1) return `$${cost.toFixed(2)}`;
+  if (cost >= 0.01) return `$${cost.toFixed(4)}`;
+  return `$${cost.toFixed(6)}`;
+}
+
+/**
+ * Format token count with K/M suffix.
+ */
+export function formatTokens(tokens: number): string {
+  if (tokens >= 1_000_000) return `${(tokens / 1_000_000).toFixed(2)}M`;
+  if (tokens >= 1_000) return `${(tokens / 1_000).toFixed(1)}K`;
+  return tokens.toString();
+}
+
+/**
+ * Get ISO date string for start of current period.
+ */
+export function getPeriodStart(period: 'daily' | 'weekly' | 'monthly' | 'yearly'): string {
+  const now = new Date();
+  switch (period) {
+    case 'daily':
+      return new Date(now.getFullYear(), now.getMonth(), now.getDate()).toISOString();
+    case 'weekly': {
+      const day = now.getDay();
+      const diff = now.getDate() - day + (day === 0 ? -6 : 1);
+      return new Date(now.getFullYear(), now.getMonth(), diff).toISOString();
+    }
+    case 'monthly':
+      return new Date(now.getFullYear(), now.getMonth(), 1).toISOString();
+    case 'yearly':
+      return new Date(now.getFullYear(), 0, 1).toISOString();
+  }
+}
+
+/**
+ * Validate and parse a date string.
+ */
+export function parseDate(dateStr: string): Date | null {
+  const date = new Date(dateStr);
+  return isNaN(date.getTime()) ? null : date;
+}
+
+/**
+ * Generate a simple unique ID.
+ */
+export function generateId(): string {
+  return `${Date.now()}-${Math.random().toString(36).substring(2, 9)}`;
+}
+
+/**
+ * Safely parse JSON with a fallback.
+ */
+export function safeJsonParse<T>(json: string, fallback: T): T {
+  try {
+    return JSON.parse(json);
+  } catch {
+    return fallback;
+  }
+}
+
+/**
+ * Truncate a string to a maximum length.
+ */
+export function truncate(str: string, maxLen: number = 100): string {
+  if (str.length <= maxLen) return str;
+  return str.substring(0, maxLen - 3) + '...';
+}

package/tsconfig.json

@@ -0,0 +1,20 @@
+{
+  "compilerOptions": {
+    "target": "ES2022",
+    "module": "Node16",
+    "moduleResolution": "Node16",
+    "lib": ["ES2022"],
+    "outDir": "./dist",
+    "rootDir": "./src",
+    "strict": true,
+    "esModuleInterop": true,
+    "skipLibCheck": true,
+    "forceConsistentCasingInFileNames": true,
+    "resolveJsonModule": true,
+    "declaration": true,
+    "declarationMap": true,
+    "sourceMap": true
+  },
+  "include": ["src/**/*"],
+  "exclude": ["node_modules", "dist"]
+}