@wavespeed/mcp 1.2.3 → 1.2.5

package/assets/result.html

@@ -300,13 +300,29 @@
     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) {

package/dist/lib/app.js

@@ -3,12 +3,14 @@
 // the metadata and show the text as before. The view is the same file the
 // remote server (api-server /mcp) serves; keep the two copies identical.
 import { readFileSync } from 'node:fs';
-export const RESULT_VIEW_URI = 'ui://wavespeed/result.html';
+// A host keeps a UI resource by its URI, so a changed card only reaches it under a new URI: bump the number.
+export const RESULT_VIEW_URI = 'ui://wavespeed/result-v2.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 +20,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';
@@ -203,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, {
@@ -217,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.',
+                },
             },
         ],
     }));
@@ -229,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) => {
@@ -246,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) {
@@ -436,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.', {
@@ -446,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,
@@ -464,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,
@@ -479,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));
@@ -504,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.3",
+  "version": "1.2.5",
   "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": {