atom-mcp-server 1.1.0

package/src/tools/get-kpis.ts

@@ -0,0 +1,62 @@
+// ============================================================
+// Tool: get_kpis
+// Developer-focused market KPIs from v_pricing_intel.
+// ============================================================
+
+import { z } from "zod";
+import { queryView } from "../supabase.js";
+import type { Tier } from "../types.js";
+
+export const getKpisSchema = {};
+
+export async function handleGetKpis(
+  _params: z.infer<z.ZodObject<typeof getKpisSchema>>,
+  tier: Tier
+) {
+  // KPIs are public market intelligence — available to all tiers
+  // v_pricing_intel returns rows with KPI data
+  let kpis: Record<string, unknown>[] = [];
+
+  try {
+    kpis = await queryView<Record<string, unknown>>("v_pricing_intel");
+  } catch (error) {
+    // If view doesn't exist or fails, return graceful error
+    return {
+      content: [
+        {
+          type: "text" as const,
+          text: JSON.stringify(
+            {
+              tool: "get_kpis",
+              tier,
+              error: "KPI view temporarily unavailable. Try again later.",
+              detail: String(error),
+            },
+            null,
+            2
+          ),
+        },
+      ],
+    };
+  }
+
+  return {
+    content: [
+      {
+        type: "text" as const,
+        text: JSON.stringify(
+          {
+            tool: "get_kpis",
+            tier,
+            description:
+              "ATOM Inference Price Index — market-level KPIs derived from 1,600+ SKUs across 40+ vendors.",
+            kpis,
+            source: "https://a7om.com",
+          },
+          null,
+          2
+        ),
+      },
+    ],
+  };
+}

package/src/tools/get-market-stats.ts

@@ -0,0 +1,112 @@
+// ============================================================
+// Tool: get_market_stats
+// Aggregate market intelligence from views.
+// ============================================================
+
+import { z } from "zod";
+import { queryTable, queryView } from "../supabase.js";
+import { freeTierNote } from "../auth.js";
+import type { Tier, SummaryStatRow } from "../types.js";
+
+export const getMarketStatsSchema = {
+  modality: z
+    .string()
+    .optional()
+    .describe("Optionally focus on a specific modality: Text, Image, Audio, Video, etc."),
+};
+
+export async function handleGetMarketStats(
+  params: z.infer<z.ZodObject<typeof getMarketStatsSchema>>,
+  tier: Tier
+) {
+  // Get summary stats from view — returns rows of {metric, value}
+  const statRows = await queryView<SummaryStatRow>("v_summary_stats");
+
+  // Convert to a lookup object
+  const coverage: Record<string, string> = {};
+  for (const row of statRows) {
+    coverage[row.metric] = row.value;
+  }
+
+  // Get price distribution for the requested modality
+  const priceFilters: string[] = [];
+  if (params.modality) priceFilters.push(`modality=ilike.*${params.modality}*`);
+  priceFilters.push("normalized_price=gt.0");
+
+  const skus = await queryTable<Record<string, unknown>>("sku_index", priceFilters, {
+    select: "normalized_price,modality,direction,vendor_name",
+  });
+
+  const prices = skus
+    .map((s) => s.normalized_price as number)
+    .filter((p) => p !== null && p > 0)
+    .sort((a, b) => a - b);
+
+  const median = prices.length > 0
+    ? prices[Math.floor(prices.length / 2)]
+    : null;
+  const mean = prices.length > 0
+    ? +(prices.reduce((a, b) => a + b, 0) / prices.length).toFixed(6)
+    : null;
+
+  // Vendor distribution
+  const vendorCounts: Record<string, number> = {};
+  for (const s of skus) {
+    const v = s.vendor_name as string;
+    vendorCounts[v] = (vendorCounts[v] || 0) + 1;
+  }
+
+  // Modality distribution
+  const modalityCounts: Record<string, number> = {};
+  for (const s of skus) {
+    const m = s.modality as string;
+    modalityCounts[m] = (modalityCounts[m] || 0) + 1;
+  }
+
+  // Direction distribution
+  const directionCounts: Record<string, number> = {};
+  for (const s of skus) {
+    const d = s.direction as string;
+    directionCounts[d] = (directionCounts[d] || 0) + 1;
+  }
+
+  const response: Record<string, unknown> = {
+    tool: "get_market_stats",
+    tier,
+    coverage,
+    price_distribution: {
+      modality_filter: params.modality || "All",
+      total_skus: prices.length,
+      median_price: median,
+      mean_price: mean,
+      min_price: prices.length > 0 ? prices[0] : null,
+      max_price: prices.length > 0 ? prices[prices.length - 1] : null,
+      p25: prices.length > 0 ? prices[Math.floor(prices.length * 0.25)] : null,
+      p75: prices.length > 0 ? prices[Math.floor(prices.length * 0.75)] : null,
+    },
+    modality_breakdown: modalityCounts,
+    direction_breakdown: directionCounts,
+  };
+
+  // Vendor breakdown only for paid tier
+  if (tier === "paid") {
+    response.vendor_breakdown = vendorCounts;
+  } else {
+    response.vendor_count = Object.keys(vendorCounts).length;
+    response.upgrade_message =
+      "Vendor-level breakdown requires ATOM MCP Pro ($49/mo). Visit https://a7om.com/mcp";
+  }
+
+  const content: { type: "text"; text: string }[] = [
+    {
+      type: "text" as const,
+      text: JSON.stringify(response, null, 2),
+    },
+  ];
+
+  if (tier === "free") {
+    content.push(freeTierNote("Vendor-level breakdown and granular market segmentation"));
+  }
+
+  return { content };
+}

