corent-mcp 0.3.0 → 0.4.1

package/dist/server.js

@@ -15,8 +15,26 @@
  */
 import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
 import { z } from "zod";
+import { MEDIA_WIDGET_HTML } from "./widget-html.js";
 export const DEFAULT_API_URL = "https://api.corent.tech";
-export const SERVER_VERSION = "0.3.0";
+export const SERVER_VERSION = "0.4.0";
+// --- MCP Apps (SEP-1865): the media tools render an inline widget in hosts
+// that support it (claude.ai web/desktop, others). Hosts without the extension
+// ignore the _meta and fall back to plain text results, so this is additive.
+const WIDGET_URI = "ui://corent/media";
+const WIDGET_MIME = "text/html;profile=mcp-app";
+// Both the canonical nested key and the legacy flat key — the spec tells hosts
+// to check both, and older host builds only know the flat one.
+const WIDGET_TOOL_META = {
+    ui: { resourceUri: WIDGET_URI },
+    "ui/resourceUri": WIDGET_URI,
+};
+// Generated media lives on Supabase storage; the logo and any brand assets on
+// corent.tech. Everything else stays blocked by the host's CSP.
+const WIDGET_CSP = {
+    connectDomains: [],
+    resourceDomains: ["https://hzokyylcctfybulkgmcb.supabase.co", "https://corent.tech"],
+};
 class CorentApiError extends Error {
     status;
     detail;
@@ -33,6 +51,8 @@ function errorCode(err) {
         return "invalid_api_key";
     if (err.status === 402)
         return "insufficient_balance";
+    if (err.status === 404)
+        return "not_found";
     if (err.status === 422 && detail.includes("content"))
         return "prompt_not_allowed";
     if (err.status === 422)
@@ -60,7 +80,7 @@ export function createCorentServer(config = {}) {
         // can advertise its tools to catalogs/agents without a key configured.
         if (!apiKey) {
             throw new CorentApiError(401, {
-                message: "CORENT_API_KEY is not set. Get a key at https://corent.tech and add it to your MCP client config.",
+                message: "No Corent credentials. On claude.ai or other remote MCP clients, reconnect the Corent connector and approve access (OAuth). For local/stdio use, set CORENT_API_KEY in your MCP client config. Keys: https://corent.tech/dashboard/api-keys",
             });
         }
         const res = await fetch(`${apiUrl}${path}`, {
@@ -104,10 +124,38 @@ export function createCorentServer(config = {}) {
             }
         };
     }
-    const server = new McpServer({ name: "corent", version: SERVER_VERSION });
+    // icons/websiteUrl: spec-standard server identity — MCP clients that support
+    // it (claude.ai included) show the square Corent mark instead of a letter tile.
+    const server = new McpServer({
+        name: "corent",
+        title: "Corent",
+        version: SERVER_VERSION,
+        icons: [{ src: "https://corent.tech/brand/intro-mark.png", mimeType: "image/png", sizes: ["512x512"] }],
+        websiteUrl: "https://corent.tech",
+    });
+    server.registerResource("corent-media-widget", WIDGET_URI, {
+        title: "Corent media widget",
+        description: "Inline card that tracks a generation and plays the result.",
+        mimeType: WIDGET_MIME,
+        _meta: { ui: { csp: WIDGET_CSP } },
+    }, async () => ({
+        contents: [
+            {
+                uri: WIDGET_URI,
+                mimeType: WIDGET_MIME,
+                text: MEDIA_WIDGET_HTML,
+                _meta: { ui: { csp: WIDGET_CSP } },
+            },
+        ],
+    }));
     // Loose output shapes: every field optional so an additive API change can
     // never break a client's schema validation.
-    const mediaMeta = z.object({ model: z.string().optional(), cost_cents: z.number().optional() }).passthrough();
+    // model is null while a video job is still processing (no provider chosen
+    // yet) -- must be nullable, not just optional, or the SDK's output validator
+    // throws on the normal poll-loop response.
+    const mediaMeta = z
+        .object({ model: z.string().nullable().optional(), cost_cents: z.number().nullable().optional() })
+        .passthrough();
     const imageOutput = {
         id: z.string().optional(),
         status: z.string().optional(),
@@ -124,9 +172,14 @@ export function createCorentServer(config = {}) {
     };
     server.registerTool("generate_image", {
         title: "Generate image",
+        _meta: WIDGET_TOOL_META,
         description: "Generate an image from a text prompt. Returns a permanent public URL. Synchronous: typically completes in 2-20 seconds. Costs a few cents, billed to the Corent account. Failed generations are never billed.",
         inputSchema: {
             prompt: z.string().describe("What to generate, in plain language"),
+            tier: z
+                .enum(["air", "lite", "premium", "pro", "max_pro"])
+                .optional()
+                .describe("Quality tier (preferred): air=cheapest/fastest, lite=everyday, premium=production, pro=advanced, max_pro=maximum fidelity (flagship models). Overrides preference."),
             preference: z
                 .enum(["fast", "cheap", "quality", "balanced"])
                 .default("balanced")
@@ -139,15 +192,20 @@ export function createCorentServer(config = {}) {
         },
         outputSchema: imageOutput,
         annotations: { readOnlyHint: false, destructiveHint: false, idempotentHint: false, openWorldHint: true },
-    }, wrap(async ({ prompt, preference, style, aspect_ratio }) => corent("/v1/images/generate", {
+    }, wrap(async ({ prompt, tier, preference, style, aspect_ratio }) => corent("/v1/images/generate", {
         method: "POST",
-        body: JSON.stringify({ prompt, preference, style, aspect_ratio }),
+        body: JSON.stringify({ prompt, tier, preference, style, aspect_ratio }),
     })));
     server.registerTool("generate_video", {
         title: "Generate video",
+        _meta: WIDGET_TOOL_META,
         description: "Start generating a video from a text prompt (optionally animating a source image). Asynchronous: returns a job id immediately; poll get_job until status is 'completed'. Video costs more than images (tens of cents to a few dollars depending on tier). Failed generations are never billed.",
         inputSchema: {
             prompt: z.string().describe("What to generate, in plain language"),
+            tier: z
+                .enum(["air", "lite", "premium", "pro", "max_pro"])
+                .optional()
+                .describe("Quality tier (preferred): air=cheapest/fastest, lite=everyday, premium=production, pro=advanced, max_pro=maximum fidelity (flagship models). Overrides preference."),
             preference: z.enum(["fast", "cheap", "quality", "balanced"]).default("balanced"),
             aspect_ratio: z.enum(["16:9", "9:16", "1:1"]).default("16:9"),
             duration_s: z.number().int().min(1).max(30).optional().describe("Requested duration in seconds"),
@@ -155,12 +213,13 @@ export function createCorentServer(config = {}) {
         },
         outputSchema: jobOutput,
         annotations: { readOnlyHint: false, destructiveHint: false, idempotentHint: false, openWorldHint: true },
-    }, wrap(async ({ prompt, preference, aspect_ratio, duration_s, image_url }) => corent("/v1/videos/generate", {
+    }, wrap(async ({ prompt, tier, preference, aspect_ratio, duration_s, image_url }) => corent("/v1/videos/generate", {
         method: "POST",
-        body: JSON.stringify({ prompt, preference, aspect_ratio, duration_s, image_url }),
+        body: JSON.stringify({ prompt, tier, preference, aspect_ratio, duration_s, image_url }),
     })));
     server.registerTool("get_job", {
         title: "Check job status",
+        _meta: WIDGET_TOOL_META,
         description: "Check the status of a generation job (mainly videos). When completed, the response includes the permanent media URL and the cost. Read-only and free.",
         inputSchema: { job_id: z.string().describe("The job id returned by generate_video or generate_image") },
         outputSchema: jobOutput,
@@ -199,6 +258,7 @@ export function createCorentServer(config = {}) {
     }, wrap(async ({ intent }) => corent("/v1/intent", { method: "POST", body: JSON.stringify({ intent }) })));
     server.registerTool("create", {
         title: "Create (plan + generate, budget-capped)",
+        _meta: WIDGET_TOOL_META,
         description: "Describe what you want in plain language and Corent plans AND generates it — choosing image vs video, tier, aspect ratio, and duration for you (the 'zero decisions' path). You MUST pass max_cost_cents as a spend ceiling; if the estimated cost exceeds it, nothing is generated and you're told the estimate so you can raise the ceiling. Image results come back with a URL; video results come back as a job id to poll with get_job. Failed generations are never billed.",
         inputSchema: {
             intent: z.string().describe("What you want, in natural language"),