@wavespeed/mcp 1.2.1 → 1.2.3

package/assets/result.html

@@ -18,9 +18,12 @@
   .prompt { color: var(--muted); margin: 0 0 10px; display: -webkit-box; -webkit-line-clamp: 2; -webkit-box-orient: vertical; overflow: hidden; }
   .grid { display: grid; gap: 8px; grid-template-columns: repeat(auto-fill, minmax(220px, 1fr)); }
   .grid.one { grid-template-columns: 1fr; }
-  .tile { background: var(--card); border: 1px solid var(--line); border-radius: 8px; overflow: hidden; }
+  .tile { position: relative; background: var(--card); border: 1px solid var(--line); border-radius: 8px; overflow: hidden; }
   .tile img, .tile video { display: block; width: 100%; max-height: 520px; object-fit: contain; background: var(--card); }
   .tile video { background: #000; }
+  .tile-spin { position: absolute; top: 40%; left: 50%; margin: -9px 0 0 -9px; pointer-events: none; }
+  .code { margin: 0; padding: 12px; max-height: 360px; overflow: auto; font: 12px/1.5 ui-monospace, SFMono-Regular, Menlo, monospace; white-space: pre-wrap; word-break: break-word; }
+  .text { padding: 12px; white-space: pre-wrap; word-break: break-word; max-height: 360px; overflow: auto; }
   .broken { padding: 28px 12px; text-align: center; color: var(--muted); font-size: 13px; }
   .tile img { cursor: zoom-in; }
   .tile audio { display: block; width: 100%; padding: 10px; }
@@ -53,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) {} }
@@ -75,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(); }
@@ -113,19 +120,21 @@
     resize();
   }
 
-  function urlOf(o) {
-    if (typeof o === "string") return o;
-    if (o && typeof o === "object") return o.url || o.download_url || o.href || "";
-    return "";
-  }
+  // File types by extension, the same lists the WaveSpeed site uses to pick a preview.
+  var EXT = {
+    image: "jpg jpeg png gif webp avif bmp tiff tif svg ico",
+    video: "mp4 mov avi mkv webm flv wmv m4v",
+    audio: "mp3 wav m4a ogg aac flac wma aiff au mid midi rmi opus",
+    model: "glb gltf",
+    subtitle: "srt vtt",
+  };
   function kindOf(url) {
     var path = url.split("?")[0].split("#")[0].toLowerCase();
     var ext = path.slice(path.lastIndexOf(".") + 1);
-    if (/^(png|jpe?g|webp|gif|avif|bmp)$/.test(ext)) return "image";
-    if (/^(mp4|webm|mov|m4v)$/.test(ext)) return "video";
-    if (/^(mp3|wav|m4a|ogg|flac|aac)$/.test(ext)) return "audio";
+    for (var k in EXT) if ((" " + EXT[k] + " ").indexOf(" " + ext + " ") >= 0) return k;
     return "file";
   }
+  var KIND_LABEL = { model: "3D model", subtitle: "Subtitles", file: "File" };
   function el(tag, cls, text) {
     var n = document.createElement(tag);
     if (cls) n.className = cls;
@@ -138,30 +147,74 @@
     return b;
   }
 