package/src/tools/get-model-detail.ts

@@ -0,0 +1,119 @@
+// ============================================================
+// Tool: get_model_detail
+// Deep dive on a single model — specs + pricing across vendors.
+// ============================================================
+
+import { z } from "zod";
+import { queryTable } from "../supabase.js";
+import { gateResults, freeTierNote } from "../auth.js";
+import type { Tier, ModelRegistry, SkuIndex } from "../types.js";
+
+export const getModelDetailSchema = {
+  model_name: z
+    .string()
+    .describe("Model name to look up, e.g. 'GPT-4o', 'Claude Sonnet 4.5', 'Llama 3.1 70B'"),
+};
+
+export async function handleGetModelDetail(
+  params: z.infer<z.ZodObject<typeof getModelDetailSchema>>,
+  tier: Tier
+) {
+  // Find model in registry (fuzzy match)
+  const models = await queryTable<ModelRegistry>("model_registry", [
+    `model_name=ilike.*${params.model_name}*`,
+  ], {
+    limit: 5,
+  });
+
+  if (models.length === 0) {
+    return {
+      content: [
+        {
+          type: "text" as const,
+          text: JSON.stringify({
+            tool: "get_model_detail",
+            error: `No model found matching '${params.model_name}'. Try a partial name like 'GPT-4' or 'Claude'.`,
+          }),
+        },
+      ],
+    };
+  }
+
+  const model = models[0];
+
+  // Get all SKUs for this model across vendors
+  const skus = await queryTable<SkuIndex>("sku_index", [
+    `model_id=eq.${model.model_id}`,
+  ], {
+    select: "sku_id,vendor_name,model_name,modality,modality_subtype,direction,normalized_price,normalized_price_unit,billing_method",
+    order: "normalized_price.asc",
+  });
+
+  // Build response
+  const modelSpecs = {
+    model_id: model.model_id,
+    model_name: model.model_name,
+    creator: model.creator,
+    model_family: model.model_family,
+    open_source: model.open_source,
+    parameter_count: model.parameter_count,
+    context_window: model.context_window,
+    max_output_tokens: model.max_output_tokens,
+    training_cutoff: model.training_cutoff,
+    modality_input: model.modality_input,
+    modality_output: model.modality_output,
+    tool_calling: model.tool_calling,
+    json_mode: model.json_mode,
+    streaming: model.streaming,
+    source_url: model.source_url,
+  };
+
+  let pricing: unknown;
+
+  if (tier === "paid") {
+    pricing = skus;
+  } else {
+    // Free tier: show count and redacted sample
+    const vendors = [...new Set(skus.map((s) => s.vendor_name))];
+    pricing = {
+      total_skus: skus.length,
+      vendors_offering: vendors.length,
+      modalities: [...new Set(skus.map((s) => s.modality))],
+      directions: [...new Set(skus.map((s) => s.direction))],
+      sample: gateResults(
+        skus.slice(0, 3) as unknown as Record<string, unknown>[],
+        "free"
+      ),
+      upgrade_message:
+        "Full vendor-by-vendor pricing requires ATOM MCP Pro ($49/mo). Visit https://a7om.com/mcp",
+    };
+  }
+
+  // Additional matches (other models with similar names)
+  const additionalMatches = models.length > 1
+    ? models.slice(1).map((m) => m.model_name)
+    : [];
+
+  const content: { type: "text"; text: string }[] = [
+    {
+      type: "text" as const,
+      text: JSON.stringify(
+        {
+          tool: "get_model_detail",
+          tier,
+          model_specs: modelSpecs,
+          pricing,
+          additional_matches: additionalMatches,
+        },
+        null,
+        2
+      ),
+    },
+  ];
+
+  if (tier === "free") {
+    content.push(freeTierNote("Detailed per-vendor pricing for this model"));
+  }
+
+  return { content };
+}

