openrouter-pricing-mcp 1.0.2 → 1.2.0

package/build/index.js

@@ -29,7 +29,6 @@ async function fetchModelsFromAPI() {
     const headers = {
         "Content-Type": "application/json",
     };
-    // Use API key if provided (optional — the models endpoint is public)
     const apiKey = process.env.OPENROUTER_API_KEY;
     if (apiKey) {
         headers["Authorization"] = `Bearer ${apiKey}`;
@@ -44,20 +43,107 @@ async function fetchModelsFromAPI() {
     return data.data;
 }
 // ─── Helpers ─────────────────────────────────────────────────────────────────
-export function formatCost(costStr) {
+const currencyFormat = new Intl.NumberFormat("en-US", {
+    style: "currency",
+    currency: "USD",
+    minimumFractionDigits: 2,
+    maximumFractionDigits: 4,
+});
+export function formatCostPer1M(costStr) {
     const cost = parseFloat(costStr);
     if (isNaN(cost))
         return costStr;
     if (cost === 0)
         return "FREE";
-    return `$${cost.toFixed(8)}`;
+    return currencyFormat.format(cost * 1_000_000);
+}
+export function formatCostUnit(costStr) {
+    const cost = parseFloat(costStr);
+    if (isNaN(cost))
+        return costStr;
+    if (cost === 0)
+        return "FREE";
+    return currencyFormat.format(cost);
 }
 // ─── Tool Handlers ───────────────────────────────────────────────────────────
+export function handleGetModelCapabilities(models, args) {
+    const modelId = String(args.model_id ?? "");
+    const model = models.find((m) => m.id === modelId);
+    if (!model) {
+        const fuzzy = models.filter((m) => m.id.toLowerCase().includes(modelId.toLowerCase()));
+        if (fuzzy.length > 0) {
+            const suggestions = fuzzy.slice(0, 5).map((m) => m.id).join("\n  ");
+            return {
+                content: [
+                    {
+                        type: "text",
+                        text: `Model "${modelId}" not found. Did you mean one of these?\n  ${suggestions}`,
+                    },
+                ],
+                isError: true,
+            };
+        }
+        return {
+            content: [{ type: "text", text: `Model "${modelId}" not found.` }],
+            isError: true,
+        };
+    }
+    const inputModalities = model.architecture?.input_modalities ?? [];
+    const outputModalities = model.architecture?.output_modalities ?? [];
+    const supportedParams = model.supported_parameters ?? [];
+    const maxCompletionTokens = model.top_provider?.max_completion_tokens;
+    const isModerated = model.top_provider?.is_moderated;
+    const tokenizer = model.architecture?.tokenizer;
+    const knowledgeCutoff = model.knowledge_cutoff;
+    // Derive capability flags from supported_parameters
+    const hasToolUse = supportedParams.includes("tools") || supportedParams.includes("tool_choice");
+    const hasReasoning = supportedParams.includes("reasoning") || supportedParams.includes("include_reasoning");
+    const hasStructuredOutput = supportedParams.includes("structured_outputs") || supportedParams.includes("response_format");
+    const hasVision = inputModalities.includes("image");
+    const hasAudioInput = inputModalities.includes("audio");
+    const hasAudioOutput = outputModalities.includes("audio");
+    const hasImageOutput = outputModalities.includes("image");
+    const lines = [
+        `Model: ${model.name} (${model.id})`,
+    ];
+    if (model.description) {
+        lines.push(``, `── Description ──`, model.description);
+    }
+    lines.push(``, `── Pricing ──`, `Prompt: ${formatCostPer1M(model.pricing.prompt)} / 1M tokens`, `Completion: ${formatCostPer1M(model.pricing.completion)} / 1M tokens`, model.pricing.image ? `Image: ${formatCostUnit(model.pricing.image)} / image` : "", model.pricing.request ? `Request: ${formatCostUnit(model.pricing.request)} / request` : "", ``, `── Context & Limits ──`, `Context Length: ${model.context_length?.toLocaleString() ?? "N/A"} tokens`);
+    if (maxCompletionTokens != null) {
+        lines.push(`Max Completion Tokens: ${maxCompletionTokens.toLocaleString()}`);
+    }
+    lines.push(``, `── Modalities ──`, `Input: ${inputModalities.length > 0 ? inputModalities.join(", ") : "text"}`, `Output: ${outputModalities.length > 0 ? outputModalities.join(", ") : "text"}`, ``, `── Capabilities ──`, `🔧 Tool Use (Function Calling): ${hasToolUse ? "✅ Yes" : "❌ No"}`, `🧠 Reasoning / Thinking: ${hasReasoning ? "✅ Yes" : "❌ No"}`, `📋 Structured Output (JSON): ${hasStructuredOutput ? "✅ Yes" : "❌ No"}`, `👁️ Vision (Image Input): ${hasVision ? "✅ Yes" : "❌ No"}`, `🎤 Audio Input: ${hasAudioInput ? "✅ Yes" : "❌ No"}`, `🔊 Audio Output: ${hasAudioOutput ? "✅ Yes" : "❌ No"}`, `🎨 Image Generation: ${hasImageOutput ? "✅ Yes" : "❌ No"}`);
+    if (tokenizer) {
+        lines.push(``, `── Technical Details ──`);
+        lines.push(`Tokenizer: ${tokenizer}`);
+    }
+    if (isModerated != null) {
+        if (!tokenizer)
+            lines.push(``, `── Technical Details ──`);
+        lines.push(`Content Moderated: ${isModerated ? "Yes" : "No"}`);
+    }
+    if (knowledgeCutoff) {
+        lines.push(`Knowledge Cutoff: ${knowledgeCutoff}`);
+    }
+    if (supportedParams.length > 0) {
+        lines.push(``, `── Supported Parameters ──`, supportedParams.join(", "));
+    }
+    // Filter out empty strings from conditional pushes
+    const filteredLines = lines.filter((l, i) => !(l === "" && i > 0 && lines[i - 1] === ""));
+    return {
+        content: [
+            {
+                type: "text",
+                text: filteredLines.join("\n"),
+            },
+        ],
+    };
+}
 export function handleGetModelPricing(models, args) {
     const modelId = String(args.model_id ?? "");
     const model = models.find((m) => m.id === modelId);
     if (!model) {
-        // Attempt fuzzy match
         const fuzzy = models.filter((m) => m.id.toLowerCase().includes(modelId.toLowerCase()));
         if (fuzzy.length > 0) {
             const suggestions = fuzzy.slice(0, 5).map((m) => m.id).join("\n  ");
@@ -76,18 +162,21 @@ export function handleGetModelPricing(models, args) {
             isError: true,
         };
     }
+    const pricingLines = [
+        `Model: ${model.name} (${model.id})`,
+    ];
+    if (model.description) {
+        pricingLines.push(``, model.description);
+    }
+    pricingLines.push(``, `Context Length: ${model.context_length?.toLocaleString() ?? "N/A"} tokens`, `Prompt Cost: ${formatCostPer1M(model.pricing.prompt)} / 1M tokens`, `Completion Cost: ${formatCostPer1M(model.pricing.completion)} / 1M tokens`, `Image Cost: ${formatCostUnit(model.pricing.image ?? "0")} / image`, `Request Cost: ${formatCostUnit(model.pricing.request ?? "0")} / request`);
+    if (model.top_provider?.max_completion_tokens) {
+        pricingLines.push(`Max Completion Tokens: ${model.top_provider.max_completion_tokens.toLocaleString()}`);
+    }
     return {
         content: [
             {
                 type: "text",
-                text: [
-                    `Model: ${model.name} (${model.id})`,
-                    `Context Length: ${model.context_length?.toLocaleString() ?? "N/A"} tokens`,
-                    `Prompt Cost: ${formatCost(model.pricing.prompt)} / token`,
-                    `Completion Cost: ${formatCost(model.pricing.completion)} / token`,
-                    `Image Cost: ${formatCost(model.pricing.image)} / token`,
-                    `Request Cost: ${formatCost(model.pricing.request)}`,
-                ].join("\n"),
+                text: pricingLines.join("\n"),
             },
         ],
     };
@@ -96,7 +185,7 @@ export function handleListAllModelsPricing(models, args) {
     const limit = Math.min(Math.max(Number(args.limit) || 50, 1), 200);
     const results = models.slice(0, limit);
     const formatted = results
-        .map((m, i) => `${i + 1}. ${m.id} — Prompt: ${formatCost(m.pricing.prompt)}, Completion: ${formatCost(m.pricing.completion)}`)
+        .map((m, i) => `${i + 1}. ${m.id} — Prompt: ${formatCostPer1M(m.pricing.prompt)}/1M, Completion: ${formatCostPer1M(m.pricing.completion)}/1M`)
         .join("\n");
     return {
         content: [
@@ -130,9 +219,23 @@ export function handleCompareModelCosts(models, args) {
             isError: true,
         };
     }
-    const header = `| Model | Prompt Cost | Completion Cost | Context Length |`;
-    const separator = `|-------|------------|----------------|----------------|`;
-    const rows = found.map((m) => `| ${m.id} | ${formatCost(m.pricing.prompt)} | ${formatCost(m.pricing.completion)} | ${m.context_length?.toLocaleString() ?? "N/A"} |`);
+    // Helper to derive capability flags
+    function getCapFlags(m) {
+        const params = m.supported_parameters ?? [];
+        const inp = m.architecture?.input_modalities ?? [];
+        return {
+            tools: params.includes("tools") || params.includes("tool_choice") ? "✅" : "❌",
+            reasoning: params.includes("reasoning") || params.includes("include_reasoning") ? "✅" : "❌",
+            vision: inp.includes("image") ? "✅" : "❌",
+            json: params.includes("structured_outputs") || params.includes("response_format") ? "✅" : "❌",
+        };
+    }
+    const header = `| Model | Prompt/1M | Completion/1M | Context | Tools | Reasoning | Vision | JSON |`;
+    const separator = `|-------|----------|--------------|---------|-------|-----------|--------|------|`;
+    const rows = found.map((m) => {
+        const caps = getCapFlags(m);
+        return `| ${m.id} | ${formatCostPer1M(m.pricing.prompt)} | ${formatCostPer1M(m.pricing.completion)} | ${m.context_length?.toLocaleString() ?? "N/A"} | ${caps.tools} | ${caps.reasoning} | ${caps.vision} | ${caps.json} |`;
+    });
     let text = `${header}\n${separator}\n${rows.join("\n")}`;
     if (notFound.length > 0) {
         text += `\n\n⚠️ Models not found: ${notFound.join(", ")}`;
@@ -152,13 +255,13 @@ export function handleGetCheapestModels(models, args) {
         .sort((a, b) => parseFloat(a.pricing[metric]) - parseFloat(b.pricing[metric]));
     const results = sorted.slice(0, limit);
     const formatted = results
-        .map((m, i) => `${i + 1}. ${m.id} — ${metric}: ${formatCost(m.pricing[metric])}${parseFloat(m.pricing[metric]) === 0 ? " 🆓" : ""}`)
+        .map((m, i) => `${i + 1}. ${m.id} — ${metric}: ${formatCostPer1M(m.pricing[metric])}/1M${parseFloat(m.pricing[metric]) === 0 ? " 🆓" : ""}`)
         .join("\n");
     return {
         content: [
             {
                 type: "text",
-                text: `Top ${results.length} cheapest models by ${metric} cost:\n\n${formatted}`,
+                text: `Top ${results.length} cheapest models by ${metric} cost (per 1M tokens):\n\n${formatted}`,
             },
         ],
     };
@@ -177,7 +280,7 @@ export function handleFindModelsByContextLength(models, args) {
         .sort((a, b) => b.context_length - a.context_length);
     const results = matching.slice(0, limit);
     const formatted = results
-        .map((m, i) => `${i + 1}. ${m.id} — Context: ${m.context_length.toLocaleString()} tokens, Prompt: ${formatCost(m.pricing.prompt)}`)
+        .map((m, i) => `${i + 1}. ${m.id} — Context: ${m.context_length.toLocaleString()} tokens, Prompt: ${formatCostPer1M(m.pricing.prompt)}/1M`)
         .join("\n");
     return {
         content: [
@@ -192,7 +295,7 @@ export function handleFindModelsByContextLength(models, args) {
 export const TOOL_DEFINITIONS = [
     {
         name: "get_model_pricing",
-        description: "Get pricing details for a specific OpenRouter model by its full ID (e.g. google/gemini-2.5-pro-preview). Includes prompt, completion, image, and request costs.",
+        description: "Get pricing details and description for a specific OpenRouter model by its full ID (e.g. google/gemini-2.5-pro-preview). Includes prompt, completion, image, and request costs formatted per 1M tokens, plus model description and max output length.",
         inputSchema: {
             type: "object",
             properties: {
@@ -220,7 +323,7 @@ export const TOOL_DEFINITIONS = [
     },
     {
         name: "compare_model_costs",
-        description: "Compare costs between multiple OpenRouter models side-by-side in a table format.",
+        description: "Compare costs and capabilities between multiple OpenRouter models side-by-side in a table format. Shows pricing, context length, and key capability flags (tools, reasoning, vision, JSON output).",
         inputSchema: {
             type: "object",
             properties: {
@@ -270,6 +373,20 @@ export const TOOL_DEFINITIONS = [
             required: ["min_context_length"],
         },
     },
+    {
+        name: "get_model_capabilities",
+        description: "Get a comprehensive model card for a specific OpenRouter model. Includes description, pricing, context limits, input/output modalities, capability flags (tool use, reasoning, vision, structured output, audio), technical details, and all supported parameters.",
+        inputSchema: {
+            type: "object",
+            properties: {
+                model_id: {
+                    type: "string",
+                    description: "The full ID of the model (e.g. openai/gpt-4o, anthropic/claude-sonnet-4)",
+                },
+            },
+            required: ["model_id"],
+        },
+    },
 ];
 // ─── Server Setup ────────────────────────────────────────────────────────────
 const HANDLER_MAP = {
@@ -278,9 +395,10 @@ const HANDLER_MAP = {
     compare_model_costs: handleCompareModelCosts,
     get_cheapest_models: handleGetCheapestModels,
     find_models_by_context_length: handleFindModelsByContextLength,
+    get_model_capabilities: handleGetModelCapabilities,
 };
 export function createServer() {
-    const server = new Server({ name: "openrouter-pricing-mcp", version: "1.0.0" }, { capabilities: { tools: {} } });
+    const server = new Server({ name: "openrouter-pricing-mcp", version: "1.2.0" }, { capabilities: { tools: {} } });
     server.setRequestHandler(ListToolsRequestSchema, async () => ({
         tools: TOOL_DEFINITIONS,
     }));

package/build/index.test.js

@@ -1,19 +1,45 @@
 import { describe, it } from "node:test";
 import assert from "node:assert/strict";
-import { formatCost, handleGetModelPricing, handleListAllModelsPricing, handleCompareModelCosts, handleGetCheapestModels, handleFindModelsByContextLength, TOOL_DEFINITIONS, } from "./index.js";
+import { formatCostPer1M, formatCostUnit, handleGetModelPricing, handleListAllModelsPricing, handleCompareModelCosts, handleGetCheapestModels, handleFindModelsByContextLength, handleGetModelCapabilities, TOOL_DEFINITIONS, } from "./index.js";
 // ─── Fixture Data ────────────────────────────────────────────────────────────
 const MOCK_MODELS = [
     {
         id: "openai/gpt-4o",
         name: "GPT-4o",
+        description: "GPT-4o is OpenAI's versatile flagship model with vision capabilities and strong reasoning.",
         pricing: { prompt: "0.0000025", completion: "0.00001", request: "0", image: "0.003613" },
         context_length: 128000,
+        architecture: {
+            modality: "text+image->text",
+            input_modalities: ["text", "image"],
+            output_modalities: ["text"],
+            tokenizer: "GPT",
+        },
+        top_provider: {
+            context_length: 128000,
+            max_completion_tokens: 16384,
+            is_moderated: true,
+        },
+        supported_parameters: ["max_tokens", "temperature", "tools", "tool_choice", "response_format", "structured_outputs"],
     },
     {
         id: "anthropic/claude-sonnet-4",
         name: "Claude Sonnet 4",
+        description: "Claude Sonnet 4 excels in coding and reasoning with improved precision and controllability.",
         pricing: { prompt: "0.000003", completion: "0.000015", request: "0", image: "0.0048" },
         context_length: 200000,
+        architecture: {
+            modality: "text+image->text",
+            input_modalities: ["text", "image"],
+            output_modalities: ["text"],
+            tokenizer: "Claude",
+        },
+        top_provider: {
+            context_length: 200000,
+            max_completion_tokens: 128000,
+            is_moderated: true,
+        },
+        supported_parameters: ["include_reasoning", "max_tokens", "reasoning", "response_format", "tools", "tool_choice", "structured_outputs"],
     },
     {
         id: "google/gemini-2.5-pro-preview",
@@ -35,18 +61,25 @@ const MOCK_MODELS = [
     },
 ];
 // ─── formatCost ──────────────────────────────────────────────────────────────
-describe("formatCost", () => {
-    it("formats a normal cost", () => {
-        assert.equal(formatCost("0.000003"), "$0.00000300");
+describe("formatCostPer1M", () => {
+    it("formats a normal cost per 1M tokens properly", () => {
+        // 0.0000025 * 1M = 2.50
+        assert.equal(formatCostPer1M("0.0000025"), "$2.50");
     });
     it("returns FREE for zero-cost", () => {
-        assert.equal(formatCost("0"), "FREE");
+        assert.equal(formatCostPer1M("0"), "FREE");
     });
     it("handles NaN gracefully by returning original string", () => {
-        assert.equal(formatCost("N/A"), "N/A");
+        assert.equal(formatCostPer1M("N/A"), "N/A");
     });
     it("handles very small costs accurately", () => {
-        assert.equal(formatCost("0.0000001"), "$0.00000010");
+        // 0.0000001 * 1M = 0.10
+        assert.equal(formatCostPer1M("0.0000001"), "$0.10");
+    });
+});
+describe("formatCostUnit", () => {
+    it("formats a per-unit cost nicely", () => {
+        assert.equal(formatCostUnit("0.003613"), "$0.0036");
     });
 });
 // ─── get_model_pricing ──────────────────────────────────────────────────────
@@ -58,6 +91,9 @@ describe("handleGetModelPricing", () => {
         assert.ok(text.includes("GPT-4o"));
         assert.ok(text.includes("openai/gpt-4o"));
         assert.ok(text.includes("128,000"));
+        assert.ok(text.includes("$2.50 / 1M")); // Checking the 1M formatting
+        assert.ok(text.includes("versatile flagship")); // description included
+        assert.ok(text.includes("16,384")); // max completion tokens
     });
     it("returns fuzzy suggestions for a partial match", () => {
         const result = handleGetModelPricing(MOCK_MODELS, { model_id: "gpt-4" });
@@ -105,6 +141,12 @@ describe("handleCompareModelCosts", () => {
         assert.ok(text.includes("openai/gpt-4o"));
         assert.ok(text.includes("anthropic/claude-sonnet-4"));
         assert.ok(text.includes("|")); // table format
+        // Should include capability columns
+        assert.ok(text.includes("Tools"));
+        assert.ok(text.includes("Reasoning"));
+        assert.ok(text.includes("Vision"));
+        assert.ok(text.includes("JSON"));
+        assert.ok(text.includes("✅")); // Both have tools
     });
     it("reports not-found models alongside found ones", () => {
         const result = handleCompareModelCosts(MOCK_MODELS, {
@@ -144,8 +186,8 @@ describe("handleGetCheapestModels", () => {
     it("defaults to prompt and limit 10", () => {
         const result = handleGetCheapestModels(MOCK_MODELS, {});
         const text = result.content[0].text;
-        assert.ok(text.includes("prompt cost"));
-    });
+        assert.ok(text.includes("prompt"));
+    }); // Wait, the default falls back to prompt cost, which prints "prompt: $... / 1M" so text includes "prompt"
 });
 // ─── find_models_by_context_length ──────────────────────────────────────────
 describe("handleFindModelsByContextLength", () => {
@@ -166,10 +208,68 @@ describe("handleFindModelsByContextLength", () => {
         assert.ok(text.includes("Prompt:"));
     });
 });
+// ─── get_model_capabilities ─────────────────────────────────────────────────
+describe("handleGetModelCapabilities", () => {
+    it("returns capabilities for a known model with full metadata", () => {
+        const result = handleGetModelCapabilities(MOCK_MODELS, { model_id: "openai/gpt-4o" });
+        assert.equal(result.isError, undefined);
+        const text = result.content[0].text;
+        assert.ok(text.includes("GPT-4o"));
+        // Description section
+        assert.ok(text.includes("Description"));
+        assert.ok(text.includes("versatile flagship"));
+        // Pricing section
+        assert.ok(text.includes("Pricing"));
+        assert.ok(text.includes("$2.50 / 1M tokens"));
+        // Capabilities
+        assert.ok(text.includes("Tool Use"));
+        assert.ok(text.includes("✅ Yes")); // tool use should be yes
+        assert.ok(text.includes("Vision"));
+        assert.ok(text.includes("✅ Yes")); // vision should be yes (image input)
+        assert.ok(text.includes("Reasoning"));
+        assert.ok(text.includes("❌ No")); // GPT-4o doesn't have reasoning
+        assert.ok(text.includes("Tokenizer: GPT"));
+        assert.ok(text.includes("16,384")); // max completion tokens
+        assert.ok(text.includes("Content Moderated: Yes"));
+    });
+    it("returns capabilities for Claude with reasoning support", () => {
+        const result = handleGetModelCapabilities(MOCK_MODELS, { model_id: "anthropic/claude-sonnet-4" });
+        const text = result.content[0].text;
+        assert.ok(text.includes("Reasoning"));
+        assert.ok(text.includes("✅ Yes")); // Claude has reasoning
+        assert.ok(text.includes("Tokenizer: Claude"));
+        assert.ok(text.includes("128,000")); // max completion tokens
+    });
+    it("handles models with minimal metadata gracefully", () => {
+        const result = handleGetModelCapabilities(MOCK_MODELS, { model_id: "meta-llama/llama-3-8b-instruct:free" });
+        assert.equal(result.isError, undefined);
+        const text = result.content[0].text;
+        assert.ok(text.includes("Llama 3 8B"));
+        // Should show all capabilities as No since no supported_parameters
+        assert.ok(text.includes("❌ No"));
+    });
+    it("returns fuzzy suggestions for partial match", () => {
+        const result = handleGetModelCapabilities(MOCK_MODELS, { model_id: "gpt-4" });
+        assert.equal(result.isError, true);
+        assert.ok(result.content[0].text.includes("Did you mean"));
+    });
+    it("returns error for unknown model", () => {
+        const result = handleGetModelCapabilities(MOCK_MODELS, { model_id: "nonexistent/xyz" });
+        assert.equal(result.isError, true);
+        assert.ok(result.content[0].text.includes("not found"));
+    });
+    it("shows supported parameters list", () => {
+        const result = handleGetModelCapabilities(MOCK_MODELS, { model_id: "openai/gpt-4o" });
+        const text = result.content[0].text;
+        assert.ok(text.includes("Supported Parameters"));
+        assert.ok(text.includes("max_tokens"));
+        assert.ok(text.includes("tools"));
+    });
+});
 // ─── TOOL_DEFINITIONS ───────────────────────────────────────────────────────
 describe("TOOL_DEFINITIONS", () => {
-    it("exports exactly 5 tools", () => {
-        assert.equal(TOOL_DEFINITIONS.length, 5);
+    it("exports exactly 6 tools", () => {
+        assert.equal(TOOL_DEFINITIONS.length, 6);
     });
     it("all tools have a name, description, and inputSchema", () => {
         for (const tool of TOOL_DEFINITIONS) {
@@ -186,6 +286,7 @@ describe("TOOL_DEFINITIONS", () => {
             "compare_model_costs",
             "get_cheapest_models",
             "find_models_by_context_length",
+            "get_model_capabilities",
         ];
         const actualNames = TOOL_DEFINITIONS.map((t) => t.name);
         assert.deepEqual(actualNames, knownNames);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "openrouter-pricing-mcp",
-  "version": "1.0.2",
+  "version": "1.2.0",
   "description": "An MCP server providing live model pricing data from OpenRouter. Query, compare, and discover the cheapest AI models.",
   "type": "module",
   "main": "build/index.js",