LLM Token Tracker

LLM Token Tracker 🧮

Token usage tracker for OpenAI, Claude, and Gemini APIs with MCP (Model Context Protocol) support. Pass accurate API costs to your users.

✨ Features

• 🎯 Simple Integration - One line to wrap your API client
• 📊 Automatic Tracking - No manual token counting
• 💰 Accurate Pricing - Up-to-date pricing for all models (2025)
• 🔄 Multiple Providers - OpenAI, Claude, and Gemini support
• 📈 User Management - Track usage per user/session
• 🌐 Currency Support - USD and KRW
• 🤖 MCP Server - Use directly in Claude Desktop!
• 🆕 Intuitive Session Tracking - Real-time usage with progress bars

📦 Installation

npm install llm-token-tracker

🚀 Quick Start

Option 1: Use as Library

const { TokenTracker } = require('llm-token-tracker');
// or import { TokenTracker } from 'llm-token-tracker';

// Initialize tracker
const tracker = new TokenTracker({
  currency: 'USD' // or 'KRW'
});

// Example: Manual tracking
const trackingId = tracker.startTracking('user-123');

// ... your API call here ...

tracker.endTracking(trackingId, {
  provider: 'openai', // or 'anthropic' or 'gemini'
  model: 'gpt-3.5-turbo',
  inputTokens: 100,
  outputTokens: 50,
  totalTokens: 150
});

// Get user's usage
const usage = tracker.getUserUsage('user-123');
console.log(`Total cost: $${usage.totalCost}`);

🔧 With Real APIs

To use with actual OpenAI/Anthropic APIs:

const OpenAI = require('openai');
const { TokenTracker } = require('llm-token-tracker');

const tracker = new TokenTracker();
const openai = tracker.wrap(new OpenAI({
  apiKey: process.env.OPENAI_API_KEY
}));

// Use normally - tracking happens automatically
const response = await openai.chat.completions.create({
  model: "gpt-3.5-turbo",
  messages: [{ role: "user", content: "Hello!" }]
});

console.log(response._tokenUsage);
// { tokens: 125, cost: 0.0002, model: "gpt-3.5-turbo" }

Option 2: Use as MCP Server

Add to Claude Desktop settings (~/Library/Application Support/Claude/claude_desktop_config.json):

{
  "mcpServers": {
    "token-tracker": {
      "command": "npx",
      "args": ["llm-token-tracker"]
    }
  }
}

Then in Claude:

• "Calculate current session usage" - See current session usage with intuitive format
• "Calculate current conversation cost" - Get cost breakdown with input/output tokens
• "Track my API usage"
• "Compare costs between GPT-4 and Claude"
• "Show my total spending today"

Available MCP Tools

1. get_current_session - 🆕 Get current session usage (RECOMMENDED)
   • Returns: Used/Remaining tokens, Input/Output breakdown, Cost, Progress bar
   • Default user_id: current-session
   • Default budget: 190,000 tokens
   • Perfect for real-time conversation tracking!
2. track_usage - Track token usage for an AI API call
   • Parameters: provider, model, input_tokens, output_tokens, user_id
3. get_usage - Get usage summary for specific user or all users
4. compare_costs - Compare costs between different models
5. clear_usage - Clear usage data for a user

Example MCP Output

💰 Current Session
━━━━━━━━━━━━━━━━━━━━━━
📊 Used: 62,830 tokens (33.1%)
✨ Remaining: 127,170 tokens
[██████░░░░░░░░░░░░░░]

📥 Input: 55,000 tokens
📤 Output: 7,830 tokens
💵 Cost: $0.2825
━━━━━━━━━━━━━━━━━━━━━━

📋 Model Breakdown:
  • anthropic/claude-sonnet-4.5: 62,830 tokens ($0.2825)

📊 Supported Models & Pricing (Updated 2025)

OpenAI (2025)

Anthropic (2025)

Google Gemini (2025)

Note: Prices shown are per 1,000 tokens. Batch API offers 50% discount. Prompt caching can reduce costs by up to 90%.

🎯 Examples

Run the example:

npm run example

Check examples/basic-usage.js for detailed usage patterns.

📝 API Reference

new TokenTracker(config)

• config.currency: 'USD' or 'KRW' (default: 'USD')
• config.webhookUrl: Optional webhook for usage notifications

tracker.wrap(client)

Wrap an OpenAI or Anthropic client for automatic tracking.

tracker.forUser(userId)

Create a user-specific tracker instance.

tracker.startTracking(userId?, sessionId?)

Start manual tracking session. Returns tracking ID.

tracker.endTracking(trackingId, usage)

End tracking and record usage.

tracker.getUserUsage(userId)

Get total usage for a user.

tracker.getAllUsersUsage()

Get usage summary for all users.

🛠 Development

# Install dependencies
npm install

# Build TypeScript
npm run build

# Watch mode
npm run dev

# Run examples
npm run example

📄 License

MIT

🤝 Contributing

Contributions are welcome! Please feel free to submit a Pull Request.

🐛 Issues

For bugs and feature requests, please create an issue.

📦 What's New in v2.4.0

• 🎉 Gemini API Support - Full integration with Google's Gemini models
• 💎 Gemini 2.0 Support - Free preview models included
• 📊 Enhanced Pricing - Up-to-date Gemini 1.5 and 2.0 pricing
• 🔧 Auto-detection - Automatic Gemini client wrapping
• 💰 Cost Comparison - Compare Gemini with OpenAI and Claude

📦 What's New in v2.3.0

• 💱 Real-time exchange rates - Automatic USD to KRW conversion
• 🌐 Uses exchangerate-api.com for accurate rates
• 💾 24-hour caching to minimize API calls
• 📊 New get_exchange_rate tool to check current rates
• 🔄 Background auto-updates with fallback support

What's New in v2.2.0

• 🗄️ File-based persistence - Session data survives server restarts
• 💾 Automatic saving to ~/.llm-token-tracker/sessions.json
• 🔄 Works for both npm and local installations
• 📊 Historical data tracking across sessions
• 🎯 Zero configuration required - just works!

What's New in v2.1.0

• 🆕 Added get_current_session tool for intuitive session tracking
• 📊 Real-time progress bars and visual indicators
• 💰 Enhanced cost breakdown with input/output token separation
• 🎨 Improved formatting with thousands separators
• 🔧 Better default user_id handling (current-session)

Built with ❤️ for developers who need transparent AI API billing.