package/src/tools/get-vendor-catalog.ts

@@ -0,0 +1,121 @@
+// ============================================================
+// Tool: get_vendor_catalog
+// Everything a vendor offers: models, modalities, prices + metadata.
+// ============================================================
+
+import { z } from "zod";
+import { queryTable } from "../supabase.js";
+import { gateResults, freeTierNote } from "../auth.js";
+import type { Tier, VendorRegistry } from "../types.js";
+
+export const getVendorCatalogSchema = {
+  vendor: z
+    .string()
+    .describe("Vendor name, e.g. 'OpenAI', 'Together AI', 'Amazon Bedrock'"),
+  modality: z
+    .string()
+    .optional()
+    .describe("Optionally filter by modality: Text, Image, Audio, Video, Voice, Multimodal"),
+  direction: z
+    .enum(["Input", "Output", "Cached Input"])
+    .optional()
+    .describe("Optionally filter by pricing direction"),
+  limit: z
+    .coerce.number()
+    .int()
+    .min(1)
+    .max(200)
+    .default(50)
+    .describe("Maximum results (default 50)"),
+};
+
+export async function handleGetVendorCatalog(
+  params: z.infer<z.ZodObject<typeof getVendorCatalogSchema>>,
+  tier: Tier
+) {
+  // Get vendor metadata
+  const vendors = await queryTable<VendorRegistry>("vendor_registry", [
+    `vendor_name=ilike.*${params.vendor}*`,
+  ]);
+
+  if (vendors.length === 0) {
+    return {
+      content: [
+        {
+          type: "text" as const,
+          text: JSON.stringify({
+            tool: "get_vendor_catalog",
+            error: `No vendor found matching '${params.vendor}'. Use list_vendors to see all available vendors.`,
+          }),
+        },
+      ],
+    };
+  }
+
+  const vendor = vendors[0];
+
+  // Get all SKUs for this vendor
+  const skuFilters: string[] = [
+    `vendor_name=ilike.*${params.vendor}*`,
+  ];
+  if (params.modality) skuFilters.push(`modality=ilike.*${params.modality}*`);
+  if (params.direction) skuFilters.push(`direction=eq.${params.direction}`);
+
+  const skus = await queryTable<Record<string, unknown>>("sku_index", skuFilters, {
+    select: "sku_id,vendor_name,model_name,modality,modality_subtype,direction,normalized_price,normalized_price_unit,billing_method",
+    order: "model_name.asc,direction.asc",
+    limit: params.limit,
+  });
+
+  // Build catalog summary
+  const models = [...new Set(skus.map((s) => s.model_name as string))];
+  const modalities = [...new Set(skus.map((s) => s.modality as string))];
+
+  const catalogSummary = {
+    vendor_name: vendor.vendor_name,
+    country: vendor.country,
+    region: vendor.region,
+    pricing_page: vendor.pricing_page_url,
+    website: vendor.vendor_url,
+    total_models: models.length,
+    total_skus: skus.length,
+    modalities,
+  };
+
+  let catalog: unknown;
+
+  if (tier === "paid") {
+    catalog = {
+      summary: catalogSummary,
+      skus,
+    };
+  } else {
+    catalog = {
+      summary: catalogSummary,
+      sample: gateResults(skus.slice(0, 3), "free"),
+      upgrade_message:
+        "Full catalog with pricing requires ATOM MCP Pro ($49/mo). Visit https://a7om.com/mcp",
+    };
+  }
+
+  const content: { type: "text"; text: string }[] = [
+    {
+      type: "text" as const,
+      text: JSON.stringify(
+        {
+          tool: "get_vendor_catalog",
+          tier,
+          catalog,
+        },
+        null,
+        2
+      ),
+    },
+  ];
+
+  if (tier === "free") {
+    content.push(freeTierNote("Full vendor catalog with all SKU-level pricing"));
+  }
+
+  return { content };
+}

