@wavespeed/mcp 1.2.2 → 1.2.4

package/assets/result.html

@@ -56,7 +56,7 @@
   // A small MCP Apps view (https://modelcontextprotocol.io/seps/1865-mcp-apps-interactive-user-interfaces-for-mcp):
   // it speaks JSON-RPC to the host over postMessage and shows what the tool returned.
   var PROTOCOL = "2026-01-26";
-  var nextId = 1, pending = {}, args = {}, current = null, pollToken = 0, started = 0, ticker = null;
+  var nextId = 1, pending = {}, args = {}, current = null, pollToken = 0, started = 0, ticker = null, hostCaps = null;
   var $ = function (id) { return document.getElementById(id); };
 
   function send(msg) { try { window.parent.postMessage(msg, "*"); } catch (e) {} }
@@ -78,7 +78,11 @@
       var p = pending[m.id];
       if (!p) return;
       delete pending[m.id];
-      if (m.error) p.reject(new Error(m.error.message || "error")); else p.resolve(m.result);
+      if (m.error) {
+        var err = new Error(m.error.message || "error");
+        err.code = m.error.code;
+        p.reject(err);
+      } else p.resolve(m.result);
       return;
     }
     if (m.method === "ui/notifications/tool-input") { args = (m.params && m.params.arguments) || {}; showPrompt(); }
@@ -251,7 +255,9 @@
         var label = el("span", "", "Generating…");
         w.appendChild(label); body.appendChild(w);
         started = started || Date.now();
-        var tick = function () { var s = Math.round((Date.now() - started) / 1000); label.textContent = "Generating… " + s + " s. Video can take a few minutes."; };
+        // Only a video run says it can take minutes; images and audio finish much sooner.
+        var isVideo = /video|t2v|i2v/i.test(String(d.model || ""));
+        var tick = function () { var s = Math.round((Date.now() - started) / 1000); label.textContent = "Generating… " + s + " s." + (isVideo ? " Video can take a few minutes." : ""); };
         tick(); ticker = setInterval(tick, 1000);
         setState(d.status);
       } else if (d.note) {
@@ -272,13 +278,51 @@
     resize();
   }
 
