@wavespeed/mcp 1.0.1 → 1.0.3

package/README.md

@@ -1,3 +1,12 @@
+<div align="center">
+  <a href="https://wavespeed.ai" target="_blank">
+    <picture>
+      <source media="(prefers-color-scheme: dark)" srcset="assets/wavespeed-logo-dark.svg">
+      <img src="assets/wavespeed-logo-light.svg" alt="WaveSpeed" width="342" height="48"/>
+    </picture>
+  </a>
+</div>
+
 # WaveSpeed MCP Server
 
 [中文](README.zh.md)

package/README.zh.md

@@ -1,3 +1,12 @@
+<div align="center">
+  <a href="https://wavespeed.ai" target="_blank">
+    <picture>
+      <source media="(prefers-color-scheme: dark)" srcset="assets/wavespeed-logo-dark.svg">
+      <img src="assets/wavespeed-logo-light.svg" alt="WaveSpeed" width="342" height="48"/>
+    </picture>
+  </a>
+</div>
+
 # WaveSpeed MCP Server
 
 [English](README.md)

package/dist/server.js

@@ -11,6 +11,7 @@
 //   hosted URL. Bare paths pass through untouched.
 // * Price quotes name the inputs they were blind to instead of presenting
 //   the formula's floor as "the" price.
+import { createRequire } from 'node:module';
 import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
 import { z } from 'zod';
 import { Client } from 'wavespeed';
@@ -19,6 +20,8 @@ import { getApiKey, getBaseUrl } from './lib/config.js';
 import { resolveLocalFiles } from './lib/local-files.js';
 import { uploadWithCache } from './lib/upload-cache.js';
 import { missingPriceVars, isFloorQuote } from './lib/pricing-vars.js';