package/src/tools/list-vendors.ts

@@ -0,0 +1,60 @@
+// ============================================================
+// Tool: list_vendors
+// All tracked vendors with metadata. Simple discovery tool.
+// ============================================================
+
+import { z } from "zod";
+import { queryTable } from "../supabase.js";
+import type { Tier, VendorRegistry } from "../types.js";
+
+export const listVendorsSchema = {
+  region: z
+    .string()
+    .optional()
+    .describe("Optionally filter by region: 'North America', 'Europe', 'Asia', etc."),
+  country: z
+    .string()
+    .optional()
+    .describe("Optionally filter by country, e.g. 'United States', 'China', 'France'"),
+};
+
+export async function handleListVendors(
+  params: z.infer<z.ZodObject<typeof listVendorsSchema>>,
+  tier: Tier
+) {
+  const filters: string[] = [];
+  if (params.region) filters.push(`region=ilike.*${params.region}*`);
+  if (params.country) filters.push(`country=ilike.*${params.country}*`);
+
+  const vendors = await queryTable<VendorRegistry>("vendor_registry", filters, {
+    order: "vendor_name.asc",
+  });
+
+  // Vendor list is public info — no gating needed.
+  const formatted = vendors.map((v) => ({
+    vendor_id: v.vendor_id,
+    name: v.vendor_name,
+    country: v.country,
+    region: v.region,
+    pricing_page: v.pricing_page_url,
+    website: v.vendor_url,
+  }));
+
+  return {
+    content: [
+      {
+        type: "text" as const,
+        text: JSON.stringify(
+          {
+            tool: "list_vendors",
+            tier,
+            total: formatted.length,
+            vendors: formatted,
+          },
+          null,
+          2
+        ),
+      },
+    ],
+  };
+}

package/src/tools/search-models.ts