+  // The card can no longer follow the run (the host refused its tool call, or it gave up): stop the timer and
+  // the spinner so it does not look as if it is still generating, and say where the result will appear.
+  function stopFollowing(message) {
+    clearInterval(ticker);
+    var body = $("body");
+    if (!(current && (current.outputs || []).length)) {
+      body.textContent = "";
+      body.appendChild(el("p", "", "Started. This card stops here; the result appears when the assistant checks on the run."));
+      setState("submitted");
+    }
+    $("note").textContent = message; $("note").hidden = false; resize();
+  }
+
+  // Say what actually went wrong when the host refused the card's tool call, so it can be told apart from a timeout.
+  function whyNot(err) {
+    if (window.console) console.warn("wavespeed-result: get_prediction failed", err, "hostCapabilities:", hostCaps);
+    if (err && err.message === "timeout") return "no answer from this chat in 130 seconds";
+    var said = "this chat said: " + ((err && err.message) || "error") + (err && err.code !== undefined ? " [" + err.code + "]" : "");
+    if (hostCaps && !hostCaps.serverTools) said += "; it does not declare tool calls for cards";
+    return said;
+  }
+
+  // Ask the server for the run through the host: the standard bridge first (tools/call), then ChatGPT's own
+  // window.openai.callTool when the standard call is refused. Both answer with the tool result (ChatGPT's may come back
+  // as {structuredContent}, {content} or {result: "<json text>"}).
+  function callServerTool(name, toolArgs) {
+    return request("tools/call", { name: name, arguments: toolArgs }).catch(function (err) {
+      var o = window.openai;
+      if (!o || typeof o.callTool !== "function" || (err && err.message === "timeout")) throw err;
+      return Promise.resolve(o.callTool(name, toolArgs)).then(function (r) {
+        if (r && typeof r.result === "string" && !r.structuredContent && !r.content) {
+          try { return { structuredContent: JSON.parse(r.result) }; } catch (e) { return { content: [{ type: "text", text: r.result }] }; }
+        }
+        return r;
+      }, function () { throw err; });
+    });
+  }
+
   // Keep asking the server while the run is going: this is what turns a video's wait into a card that fills in.
   function poll(id, token) {
     var asked = 0;
     (function again() {
       if (token !== pollToken) return;
       asked++;
-      request("tools/call", { name: "get_prediction", arguments: { id: id, wait_seconds: 25 } }).then(function (res) {
+      callServerTool("get_prediction", { id: id, wait_seconds: 25 }).then(function (res) {
         if (token !== pollToken) return;
         var d = res && res.structuredContent;
         if (!d) {
@@ -286,16 +330,15 @@
           try { d = JSON.parse(text); } catch (e) {}
         }
         if (res && res.isError && !(d && d.error)) d = { error: ((res.content || [])[0] || {}).text || "The run failed." };
-        if (!d) { $("note").textContent = "Still running. Ask the assistant to check on it."; $("note").hidden = false; return; }
+        if (!d) { stopFollowing("Still running. Ask the assistant to check on it."); return; }
         if (d.error || d.done || ["completed", "failed", "cancelled", "timeout", "deleted"].indexOf(d.status) >= 0) {
           if (d.status && d.status !== "completed" && !d.error) d.error = "Prediction " + d.status + (d.error ? ": " + d.error : "");
           current = d; render(d); return;
         }
-        if (asked < 80) again(); else { $("note").textContent = "Still running on the server. Ask the assistant to check on it."; $("note").hidden = false; }
-      }).catch(function () {
+        if (asked < 80) again(); else stopFollowing("Still running on the server. Ask the assistant to check on it.");
+      }).catch(function (err) {
         if (token !== pollToken) return;
-        $("note").textContent = "This chat cannot refresh the card by itself. Ask the assistant to check on the run.";
-        $("note").hidden = false; resize();
+        stopFollowing("The card could not check on the run (" + whyNot(err) + "). Ask the assistant to check on it.");
       });
     })();
   }
@@ -311,7 +354,7 @@
 
   // Handshake with the host, then wait for tool-input and tool-result.
   request("ui/initialize", { protocolVersion: PROTOCOL, appInfo: { name: "wavespeed-result", version: "1.0.0" }, appCapabilities: {} })
-    .then(function (r) { if (r && r.hostContext) applyContext(r.hostContext); notify("ui/notifications/initialized", {}); })
+    .then(function (r) { hostCaps = (r && r.hostCapabilities) || {}; if (r && r.hostContext) applyContext(r.hostContext); notify("ui/notifications/initialized", {}); })
     .catch(function () { notify("ui/notifications/initialized", {}); });
 
   // ChatGPT's own bridge, when the host offers it instead of the standard messages.

package/dist/lib/app.js

@@ -6,9 +6,10 @@ import { readFileSync } from 'node:fs';
 export const RESULT_VIEW_URI = 'ui://wavespeed/result.html';
 export const APP_MIME_TYPE = 'text/html;profile=mcp-app';
 export const TOP_UP_URL = 'https://wavespeed.ai/top-up';
+// get_prediction is not one of them: the card asks it for itself while the run goes on, and an assistant that polls it
+// would draw a second card for the same result. It is marked callable from the card instead (see toolMeta).
 export const UI_TOOLS = new Set([
     'run_model',
-    'get_prediction',
     'generate_image',
     'edit_image',
     'generate_video',
@@ -18,6 +19,38 @@ export const UI_TOOLS = new Set([
 // "https:" alone should allow every https source, but a host may only honour named hosts when it builds the
 // card's CSP, so the usual media hosts are named too, as wildcards.
 export const VIEW_RESOURCE_DOMAINS = ['https:', 'https://*.cloudfront.net', 'https://*.wavespeed.ai', 'https://*.amazonaws.com'];
+// The one-line status ChatGPT shows while a tool runs and after it ran (it allows 64 characters).
+const TOOL_STATUS = {
+    run_model: ['Starting the run…', 'Run started'],
+    generate_image: ['Starting the run…', 'Run started'],
+    edit_image: ['Starting the run…', 'Run started'],
+    generate_video: ['Starting the run…', 'Run started'],
+    edit_video: ['Starting the run…', 'Run started'],
+    extend_video: ['Starting the run…', 'Run started'],
+    get_prediction: ['Checking the run…', 'Run checked'],
+};
+/**
+ * The tool's _meta for the card: the run tools open it; get_prediction does not but says the card may call it
+ * (ChatGPT only lets a widget call a tool marked openai/widgetAccessible; MCP Apps hosts read ui.visibility, whose
+ * default already includes "app"). Same as the remote server (api-server mcp_app.go).
+ */
+export function toolMeta(name) {
+    const linked = UI_TOOLS.has(name);
+    if (!linked && name !== 'get_prediction')
+        return undefined;
+    const meta = {
+        ui: linked ? { resourceUri: RESULT_VIEW_URI, visibility: ['model', 'app'] } : { visibility: ['model', 'app'] },
+        'openai/widgetAccessible': true,
+    };
+    if (linked)
+        meta['openai/outputTemplate'] = RESULT_VIEW_URI;
+    const status = TOOL_STATUS[name];
+    if (status) {
+        meta['openai/toolInvocation/invoking'] = status[0];
+        meta['openai/toolInvocation/invoked'] = status[1];
+    }
+    return meta;
+}
 export function loadView() {
     return readFileSync(new URL('../../assets/result.html', import.meta.url), 'utf8');
 }

package/dist/server.js

@@ -24,7 +24,7 @@ import { MAX_DELETE_IDS, prepareDelete, summarizeDelete } from './lib/delete-pre
 // Where to add credit; handed to the assistant next to the balance.
 const TOP_UP_URL = 'https://wavespeed.ai/top-up';
 import { resolveLocalFiles } from './lib/local-files.js';
-import { APP_MIME_TYPE, RESULT_VIEW_URI, UI_TOOLS, VIEW_RESOURCE_DOMAINS, insufficientCreditsResult, isInsufficientCredits, loadView, } from './lib/app.js';
+import { APP_MIME_TYPE, RESULT_VIEW_URI, toolMeta, VIEW_RESOURCE_DOMAINS, insufficientCreditsResult, isInsufficientCredits, loadView, } from './lib/app.js';
 import { uploadWithCache } from './lib/upload-cache.js';
 import { findUnknownInputs } from './lib/validate-inputs.js';
 import { IMAGE_MODEL_KEYS, VIDEO_MODEL_KEYS } from './lib/media-models.js';
@@ -160,7 +160,16 @@ function ok(payload) {
     return res;
 }
 export function createServer() {
-    const server = new McpServer({ name: 'wavespeed', version: PKG_VERSION }, {
+    const server = new McpServer({
+        name: 'wavespeed',
+        title: 'WaveSpeed',
+        version: PKG_VERSION,
+        websiteUrl: 'https://wavespeed.ai',
+        icons: [
+            { src: 'https://wavespeed.ai/favicon.svg', mimeType: 'image/svg+xml', sizes: ['any'] },
+            { src: 'https://wavespeed.ai/apple-touch-icon.png', mimeType: 'image/png', sizes: ['192x192'] },
+        ],
+    }, {
         instructions: [
             'WaveSpeed AI media generation (image, video, audio, 3D).',
             'Pattern: list_models to find a model, get_model_schema to see its',
@@ -194,8 +203,9 @@ export function createServer() {
             }
         };
         const tool = registerTool(...args);
-        if (UI_TOOLS.has(name))
-            tool.update({ _meta: { ui: { resourceUri: RESULT_VIEW_URI } } });
+        const meta = toolMeta(name);
+        if (meta)
+            tool.update({ _meta: meta });
         return tool;
     };
     server.registerResource('result-card', RESULT_VIEW_URI, {
@@ -208,7 +218,13 @@ export function createServer() {
                 uri: uri.href,
                 mimeType: APP_MIME_TYPE,
                 text: loadView(),
-                _meta: { ui: { csp: { resourceDomains: VIEW_RESOURCE_DOMAINS, connectDomains: [] }, prefersBorder: true } },
+                _meta: {
+                    ui: { csp: { resourceDomains: VIEW_RESOURCE_DOMAINS, connectDomains: [] }, prefersBorder: true },
+                    // The same, in the names ChatGPT's own extensions use, for versions that read only those.
+                    'openai/widgetCSP': { resource_domains: VIEW_RESOURCE_DOMAINS, connect_domains: [] },
+                    'openai/widgetPrefersBorder': true,
+                    'openai/widgetDescription': 'Shows the images, videos and audio of a WaveSpeed run and follows it until it finishes.',
+                },
             },
         ],
     }));
@@ -220,7 +236,7 @@ export function createServer() {
             .describe('Filter by modality type, e.g. text-to-image, image-to-video'),
         limit: z.number().int().min(1).max(200).default(30).describe('Max results'),
         refresh: z.boolean().default(false).describe('Bypass the 1h catalog cache'),
-    }, { readOnlyHint: true, openWorldHint: true }, async ({ query, type, limit, refresh }) => {
+    }, { readOnlyHint: true, destructiveHint: false, openWorldHint: true }, async ({ query, type, limit, refresh }) => {
         const models = await fetchModels({ refresh });
         const q = query?.toLowerCase();
         const filtered = models.filter((m) => {
@@ -237,7 +253,7 @@ export function createServer() {
             models: filtered.slice(0, limit).map(compactModel),
         });
     });
-    server.tool('get_model_schema', "Get a model's real input schema (required fields, properties, defaults). Call this before run_model so inputs match what the model actually accepts.", { model: z.string().describe('Model ID from list_models, e.g. bytedance/seedream-v5.0-pro') }, { readOnlyHint: true, openWorldHint: true }, async ({ model }) => {
+    server.tool('get_model_schema', "Get a model's real input schema (required fields, properties, defaults). Call this before run_model so inputs match what the model actually accepts.", { model: z.string().describe('Model ID from list_models, e.g. bytedance/seedream-v5.0-pro') }, { readOnlyHint: true, destructiveHint: false, openWorldHint: true }, async ({ model }) => {
         const models = await fetchModels();
         const meta = models.find((m) => m.model_id === model);
         if (!meta) {
@@ -427,7 +443,7 @@ export function createServer() {
         const data = await fetchPricing(model, priceInput);
         return ok({ ...data, estimate: true, disclaimer: PRICE_DISCLAIMER, playground_url: modelPlaygroundUrl(model) });
     });
-    server.tool('get_balance', 'Show the WaveSpeed account credit balance, with the link to top up.', {}, { readOnlyHint: true, openWorldHint: true }, async () => {
+    server.tool('get_balance', 'Show the WaveSpeed account credit balance, with the link to top up.', {}, { readOnlyHint: true, destructiveHint: false, openWorldHint: true }, async () => {
         return ok({ ...(await fetchBalance()), top_up_url: TOP_UP_URL });
     });
     server.tool('list_predictions', 'List past predictions (run history), newest window first. Defaults to the last 24 hours; pass since/until (ISO or YYYY-MM-DD) for other ranges. Use get_prediction with an id for one task in full.', {
@@ -437,7 +453,7 @@ export function createServer() {
         status: z.enum(PREDICTION_STATUSES).optional(),
         since: z.string().optional().describe('Created after this time (ISO or YYYY-MM-DD)'),
         until: z.string().optional().describe('Created before this time (ISO or YYYY-MM-DD)'),
-    }, { readOnlyHint: true, openWorldHint: true }, async ({ page, limit, model, status, since, until }) => {
+    }, { readOnlyHint: true, destructiveHint: false, openWorldHint: true }, async ({ page, limit, model, status, since, until }) => {
         return ok(await fetchHistory({
             page,
             pageSize: limit,
@@ -455,7 +471,7 @@ export function createServer() {
         key: z.string().optional().describe('Filter by the API key uuid that generated the charge'),
         since: z.string().optional().describe('Records after this time (ISO or YYYY-MM-DD)'),
         until: z.string().optional().describe('Records before this time (ISO or YYYY-MM-DD)'),
-    }, { readOnlyHint: true, openWorldHint: true }, async ({ page, limit, type, model, key, since, until }) => {
+    }, { readOnlyHint: true, destructiveHint: false, openWorldHint: true }, async ({ page, limit, type, model, key, since, until }) => {
         return ok(await fetchBillings({
             page,
             pageSize: limit,
@@ -470,7 +486,7 @@ export function createServer() {
         since: z.string().optional().describe('Start (ISO or YYYY-MM-DD); default 7 days ago'),
         until: z.string().optional().describe('End (ISO or YYYY-MM-DD); default now'),
         model: z.string().optional().describe('Filter by model uuid'),
-    }, { readOnlyHint: true, openWorldHint: true }, async ({ since, until, model }) => {
+    }, { readOnlyHint: true, destructiveHint: false, openWorldHint: true }, async ({ since, until, model }) => {
         const end = toIso(until, 'until') ?? new Date().toISOString();
         const start = toIso(since, 'since') ?? new Date(Date.parse(end) - 7 * 24 * 60 * 60 * 1000).toISOString();
         return ok(await fetchUsageStats(start, end, model));
@@ -495,7 +511,7 @@ export function createServer() {
             .max(1800)
             .default(0)
             .describe('Max seconds to wait for a terminal status; 0 = return the current status now'),
-    }, { readOnlyHint: true, openWorldHint: true }, async ({ id, wait_seconds }, extra) => {
+    }, { readOnlyHint: true, destructiveHint: false, openWorldHint: true }, async ({ id, wait_seconds }, extra) => {
         const started = Date.now();
         const { item, done } = await waitForPrediction(id, {
             intervalMs: POLL_INTERVAL_MS,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@wavespeed/mcp",
-  "version": "1.2.2",
+  "version": "1.2.4",
   "description": "WaveSpeed MCP server — run any model on the live WaveSpeed catalog (image, video, audio, 3D) with schema introspection, local-file upload, and price quotes.",
   "type": "module",
   "bin": {