@vespermcp/mcp-server 1.3.0 → 1.3.1

package/build/lib/mcp-analytics.js

@@ -0,0 +1,164 @@
+import { getSupabaseAdminClient, resolveUserIdFromApiKey } from "./plan-resolve.js";
+import { inferSourceFromDatasetId, normalizeProviderSource, PLAN_GATE_EXEMPT_TOOLS, } from "./plan-gate.js";
+function truncateDatasetName(raw) {
+    const t = raw.trim();
+    if (!t)
+        return "unknown";
+    return t.length > 256 ? t.slice(0, 253) + "..." : t;
+}
+function pickDatasetName(args) {
+    const a = args || {};
+    const candidates = [
+        a.dataset_id,
+        a.query,
+        a.datasetId,
+        a.url,
+        a.file_path,
+    ];
+    for (const c of candidates) {
+        if (typeof c === "string" && c.trim()) {
+            return truncateDatasetName(c);
+        }
+    }
+    return "unknown";
+}
+function pickSource(toolName, args) {
+    const a = args || {};
+    const explicit = normalizeProviderSource(String(a.source ?? ""));
+    if (explicit)
+        return explicit;
+    const fromId = inferSourceFromDatasetId(String(a.dataset_id ?? a.query ?? ""));
+    if (fromId)
+        return fromId;
+    if (toolName === "vesper_web_find" && Array.isArray(a.sources) && a.sources.length > 0) {
+        return String(a.sources[0]).toLowerCase();
+    }
+    return null;
+}
+function pickFormat(args) {
+    const a = args || {};
+    const f = a.format ?? a.target_format ?? a.output_format;
+    if (typeof f === "string" && f.trim())
+        return f.trim().toLowerCase().slice(0, 32);
+    return null;
+}
+function mapToolToEvent(toolName) {
+    switch (toolName) {
+        case "vesper_search":
+        case "discover_datasets":
+        case "vesper_web_find":
+        case "vesper.extract_web":
+        case "get_dataset_info":
+            return { event_type: "dataset_search" };
+        case "download_dataset":
+        case "vesper_download_assets":
+            return { event_type: "dataset_download" };
+        case "quality_analyze":
+        case "analyze_quality":
+        case "analyze_image_quality":
+        case "analyze_media_quality":
+        case "generate_quality_report":
+        case "preview_cleaning":
+            return { event_type: "quality_analysis" };
+        case "prepare_dataset":
+            return { event_type: "dataset_prepare" };
+        case "export_dataset":
+        case "vesper_convert_format":
+            return { event_type: "export" };
+        default:
+            return { event_type: "data_processed" };
+    }
+}
+function mapToolToEventWithArgs(toolName, args) {
+    if (toolName === "unified_dataset_api") {
+        const op = String(args.operation ?? "").trim().toLowerCase();
+        if (op === "discover" || op === "providers" || op === "info") {
+            return { event_type: "dataset_search" };
+        }
+        if (op === "download") {
+            return { event_type: "dataset_download" };
+        }
+    }
+    return mapToolToEvent(toolName);
+}
+async function hasAnalyticsConsent(userId) {
+    const supabase = getSupabaseAdminClient();
+    if (!supabase)
+        return false;
+    const { data, error } = await supabase
+        .from("analytics_consent")
+        .select("consented")
+        .eq("user_id", userId)
+        .maybeSingle();
+    if (error) {
+        return false;
+    }
+    return data?.consented === true;
+}
+/**
+ * After each MCP tool call: insert one row into `analytics_events` (same table as
+ * `/api/analytics/ingest` and the landing Operations tab) when the user has opted in
+ * and `VESPER_API_KEY` / `api_key` resolves to a user.
+ */
+export async function recordMcpToolAnalyticsAfterCall(opts) {
+    if (process.env.VESPER_DISABLE_MCP_ANALYTICS === "1" || process.env.VESPER_DISABLE_MCP_ANALYTICS === "true") {
+        return;
+    }
+    const toolName = String(opts.toolName || "").trim();
+    if (!toolName || PLAN_GATE_EXEMPT_TOOLS.has(toolName)) {
+        return;
+    }
+    const supabase = getSupabaseAdminClient();
+    if (!supabase) {
+        return;
+    }
+    const args = opts.args || {};
+    const apiKey = String(args.api_key ?? process.env.VESPER_API_KEY ?? "").trim();
+    if (!apiKey) {
+        return;
+    }
+    const userId = await resolveUserIdFromApiKey(apiKey);
+    if (!userId) {
+        return;
+    }
+    if (!(await hasAnalyticsConsent(userId))) {
+        return;
+    }
+    const { event_type } = mapToolToEventWithArgs(toolName, args);
+    const dataset_name = pickDatasetName(args);
+    const source = pickSource(toolName, args);
+    const format = pickFormat(args);
+    const metadata = {
+        mcp_tool: toolName,
+        ok: opts.result !== undefined && opts.result.isError !== true,
+    };
+    if (opts.result?.isError) {
+        const first = opts.result.content?.[0];
+        if (first && typeof first.text === "string") {
+            metadata.error_preview = first.text.slice(0, 500);
+        }
+    }
+    const row = {
+        user_id: userId,
+        event_type,
+        dataset_name,
+        source: source || null,
+        format: format || null,
+        size_bytes: null,
+        quality_score: null,
+        metadata,
+        created_at: new Date().toISOString(),
+    };
+    let { error } = await supabase.from("analytics_events").insert(row);
+    // Older DBs may lack `dataset_prepare` in CHECK constraint — fall back.
+    if (error && event_type === "dataset_prepare" && /check|constraint/i.test(error.message || "")) {
+        ({ error } = await supabase.from("analytics_events").insert({
+            ...row,
+            event_type: "data_processed",
+            metadata: { ...metadata, event_type_fallback: "dataset_prepare" },
+        }));
+    }
+    if (error) {
+        console.error("[mcp-analytics] insert failed:", error.message);
+    }
+}

package/build/lib/plan-resolve.js

@@ -45,7 +45,7 @@ async function getUserPlanForUserId(userId) {
     }
     return "free";
 }
-async function resolveUserFromApiKey(apiKey) {
+async function lookupUserByApiKey(apiKey) {
     const supabase = getSupabase();
     if (!supabase)
         return null;
@@ -59,6 +59,14 @@ async function resolveUserFromApiKey(apiKey) {
     }
     return { userId: data.user_id };
 }
+/** For analytics / profile sync — same lookup as plan gate. */
+export async function resolveUserIdFromApiKey(apiKey) {
+    const row = await lookupUserByApiKey(apiKey);
+    return row?.userId ?? null;
+}
+export function getSupabaseAdminClient() {
+    return getSupabase();
+}
 /**
  * Central gate: same rules as `landing/lib/plan-entitlements` + landing analytics ingest.
  */
@@ -77,7 +85,7 @@ export async function enforcePlanGateForTool(toolName, args) {
             message: "Plan enforcement is enabled (Supabase configured). Set `VESPER_API_KEY` in the MCP env or pass `api_key` on tool calls to your Vesper API key so your tier can be verified.",
         };
     }
-    const user = await resolveUserFromApiKey(apiKey);
+    const user = await lookupUserByApiKey(apiKey);
     if (!user) {
         return {
             ok: false,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@vespermcp/mcp-server",
-  "version": "1.3.0",
+  "version": "1.3.1",
   "description": "AI-powered dataset discovery, quality analysis, and preparation MCP server with multimodal support (text, image, audio, video)",
   "type": "module",
   "main": "build/index.js",