-  function tile(url) {
-    var kind = kindOf(url), t = el("div", "tile"), media;
-    if (!/^https?:\/\//i.test(url)) return null;
+  // Like the site: one thing plays at a time.
+  function onlyOnePlays(media) {
+    media.addEventListener("play", function () {
+      document.querySelectorAll("video, audio").forEach(function (m) { if (m !== media) m.pause(); });
+    });
+  }
+  // A spinner stands in until the media has something to draw (the site does the same).
+  function withSpinner(t, media, readyEvent) {
+    var sp = el("div", "spin tile-spin");
+    t.appendChild(sp);
+    var done = function () { if (sp.parentNode) sp.parentNode.removeChild(sp); };
+    media.addEventListener(readyEvent, done);
+    media.addEventListener("error", done);
+  }
+
+  function brokenNote(t, media, what) {
+    // The host may not let the card load this address; say so and keep the Open button working.
+    media.addEventListener("error", function () {
+      if (media.parentNode !== t) return;
+      t.replaceChild(el("div", "broken", "This chat can't show the " + what + " here. Use Open / save."), media); resize();
+    });
+  }
+
+  // One tile per output. Outputs are links to files, plain text, or JSON.
+  function tile(o) {
+    var t = el("div", "tile");
+    if (o !== null && typeof o === "object") {
+      var text = JSON.stringify(o, null, 2);
+      t.appendChild(el("pre", "code", text));
+      var jm = el("div", "meta"); jm.appendChild(el("span", "", "JSON"));
+      var cb = el("button", "", "Copy");
+      cb.addEventListener("click", function () {
+        try { navigator.clipboard.writeText(text).then(function () { cb.textContent = "Copied"; }, function () { cb.textContent = "Copy failed"; }); }
+        catch (e) { cb.textContent = "Copy failed"; }
+      });
+      jm.appendChild(cb); t.appendChild(jm);
+      return t;
+    }
+    var url = typeof o === "string" ? o : String(o);
+    if (!/^https?:\/\//i.test(url)) { t.appendChild(el("div", "text", url)); return t; }
+    var kind = kindOf(url), media;
     if (kind === "image") {
       media = el("img"); media.src = url; media.alt = "Generated image"; media.loading = "lazy";
       media.addEventListener("click", function () { openLink(url); });
       media.addEventListener("load", resize);
-      // The host may not let the card load this address; say so and keep the Open button working.
-      media.addEventListener("error", function () {
-        var m = el("div", "broken", "This chat can't show the image here. Use Open / save.");
-        t.replaceChild(m, media); resize();
-      });
+      brokenNote(t, media, "image");
       t.appendChild(media);
+      withSpinner(t, media, "load");
     } else if (kind === "video") {
-      media = el("video"); media.src = url; media.controls = true; media.playsInline = true; media.preload = "metadata";
+      // #t=0.001 makes the browser draw the first frame; without it iOS shows a black box until play.
+      media = el("video"); media.src = url + (url.indexOf("#") < 0 ? "#t=0.001" : ""); media.controls = true; media.playsInline = true; media.preload = "metadata"; media.loop = true;
       media.addEventListener("loadedmetadata", resize);
+      onlyOnePlays(media);
+      brokenNote(t, media, "video");
       t.appendChild(media);
+      withSpinner(t, media, "loadeddata");
     } else if (kind === "audio") {
       media = el("audio"); media.src = url; media.controls = true; media.preload = "metadata";
+      onlyOnePlays(media);
+      brokenNote(t, media, "audio");
       t.appendChild(media);
+    } else {
+      // 3D models, subtitles, documents and other links cannot be drawn here: show the link.
+      t.appendChild(el("div", "text", url.split("?")[0].split("/").pop() || url));
     }
-    var meta = el("div", "meta"), name = url.split("?")[0].split("/").pop() || "file";
-    meta.appendChild(el("span", "", kind === "file" ? name : kind));
-    meta.appendChild(openButton(url, kind === "file" ? "Open" : "Open / save"));
+    var meta = el("div", "meta");
+    meta.appendChild(el("span", "", KIND_LABEL[kind] || kind));
+    meta.appendChild(openButton(url, kind === "file" || kind === "model" || kind === "subtitle" ? "Open" : "Open / save"));
     t.appendChild(meta);
     return t;
   }
@@ -191,10 +244,10 @@
       body.appendChild(el("div", "err", String(d.error)));
       setState("failed");
     } else {
-      var urls = (d.outputs || []).map(urlOf).filter(Boolean);
-      if (urls.length) {
-        var grid = el("div", "grid" + (urls.length === 1 ? " one" : ""));
-        urls.forEach(function (u) { var t = tile(u); if (t) grid.appendChild(t); });
+      var outs = (d.outputs || []).filter(function (o) { return o !== null && o !== undefined && o !== ""; });
+      if (outs.length) {
+        var grid = el("div", "grid" + (outs.length === 1 ? " one" : ""));
+        outs.forEach(function (o) { grid.appendChild(tile(o)); });
         body.appendChild(grid);
         setState(d.status === "completed" || d.done ? "done" : (d.status || ""));
       } else if (d.status && !d.done) {
@@ -202,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) {
@@ -223,6 +278,28 @@
     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;
+  }
+
   // 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;
@@ -237,16 +314,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.");
       });
     })();
   }
@@ -262,7 +338,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/server.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',

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@wavespeed/mcp",
-  "version": "1.2.1",
+  "version": "1.2.3",
   "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": {