uniqueness-mcp 0.2.0 → 0.3.0

package/index.mjs

@@ -14,7 +14,7 @@ import { z } from "zod";
 const BASE = process.env.UNIQUENESS_API_URL || "https://uniquenessengine.com";
 const KEY = process.env.UNIQUENESS_API_KEY || "";
 
-const server = new McpServer({ name: "uniqueness", version: "0.2.0" });
+const server = new McpServer({ name: "uniqueness", version: "0.3.0" });
 
 server.tool(
   "get_connection_brief",
@@ -44,19 +44,50 @@ server.tool(
     if (args.name) body.name = args.name;
     if (args.company) body.company = args.company;
     if (args.email) body.email = args.email;
-    if (args.deep) body.mode = "deep";
+    if (args.deep) { body.tier = "deep"; body.mode = "deep"; }
 
-    const res = await fetch(`${BASE}/api/brief`, {
+    // Async enrich+poll (mirrors cli/uniqueness.mjs `brief`). The sync /api/brief
+    // path returns {} to the client past the ~124s delivery cliff on fresh
+    // (uncached) profiles; /api/enrich submits a job we can poll to completion.
+    // A cache hit / immediate brief comes back inline on the submit.
+    const auth = { "content-type": "application/json", authorization: `Bearer ${KEY}` };
+    const sub = await fetch(`${BASE}/api/enrich`, {
       method: "POST",
-      headers: { "content-type": "application/json", authorization: `Bearer ${KEY}` },
+      headers: auth,
       body: JSON.stringify(body),
     });
-    const json = await res.json().catch(() => ({}));
-    if (res.status === 401) return { isError: true, content: [{ type: "text", text: "401 unauthorized — invalid API key." }] };
-    if (res.status === 402) return { isError: true, content: [{ type: "text", text: "Out of credits. " + (json.message || "") }] };
-    if (!res.ok) return { isError: true, content: [{ type: "text", text: `brief failed (${res.status}): ${json.error || ""}` }] };
+    const subJson = await sub.json().catch(() => ({}));
+    if (sub.status === 401) return { isError: true, content: [{ type: "text", text: "401 unauthorized — invalid API key." }] };
+    if (sub.status === 402) return { isError: true, content: [{ type: "text", text: "Out of credits. " + (subJson.message || "") }] };
+
+    // Fast path: cache hit / immediate brief.
+    let brief = subJson.brief || null;
+    let jobId = subJson.job_id || null;
+    if (!brief) {
+      if (!jobId) {
+        return { isError: true, content: [{ type: "text", text: `enrich failed (${sub.status}): ${subJson.error || subJson.message || ""}` }] };
+      }
+      // Poll GET /api/jobs/:id every 3s until a terminal state or the deadline.
+      // Briefs run ~45-180s, so give it up to ~210s before handing back the job_id.
+      const deadline = Date.now() + 210000;
+      while (Date.now() < deadline && !brief) {
+        await new Promise((r) => setTimeout(r, 3000));
+        const pr = await fetch(`${BASE}/api/jobs/${jobId}`, { headers: { authorization: `Bearer ${KEY}` } });
+        const pj = await pr.json().catch(() => ({}));
+        const st = pj.status;
+        if (st === "done") brief = pj.result;
+        else if (st === "refused") brief = pj.result || { status: "needs_review", note: "Identity could not be confidently resolved." };
+        else if (st === "failed") return { isError: true, content: [{ type: "text", text: `job failed: ${pj.error || "unknown"}` }] };
+      }
+      if (!brief) {
+        // Never return {} — hand back the job_id so the caller can re-run to keep polling.
+        return {
+          content: [{ type: "text", text: `Still processing (job ${jobId}). The brief is taking longer than usual — re-run get_connection_brief with the same inputs to keep polling.` }],
+        };
+      }
+    }
 
-    return { content: [{ type: "text", text: JSON.stringify(json, null, 2) }] };
+    return { content: [{ type: "text", text: JSON.stringify(brief, null, 2) }] };
   }
 );
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "uniqueness-mcp",
-  "version": "0.2.0",
+  "version": "0.3.0",
   "description": "MCP server for Uniqueness Engine — fact-checked connection briefs as an agent tool.",
   "type": "module",
   "bin": {