@tokscale/cli 1.0.17 → 1.0.18

package/dist/cli.js

@@ -13,13 +13,13 @@ import pc from "picocolors";
 import { login, logout, whoami } from "./auth.js";
 import { submit } from "./submit.js";
 import { generateWrapped } from "./wrapped.js";
-import { PricingFetcher } from "./pricing.js";
 import { loadCursorCredentials, saveCursorCredentials, clearCursorCredentials, validateCursorSession, readCursorUsage, getCursorCredentialsPath, syncCursorCache, } from "./cursor.js";
 import { createUsageTable, formatUsageRow, formatTotalsRow, formatNumber, formatCurrency, formatModelName, } from "./table.js";
-import { isNativeAvailable, getNativeVersion, parseLocalSourcesAsync, finalizeReportAsync, finalizeMonthlyReportAsync, finalizeGraphAsync, } from "./native.js";
+import { getNativeVersion, parseLocalSourcesAsync, finalizeReportAsync, finalizeMonthlyReportAsync, finalizeGraphAsync, } from "./native.js";
 import { createSpinner } from "./spinner.js";
 import * as fs from "node:fs";
 import { performance } from "node:perf_hooks";
+import { loadSettings } from "./tui/config/settings.js";
 let cachedTUILoader = null;
 let tuiLoadAttempted = false;
 async function tryLoadTUI() {
@@ -131,7 +131,7 @@ async function main() {
     const program = new Command();
     program
         .name("tokscale")
-        .description("Tokscale - Track AI coding costs across OpenCode, Claude Code, Codex, Gemini, and Cursor")
+        .description("Tokscale - Track AI coding costs across OpenCode, Claude Code, Codex, Gemini, Cursor, and Amp")
         .version(pkg.version);
     program
         .command("monthly")
@@ -143,6 +143,8 @@ async function main() {
         .option("--codex", "Show only Codex CLI usage")
         .option("--gemini", "Show only Gemini CLI usage")
         .option("--cursor", "Show only Cursor IDE usage")
+        .option("--amp", "Show only Amp usage")
+        .option("--droid", "Show only Factory Droid usage")
         .option("--today", "Show only today's usage")
         .option("--week", "Show last 7 days")
         .option("--month", "Show current month")
@@ -150,12 +152,13 @@ async function main() {
         .option("--until <date>", "End date (YYYY-MM-DD)")
         .option("--year <year>", "Filter to specific year")
         .option("--benchmark", "Show processing time")
+        .option("--no-spinner", "Disable spinner (for AI agents and scripts - keeps stdout clean)")
         .action(async (options) => {
         if (options.json) {
             await outputJsonReport("monthly", options);
         }
         else if (options.light) {
-            await showMonthlyReport(options);
+            await showMonthlyReport(options, { spinner: options.spinner });
         }
         else {
             const launchTUI = await tryLoadTUI();
@@ -164,7 +167,7 @@ async function main() {
             }
             else {
                 showTUIUnavailableMessage();
-                await showMonthlyReport(options);
+                await showMonthlyReport(options, { spinner: options.spinner });
             }
         }
     });
@@ -178,6 +181,8 @@ async function main() {
         .option("--codex", "Show only Codex CLI usage")
         .option("--gemini", "Show only Gemini CLI usage")
         .option("--cursor", "Show only Cursor IDE usage")
+        .option("--amp", "Show only Amp usage")
+        .option("--droid", "Show only Factory Droid usage")
         .option("--today", "Show only today's usage")
         .option("--week", "Show last 7 days")
         .option("--month", "Show current month")
@@ -185,12 +190,13 @@ async function main() {
         .option("--until <date>", "End date (YYYY-MM-DD)")
         .option("--year <year>", "Filter to specific year")
         .option("--benchmark", "Show processing time")
+        .option("--no-spinner", "Disable spinner (for AI agents and scripts - keeps stdout clean)")
         .action(async (options) => {
         if (options.json) {
             await outputJsonReport("models", options);
         }
         else if (options.light) {
-            await showModelReport(options);
+            await showModelReport(options, { spinner: options.spinner });
         }
         else {
             const launchTUI = await tryLoadTUI();
@@ -199,7 +205,7 @@ async function main() {
             }
             else {
                 showTUIUnavailableMessage();
-                await showModelReport(options);
+                await showModelReport(options, { spinner: options.spinner });
             }
         }
     });
@@ -212,6 +218,8 @@ async function main() {
         .option("--codex", "Include only Codex CLI data")
         .option("--gemini", "Include only Gemini CLI data")
         .option("--cursor", "Include only Cursor IDE data")
+        .option("--amp", "Include only Amp data")
+        .option("--droid", "Include only Factory Droid data")
         .option("--today", "Show only today's usage")
         .option("--week", "Show last 7 days")
         .option("--month", "Show current month")
@@ -219,6 +227,7 @@ async function main() {
         .option("--until <date>", "End date (YYYY-MM-DD)")
         .option("--year <year>", "Filter to specific year")
         .option("--benchmark", "Show processing time")
+        .option("--no-spinner", "Disable spinner (for AI agents and scripts - keeps stdout clean)")
         .action(async (options) => {
         await handleGraphCommand(options);
     });
@@ -232,6 +241,8 @@ async function main() {
         .option("--codex", "Include only Codex CLI data")
         .option("--gemini", "Include only Gemini CLI data")
         .option("--cursor", "Include only Cursor IDE data")
+        .option("--amp", "Include only Amp data")
+        .option("--droid", "Include only Factory Droid data")
         .option("--no-spinner", "Disable loading spinner (for scripting)")
         .option("--short", "Display total tokens in abbreviated format (e.g., 7.14B)")
         .addOption(new Option("--agents", "Show Top OpenCode Agents (default)").conflicts("clients"))
@@ -269,6 +280,8 @@ async function main() {
         .option("--codex", "Include only Codex CLI data")
         .option("--gemini", "Include only Gemini CLI data")
         .option("--cursor", "Include only Cursor IDE data")
+        .option("--amp", "Include only Amp data")
+        .option("--droid", "Include only Factory Droid data")
         .option("--since <date>", "Start date (YYYY-MM-DD)")
         .option("--until <date>", "End date (YYYY-MM-DD)")
         .option("--year <year>", "Filter to specific year")
@@ -280,6 +293,8 @@ async function main() {
             codex: options.codex,
             gemini: options.gemini,
             cursor: options.cursor,
+            amp: options.amp,
+            droid: options.droid,
             since: options.since,
             until: options.until,
             year: options.year,
@@ -297,6 +312,8 @@ async function main() {
         .option("--codex", "Show only Codex CLI usage")
         .option("--gemini", "Show only Gemini CLI usage")
         .option("--cursor", "Show only Cursor IDE usage")
+        .option("--amp", "Show only Amp usage")
+        .option("--droid", "Show only Factory Droid usage")
         .option("--today", "Show only today's usage")
         .option("--week", "Show last 7 days")
         .option("--month", "Show current month")
@@ -313,9 +330,15 @@ async function main() {
             process.exit(1);
         }
     });
-    // =========================================================================
-    // Cursor IDE Authentication Commands
-    // =========================================================================
+    program
+        .command("pricing <model-id>")
+        .description("Look up pricing for a model")
+        .option("--json", "Output as JSON")
+        .option("--provider <source>", "Force pricing source: 'litellm' or 'openrouter'")
+        .option("--no-spinner", "Disable spinner (for AI agents and scripts - keeps stdout clean)")
+        .action(async (modelId, options) => {
+        await handlePricingCommand(modelId, options);
+    });
     const cursorCommand = program
         .command("cursor")
         .description("Cursor IDE integration commands");
@@ -343,7 +366,7 @@ async function main() {
     // Global flags should go to main program
     const isGlobalFlag = ['--help', '-h', '--version', '-V'].includes(firstArg);
     const hasSubcommand = args.length > 0 && !firstArg.startsWith('-');
-    const knownCommands = ['monthly', 'models', 'graph', 'wrapped', 'login', 'logout', 'whoami', 'submit', 'cursor', 'tui', 'help'];
+    const knownCommands = ['monthly', 'models', 'graph', 'wrapped', 'login', 'logout', 'whoami', 'submit', 'cursor', 'tui', 'pricing', 'help'];
     const isKnownCommand = hasSubcommand && knownCommands.includes(firstArg);
     if (isKnownCommand || isGlobalFlag) {
         // Run the specified subcommand or show full help/version
@@ -360,6 +383,7 @@ async function main() {
             .option("--codex", "Show only Codex CLI usage")
             .option("--gemini", "Show only Gemini CLI usage")
             .option("--cursor", "Show only Cursor IDE usage")
+            .option("--amp", "Show only Amp usage")
             .option("--today", "Show only today's usage")
             .option("--week", "Show last 7 days")
             .option("--month", "Show current month")
@@ -367,13 +391,14 @@ async function main() {
             .option("--until <date>", "End date (YYYY-MM-DD)")
             .option("--year <year>", "Filter to specific year")
             .option("--benchmark", "Show processing time")
+            .option("--no-spinner", "Disable spinner (for AI agents and scripts - keeps stdout clean)")
             .parse();
         const opts = defaultProgram.opts();
         if (opts.json) {
             await outputJsonReport("models", opts);
         }
         else if (opts.light) {
-            await showModelReport(opts);
+            await showModelReport(opts, { spinner: opts.spinner });
         }
         else {
             const launchTUI = await tryLoadTUI();
@@ -382,13 +407,13 @@ async function main() {
             }
             else {
                 showTUIUnavailableMessage();
-                await showModelReport(opts);
+                await showModelReport(opts, { spinner: opts.spinner });
             }
         }
     }
 }
 function getEnabledSources(options) {
-    const hasFilter = options.opencode || options.claude || options.codex || options.gemini || options.cursor;
+    const hasFilter = options.opencode || options.claude || options.codex || options.gemini || options.cursor || options.amp || options.droid;
     if (!hasFilter)
         return undefined; // All sources
     const sources = [];
@@ -402,19 +427,12 @@ function getEnabledSources(options) {
         sources.push("gemini");
     if (options.cursor)
         sources.push("cursor");
+    if (options.amp)
+        sources.push("amp");
+    if (options.droid)
+        sources.push("droid");
     return sources;
 }
-function logNativeStatus() {
-    if (!isNativeAvailable()) {
-        console.log(pc.yellow("  Note: Using TypeScript fallback (native module not available)"));
-        console.log(pc.gray("  Run 'bun run build:core' for ~10x faster processing.\n"));
-    }
-}
-async function fetchPricingData() {
-    const fetcher = new PricingFetcher();
-    await fetcher.fetchPricing();
-    return fetcher;
-}
 /**
  * Sync Cursor usage data from API to local cache.
  * Only attempts sync if user is authenticated with Cursor.
@@ -432,22 +450,10 @@ async function syncCursorData() {
         error: result.error,
     };
 }
-/**
- * Load all data sources in parallel (two-phase optimization):
- * - Cursor API sync (network)
- * - Pricing fetch (network)
- * - Local file parsing (CPU/IO) - OpenCode, Claude, Codex, Gemini
- *
- * This overlaps network I/O with local file parsing for better performance.
- */
-async function loadDataSourcesParallel(localSources, dateFilters) {
-    // Skip local parsing if no local sources requested (e.g., cursor-only mode)
+async function loadDataSourcesParallel(localSources, dateFilters, onPhase) {
     const shouldParseLocal = localSources.length > 0;
-    // Use Promise.allSettled for graceful degradation
-    const [cursorResult, pricingResult, localResult] = await Promise.allSettled([
+    const [cursorResult, localResult] = await Promise.allSettled([
         syncCursorData(),
-        fetchPricingData(),
-        // Parse local sources in parallel (excludes Cursor) - skip if empty
         shouldParseLocal
             ? parseLocalSourcesAsync({
                 sources: localSources.filter(s => s !== 'cursor'),
@@ -457,20 +463,15 @@ async function loadDataSourcesParallel(localSources, dateFilters) {
             })
             : Promise.resolve(null),
     ]);
-    // Handle partial failures gracefully
     const cursorSync = cursorResult.status === 'fulfilled'
         ? cursorResult.value
         : { attempted: true, synced: false, rows: 0, error: 'Cursor sync failed' };
-    const fetcher = pricingResult.status === 'fulfilled'
-        ? pricingResult.value
-        : new PricingFetcher(); // Empty pricing → costs = 0
     const localMessages = localResult.status === 'fulfilled'
         ? localResult.value
         : null;
-    return { fetcher, cursorSync, localMessages };
+    return { cursorSync, localMessages };
 }
-async function showModelReport(options) {
-    logNativeStatus();
+async function showModelReport(options, extraOptions) {
     const dateFilters = getDateFilters(options);
     const enabledSources = getEnabledSources(options);
     const onlyCursor = enabledSources?.length === 1 && enabledSources[0] === 'cursor';
@@ -493,27 +494,28 @@ async function showModelReport(options) {
         console.log(pc.gray(`  Using: Rust native module v${getNativeVersion()}`));
     }
     console.log();
-    // Start spinner for loading phase
-    const spinner = createSpinner({ color: "cyan" });
-    spinner.start(pc.gray("Loading data sources..."));
-    // Filter out cursor for local parsing (it's synced separately via network)
-    const localSources = (enabledSources || ['opencode', 'claude', 'codex', 'gemini', 'cursor'])
+    const useSpinner = extraOptions?.spinner !== false;
+    const spinner = useSpinner ? createSpinner({ color: "cyan" }) : null;
+    const localSources = (enabledSources || ['opencode', 'claude', 'codex', 'gemini', 'cursor', 'amp', 'droid'])
         .filter(s => s !== 'cursor');
-    // Two-phase parallel loading: network (Cursor + pricing) overlaps with local file parsing
-    // If cursor-only, skip local parsing entirely
-    const { fetcher, cursorSync, localMessages } = await loadDataSourcesParallel(onlyCursor ? [] : localSources, dateFilters);
+    spinner?.start(pc.gray("Scanning session data..."));
+    const { cursorSync, localMessages } = await loadDataSourcesParallel(onlyCursor ? [] : localSources, dateFilters, (phase) => spinner?.update(phase));
     if (!localMessages && !onlyCursor) {
-        spinner.error('Failed to parse local session files');
+        if (spinner) {
+            spinner.error('Failed to parse local session files');
+        }
+        else {
+            console.error('Failed to parse local session files');
+        }
         process.exit(1);
     }
-    spinner.update(pc.gray("Finalizing report..."));
+    spinner?.update(pc.gray("Finalizing report..."));
     const startTime = performance.now();
     let report;
     try {
-        const emptyMessages = { messages: [], opencodeCount: 0, claudeCount: 0, codexCount: 0, geminiCount: 0, processingTimeMs: 0 };
+        const emptyMessages = { messages: [], opencodeCount: 0, claudeCount: 0, codexCount: 0, geminiCount: 0, ampCount: 0, droidCount: 0, processingTimeMs: 0 };
         report = await finalizeReportAsync({
             localMessages: localMessages || emptyMessages,
-            pricing: fetcher.toPricingEntries(),
             includeCursor: includeCursor && cursorSync.synced,
             since: dateFilters.since,
             until: dateFilters.until,
@@ -521,11 +523,16 @@ async function showModelReport(options) {
         });
     }
     catch (e) {
-        spinner.error(`Error: ${e.message}`);
+        if (spinner) {
+            spinner.error(`Error: ${e.message}`);
+        }
+        else {
+            console.error(`Error: ${e.message}`);
+        }
         process.exit(1);
     }
     const processingTime = performance.now() - startTime;
-    spinner.stop();
+    spinner?.stop();
     if (report.entries.length === 0) {
         if (onlyCursor && !cursorSync.synced) {
             console.log(pc.yellow("  No Cursor data available."));
@@ -538,7 +545,11 @@ async function showModelReport(options) {
     }
     // Create table
     const table = createUsageTable("Source/Model");
-    for (const entry of report.entries) {
+    const settings = loadSettings();
+    const filteredEntries = settings.includeUnusedModels
+        ? report.entries
+        : report.entries.filter(e => e.input + e.output + e.cacheRead + e.cacheWrite > 0);
+    for (const entry of filteredEntries) {
         const sourceLabel = getSourceLabel(entry.source);
         const modelDisplay = `${pc.dim(sourceLabel)} ${formatModelName(entry.model)}`;
         table.push(formatUsageRow(modelDisplay, [entry.model], entry.input, entry.output, entry.cacheWrite, entry.cacheRead, entry.cost));
@@ -563,8 +574,7 @@ async function showModelReport(options) {
     }
     console.log();
 }
-async function showMonthlyReport(options) {
-    logNativeStatus();
+async function showMonthlyReport(options, extraOptions) {
     const dateRange = getDateRangeLabel(options);
     const title = dateRange
         ? `Monthly Token Usage Report (${dateRange})`
@@ -574,28 +584,30 @@ async function showMonthlyReport(options) {
         console.log(pc.gray(`  Using: Rust native module v${getNativeVersion()}`));
     }
     console.log();
-    // Start spinner for loading phase
-    const spinner = createSpinner({ color: "cyan" });
-    spinner.start(pc.gray("Loading data sources..."));
+    const useSpinner = extraOptions?.spinner !== false;
+    const spinner = useSpinner ? createSpinner({ color: "cyan" }) : null;
     const dateFilters = getDateFilters(options);
     const enabledSources = getEnabledSources(options);
-    // Filter out cursor for local parsing (it's synced separately via network)
-    const localSources = (enabledSources || ['opencode', 'claude', 'codex', 'gemini', 'cursor'])
+    const localSources = (enabledSources || ['opencode', 'claude', 'codex', 'gemini', 'cursor', 'amp', 'droid'])
         .filter(s => s !== 'cursor');
     const includeCursor = !enabledSources || enabledSources.includes('cursor');
-    // Two-phase parallel loading: network (Cursor + pricing) overlaps with local file parsing
-    const { fetcher, cursorSync, localMessages } = await loadDataSourcesParallel(localSources, dateFilters);
+    spinner?.start(pc.gray("Scanning session data..."));
+    const { cursorSync, localMessages } = await loadDataSourcesParallel(localSources, dateFilters, (phase) => spinner?.update(phase));
     if (!localMessages) {
-        spinner.error('Failed to parse local session files');
+        if (spinner) {
+            spinner.error('Failed to parse local session files');
+        }
+        else {
+            console.error('Failed to parse local session files');
+        }
         process.exit(1);
     }
-    spinner.update(pc.gray("Finalizing report..."));
+    spinner?.update(pc.gray("Finalizing report..."));
     const startTime = performance.now();
     let report;
     try {
         report = await finalizeMonthlyReportAsync({
             localMessages,
-            pricing: fetcher.toPricingEntries(),
             includeCursor: includeCursor && cursorSync.synced,
             since: dateFilters.since,
             until: dateFilters.until,
@@ -603,18 +615,27 @@ async function showMonthlyReport(options) {
         });
     }
     catch (e) {
-        spinner.error(`Error: ${e.message}`);
+        if (spinner) {
+            spinner.error(`Error: ${e.message}`);
+        }
+        else {
+            console.error(`Error: ${e.message}`);
+        }
         process.exit(1);
     }
     const processingTime = performance.now() - startTime;
-    spinner.stop();
+    spinner?.stop();
     if (report.entries.length === 0) {
         console.log(pc.yellow("  No usage data found.\n"));
         return;
     }
     // Create table
     const table = createUsageTable("Month");
-    for (const entry of report.entries) {
+    const settings = loadSettings();
+    const filteredEntries = settings.includeUnusedModels
+        ? report.entries
+        : report.entries.filter(e => e.input + e.output + e.cacheRead + e.cacheWrite > 0);
+    for (const entry of filteredEntries) {
         table.push(formatUsageRow(entry.month, entry.models, entry.input, entry.output, entry.cacheWrite, entry.cacheRead, entry.cost));
     }
     // Add totals row
@@ -639,23 +660,21 @@ async function showMonthlyReport(options) {
     console.log();
 }
 async function outputJsonReport(reportType, options) {
-    logNativeStatus();
     const dateFilters = getDateFilters(options);
     const enabledSources = getEnabledSources(options);
     const onlyCursor = enabledSources?.length === 1 && enabledSources[0] === 'cursor';
     const includeCursor = !enabledSources || enabledSources.includes('cursor');
-    const localSources = (enabledSources || ['opencode', 'claude', 'codex', 'gemini', 'cursor'])
+    const localSources = (enabledSources || ['opencode', 'claude', 'codex', 'gemini', 'cursor', 'amp', 'droid'])
         .filter(s => s !== 'cursor');
-    const { fetcher, cursorSync, localMessages } = await loadDataSourcesParallel(onlyCursor ? [] : localSources, dateFilters);
+    const { cursorSync, localMessages } = await loadDataSourcesParallel(onlyCursor ? [] : localSources, dateFilters);
     if (!localMessages && !onlyCursor) {
         console.error(JSON.stringify({ error: "Failed to parse local session files" }));
         process.exit(1);
     }
-    const emptyMessages = { messages: [], opencodeCount: 0, claudeCount: 0, codexCount: 0, geminiCount: 0, processingTimeMs: 0 };
+    const emptyMessages = { messages: [], opencodeCount: 0, claudeCount: 0, codexCount: 0, geminiCount: 0, ampCount: 0, droidCount: 0, processingTimeMs: 0 };
     if (reportType === "models") {
         const report = await finalizeReportAsync({
             localMessages: localMessages || emptyMessages,
-            pricing: fetcher.toPricingEntries(),
             includeCursor: includeCursor && cursorSync.synced,
             since: dateFilters.since,
             until: dateFilters.until,
@@ -666,7 +685,6 @@ async function outputJsonReport(reportType, options) {
     else {
         const report = await finalizeMonthlyReportAsync({
             localMessages: localMessages || emptyMessages,
-            pricing: fetcher.toPricingEntries(),
             includeCursor: includeCursor && cursorSync.synced,
             since: dateFilters.since,
             until: dateFilters.until,
@@ -676,18 +694,15 @@ async function outputJsonReport(reportType, options) {
     }
 }
 async function handleGraphCommand(options) {
-    logNativeStatus();
-    // Start spinner for loading phase (only if outputting to file, not stdout)
-    const spinner = options.output ? createSpinner({ color: "cyan" }) : null;
-    spinner?.start(pc.gray("Loading data sources..."));
+    const useSpinner = options.output && options.spinner !== false;
+    const spinner = useSpinner ? createSpinner({ color: "cyan" }) : null;
     const dateFilters = getDateFilters(options);
     const enabledSources = getEnabledSources(options);
-    // Filter out cursor for local parsing (it's synced separately via network)
-    const localSources = (enabledSources || ['opencode', 'claude', 'codex', 'gemini', 'cursor'])
+    const localSources = (enabledSources || ['opencode', 'claude', 'codex', 'gemini', 'cursor', 'amp', 'droid'])
         .filter(s => s !== 'cursor');
     const includeCursor = !enabledSources || enabledSources.includes('cursor');
-    // Two-phase parallel loading: network (Cursor + pricing) overlaps with local file parsing
-    const { fetcher, cursorSync, localMessages } = await loadDataSourcesParallel(localSources, dateFilters);
+    spinner?.start(pc.gray("Scanning session data..."));
+    const { cursorSync, localMessages } = await loadDataSourcesParallel(localSources, dateFilters, (phase) => spinner?.update(phase));
     if (!localMessages) {
         spinner?.error('Failed to parse local session files');
         process.exit(1);
@@ -696,7 +711,6 @@ async function handleGraphCommand(options) {
     const startTime = performance.now();
     const data = await finalizeGraphAsync({
         localMessages,
-        pricing: fetcher.toPricingEntries(),
         includeCursor: includeCursor && cursorSync.synced,
         since: dateFilters.since,
         until: dateFilters.until,
@@ -760,6 +774,111 @@ async function handleWrappedCommand(options) {
         process.exit(1);
     }
 }
+async function handlePricingCommand(modelId, options) {
+    const validProviders = ["litellm", "openrouter"];
+    if (options.provider && !validProviders.includes(options.provider.toLowerCase())) {
+        console.log(pc.red(`\n  Invalid provider: ${options.provider}`));
+        console.log(pc.gray(`  Valid providers: ${validProviders.join(", ")}\n`));
+        process.exit(1);
+    }
+    const useSpinner = options.spinner !== false;
+    const spinner = useSpinner ? createSpinner({ color: "cyan" }) : null;
+    const providerLabel = options.provider ? ` from ${options.provider}` : "";
+    spinner?.start(pc.gray(`Fetching pricing data${providerLabel}...`));
+    let core;
+    try {
+        const mod = await import("@tokscale/core");
+        core = (mod.default ?? mod);
+    }
+    catch (importErr) {
+        spinner?.stop();
+        const errorMsg = importErr.message || "Unknown error";
+        if (options.json) {
+            console.log(JSON.stringify({ error: "Native module not available", details: errorMsg }, null, 2));
+        }
+        else {
+            console.log(pc.red(`\n  Native module not available: ${errorMsg}`));
+            console.log(pc.gray("  Run 'bun run build:core' to build the native module.\n"));
+        }
+        process.exit(1);
+    }
+    try {
+        const provider = options.provider?.toLowerCase() || undefined;
+        const nativeResult = await core.lookupPricing(modelId, provider);
+        spinner?.stop();
+        const result = {
+            matchedKey: nativeResult.matchedKey,
+            source: nativeResult.source,
+            pricing: {
+                input_cost_per_token: nativeResult.pricing.inputCostPerToken,
+                output_cost_per_token: nativeResult.pricing.outputCostPerToken,
+                cache_read_input_token_cost: nativeResult.pricing.cacheReadInputTokenCost,
+                cache_creation_input_token_cost: nativeResult.pricing.cacheCreationInputTokenCost,
+            },
+        };
+        if (options.json) {
+            console.log(JSON.stringify({
+                modelId,
+                matchedKey: result.matchedKey,
+                source: result.source,
+                pricing: {
+                    inputCostPerToken: result.pricing.input_cost_per_token ?? 0,
+                    outputCostPerToken: result.pricing.output_cost_per_token ?? 0,
+                    cacheReadInputTokenCost: result.pricing.cache_read_input_token_cost,
+                    cacheCreationInputTokenCost: result.pricing.cache_creation_input_token_cost,
+                },
+            }, null, 2));
+        }
+        else {
+            const sourceLabel = result.source.toLowerCase() === "litellm" ? pc.blue("LiteLLM") : pc.magenta("OpenRouter");
+            const inputCost = result.pricing.input_cost_per_token ?? 0;
+            const outputCost = result.pricing.output_cost_per_token ?? 0;
+            const cacheReadCost = result.pricing.cache_read_input_token_cost;
+            const cacheWriteCost = result.pricing.cache_creation_input_token_cost;
+            console.log(pc.cyan(`\n  Pricing for: ${pc.white(modelId)}`));
+            console.log(pc.gray(`  Matched key: ${result.matchedKey}`));
+            console.log(pc.gray(`  Source: `) + sourceLabel);
+            console.log();
+            console.log(pc.white(`  Input:  `) + formatPricePerMillion(inputCost));
+            console.log(pc.white(`  Output: `) + formatPricePerMillion(outputCost));
+            if (cacheReadCost !== undefined) {
+                console.log(pc.white(`  Cache Read:  `) + formatPricePerMillion(cacheReadCost));
+            }
+            if (cacheWriteCost !== undefined) {
+                console.log(pc.white(`  Cache Write: `) + formatPricePerMillion(cacheWriteCost));
+            }
+            console.log();
+        }
+    }
+    catch (err) {
+        spinner?.stop();
+        const errorMsg = err.message || "Unknown error";
+        // Check if this is a "model not found" error from Rust or a different error
+        const isModelNotFound = errorMsg.toLowerCase().includes("not found") ||
+            errorMsg.toLowerCase().includes("no pricing");
+        if (options.json) {
+            if (isModelNotFound) {
+                console.log(JSON.stringify({ error: "Model not found", modelId }, null, 2));
+            }
+            else {
+                console.log(JSON.stringify({ error: errorMsg, modelId }, null, 2));
+            }
+        }
+        else {
+            if (isModelNotFound) {
+                console.log(pc.red(`\n  Model not found: ${modelId}\n`));
+            }
+            else {
+                console.log(pc.red(`\n  Error looking up pricing: ${errorMsg}\n`));
+            }
+        }
+        process.exit(1);
+    }
+}
+function formatPricePerMillion(costPerToken) {
+    const perMillion = costPerToken * 1_000_000;
+    return pc.green(`$${perMillion.toFixed(2)}`) + pc.gray(" / 1M tokens");
+}
 function getSourceLabel(source) {
     switch (source) {
         case "opencode":
@@ -772,6 +891,10 @@ function getSourceLabel(source) {
             return "Gemini";
         case "cursor":
             return "Cursor";
+        case "amp":
+            return "Amp";
+        case "droid":
+            return "Droid";
         default:
             return source;
     }