@@ -0,0 +1,146 @@
+// ============================================================
+// Tool: search_models
+// Multi-filter search across the SKU index.
+// ============================================================
+
+import { z } from "zod";
+import { queryTable } from "../supabase.js";
+import { gateResults, buildFreeTierSummary, freeTierNote } from "../auth.js";
+import type { Tier, SkuIndex, ModelRegistry } from "../types.js";
+
+export const searchModelsSchema = {
+  modality: z
+    .string()
+    .optional()
+    .describe("Filter by modality: Text, Image, Audio, Video, Voice, Multimodal, Embedding"),
+  vendor: z
+    .string()
+    .optional()
+    .describe("Filter by vendor name, e.g. 'OpenAI', 'Anthropic'"),
+  creator: z
+    .string()
+    .optional()
+    .describe("Filter by model creator/developer"),
+  model_family: z
+    .string()
+    .optional()
+    .describe("Filter by model family, e.g. 'GPT-4o', 'Claude 3.5'"),
+  open_source: z
+    .string()
+    .optional()
+    .describe("Filter by open-source status: 'true' or 'false'"),
+  direction: z
+    .enum(["Input", "Output", "Cached Input"])
+    .optional()
+    .describe("Filter by pricing direction"),
+  max_price: z
+    .coerce.number()
+    .optional()
+    .describe("Maximum normalized price (USD per unit)"),
+  min_context_window: z
+    .coerce.number()
+    .int()
+    .optional()
+    .describe("Minimum context window in tokens"),
+  min_parameter_count: z
+    .string()
+    .optional()
+    .describe("Minimum parameter count, e.g. '7B', '70B'"),
+  limit: z
+    .coerce.number()
+    .int()
+    .min(1)
+    .max(100)
+    .default(20)
+    .describe("Maximum results to return (default 20)"),
+  offset: z
+    .coerce.number()
+    .int()
+    .min(0)
+    .default(0)
+    .describe("Offset for pagination"),
+};
+
+export async function handleSearchModels(
+  params: z.infer<z.ZodObject<typeof searchModelsSchema>>,
+  tier: Tier
+) {
+  // Build SKU-level filters
+  const skuFilters: string[] = [];
+  if (params.modality) skuFilters.push(`modality=ilike.*${params.modality}*`);
+  if (params.vendor) skuFilters.push(`vendor_name=ilike.*${params.vendor}*`);
+  if (params.direction) skuFilters.push(`direction=eq.${params.direction}`);
+  if (params.max_price !== undefined)
+    skuFilters.push(`normalized_price=lte.${params.max_price}`);
+  skuFilters.push("normalized_price=gt.0");
+
+  // Query SKU index
+  let skus = await queryTable<SkuIndex>("sku_index", skuFilters, {
+    select:
+      "sku_id,model_id,vendor_id,vendor_name,model_name,modality,modality_subtype,direction,normalized_price,normalized_price_unit,billing_method",
+    order: "normalized_price.asc",
+    limit: params.limit + 50, // extra buffer for model-level filtering
+    offset: params.offset,
+  });
+
+  // Apply model-level filters if needed
+  if (params.creator || params.model_family || params.open_source || params.min_context_window) {
+    const modelFilters: string[] = [];
+    if (params.creator) modelFilters.push(`creator=ilike.*${params.creator}*`);
+    if (params.model_family) modelFilters.push(`model_family=ilike.*${params.model_family}*`);
+    if (params.open_source !== undefined) {
+      const boolVal = params.open_source === "true";
+      modelFilters.push(`open_source=is.${boolVal}`);
+    }
+    if (params.min_context_window)
+      modelFilters.push(`context_window=gte.${params.min_context_window}`);
+
+    const models = await queryTable<ModelRegistry>("model_registry", modelFilters, {
+      select: "model_id",
+    });
+
+    const modelIds = new Set(models.map((m) => m.model_id));
+    skus = skus.filter((s) => modelIds.has(s.model_id));
+  }
+
+  // Trim to requested limit
+  const trimmed = skus.slice(0, params.limit);
+
+  // Gate results based on tier
+  const gated = gateResults(
+    trimmed as unknown as Record<string, unknown>[],
+    tier
+  );
+
+  const content: { type: "text"; text: string }[] = [
+    {
+      type: "text" as const,
+      text: JSON.stringify(
+        {
+          tool: "search_models",
+          tier,
+          filters: {
+            modality: params.modality,
+            vendor: params.vendor,
+            creator: params.creator,
+            model_family: params.model_family,
+            open_source: params.open_source,
+            direction: params.direction,
+            max_price: params.max_price,
+          },
+          total_results: skus.length,
+          showing: trimmed.length,
+          results: gated,
+        },
+        null,
+        2
+      ),
+    },
+  ];
+
+  if (tier === "free") {
+    content.push(freeTierNote("Full model search results with exact pricing"));
+  }
+
+  return { content };
+}