+const require = createRequire(import.meta.url);
+const { version: PKG_VERSION } = require('../package.json');
 const PRICE_DISCLAIMER = 'Estimate only, for reference — the amount actually charged for a run is authoritative.';
 function requestSchema(m) {
     return m.api_schema?.api_schemas?.[0]?.request_schema;
@@ -43,7 +46,7 @@ function ok(payload) {
     return { content: [{ type: 'text', text: JSON.stringify(payload, null, 2) }] };
 }
 export function createServer() {
-    const server = new McpServer({ name: 'wavespeed', version: '1.0.0' }, {
+    const server = new McpServer({ name: 'wavespeed', version: PKG_VERSION }, {
         instructions: [
             'WaveSpeed AI media generation (image, video, audio, 3D).',
             'Pattern: list_models to find a model, get_model_schema to see its',
@@ -61,7 +64,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'),
-    }, async ({ query, type, limit, refresh }) => {
+    }, { readOnlyHint: true, openWorldHint: true }, async ({ query, type, limit, refresh }) => {
         const models = await fetchModels({ refresh });
         const q = query?.toLowerCase();
         const filtered = models.filter((m) => {
@@ -78,7 +81,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. google/nano-banana-2/text-to-image') }, 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. google/nano-banana-2/text-to-image') }, { readOnlyHint: true, openWorldHint: true }, async ({ model }) => {
         const models = await fetchModels();
         const meta = models.find((m) => m.model_id === model);
         if (!meta) {
@@ -106,7 +109,7 @@ export function createServer() {
             .max(1800)
             .default(600)
             .describe('Max seconds to wait; 0 = submit only and return the prediction id'),
-    }, async ({ model, input, wait_seconds }) => {
+    }, { readOnlyHint: false, destructiveHint: false, idempotentHint: false, openWorldHint: true }, async ({ model, input, wait_seconds }) => {
         const resolved = await resolveLocalFiles(input, {
             upload: async (p) => (await uploadFile(p)).url,
         });
@@ -131,8 +134,16 @@ export function createServer() {
     server.tool('get_price', 'Estimate the cost of a run before executing it (no charge). Provide the same input you would pass to run_model — pricing often depends on inputs like duration or resolution.', {
         model: z.string().describe('Model ID from list_models'),
         input: z.record(z.unknown()).default({}).describe('Inputs the quote should account for'),
-    }, async ({ model, input }) => {
-        const data = await fetchPricing(model, input);
+    }, { readOnlyHint: false, destructiveHint: false, openWorldHint: true }, async ({ model, input }) => {
+        // Resolve @path markers exactly like run_model: pricing formulas often
+        // read the file itself (get_duration_v3(audio)), so quoting with the
+        // literal "@./a.mp3" string would silently collapse to the floor. The
+        // 24h content-hash cache means the later run reuses this upload.
+        const resolved = await resolveLocalFiles(input, {
+            upload: async (p) => (await uploadFile(p)).url,
+        });
+        const priceInput = resolved.input;
+        const data = await fetchPricing(model, priceInput);
         // The pricing endpoint accepts partial inputs without complaint — a
         // formula whose variables are all missing collapses to base_price, the
         // floor of the model's range. Name what the quote could not see.
@@ -141,8 +152,8 @@ export function createServer() {
         try {
             const models = await fetchModels();
             const meta = models.find((m) => m.model_id === model);
-            unpriced = missingPriceVars(meta?.formula, input, requestSchema(meta ?? {})?.properties);
-            atFloor = isFloorQuote(meta?.formula, input);
+            unpriced = missingPriceVars(meta?.formula, priceInput, requestSchema(meta ?? {})?.properties);
+            atFloor = isFloorQuote(meta?.formula, priceInput);
         }
         catch {
             /* no catalog — the generic disclaimer still applies */
@@ -155,13 +166,13 @@ export function createServer() {
             disclaimer: PRICE_DISCLAIMER,
         });
     });
-    server.tool('get_balance', 'Show the WaveSpeed account credit balance.', {}, async () => {
+    server.tool('get_balance', 'Show the WaveSpeed account credit balance.', {}, { readOnlyHint: true, openWorldHint: true }, async () => {
         return ok(await fetchBalance());
     });
-    server.tool('upload_file', 'Upload a local file to WaveSpeed and get its hosted URL (identical bytes reuse the same upload for 24h). Usually unnecessary — run_model handles "@./path" inputs itself.', { path: z.string().describe('Local file path') }, async ({ path: filePath }) => {
+    server.tool('upload_file', 'Upload a local file to WaveSpeed and get its hosted URL (identical bytes reuse the same upload for 24h). Usually unnecessary — run_model handles "@./path" inputs itself.', { path: z.string().describe('Local file path') }, { readOnlyHint: false, destructiveHint: false, idempotentHint: true, openWorldHint: true }, async ({ path: filePath }) => {
         return ok(await uploadFile(filePath));
     });
-    server.tool('get_prediction', 'Fetch the status and outputs of a past or in-flight prediction by id — use to recover a run that hit the wait limit.', { id: z.string().describe('Prediction id returned by run_model') }, async ({ id }) => {
+    server.tool('get_prediction', 'Fetch the status and outputs of a past or in-flight prediction by id — use to recover a run that hit the wait limit.', { id: z.string().describe('Prediction id returned by run_model') }, { readOnlyHint: true, openWorldHint: true }, async ({ id }) => {
         const item = await fetchPrediction(id);
         return ok({
             id: item.id,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@wavespeed/mcp",
-  "version": "1.0.1",
+  "version": "1.0.3",
   "description": "WaveSpeed MCP server \u2014 run any model on the live WaveSpeed catalog (image, video, audio, 3D) with schema introspection, local-file upload, and price quotes.",
   "type": "module",
   "bin": {
@@ -47,5 +47,6 @@
     "tsx": "^4.19.0",
     "typescript": "^5.7.0",
     "vitest": "^3.0.0"
-  }
+  },
+  "mcpName": "ai.wavespeed/mcp"
 }