@wavespeed/mcp 1.0.14 → 1.0.16

package/README.md

@@ -25,16 +25,16 @@ Built on the same design as the open-source [`@wavespeed/cli`](https://github.co
 - **No hardcoded models.** The catalog comes from the live API (1h cache); new platform models work the day they ship.
 - **One generation verb.** `run_model` executes anything; `get_model_schema` tells the agent what inputs a model actually accepts, so it reads before it writes.
 - **Inputs are never mutated.** The one explicit transform is the `@path` marker — a `"@./photo.jpg"` value inside `input` uploads the file and substitutes its hosted URL. Bare paths are passed through untouched.
-- **Honest pricing.** `get_price` quotes before you spend and names the inputs the quote was blind to (`unpriced_inputs`) instead of presenting a formula's floor as "the" price.
+- **Quote before you spend.** `get_price` takes the same input as `run_model` and returns an estimate; the amount actually charged for a run is authoritative.
 
 ## Tools
 
 | Tool | What it does |
 |---|---|
 | `list_models` | Search the live catalog by text or modality type |
-| `get_model_schema` | A model's real input schema (required, properties, defaults) |
+| `get_model_schema` | A model's real input schema (required, properties, defaults), with a link to the model's docs page |
 | `run_model` | Submit + wait; rejects input keys the model doesn't accept; `@path` inputs upload automatically; returns output URLs |
-| `get_price` | Cost estimate with `unpriced_inputs` / `at_base_price` disclosure |
+| `get_price` | Cost estimate for the inputs you pass (no charge), with a link to the model's docs page |
 | `get_balance` | Account credit balance, with the top-up link |
 | `list_predictions` | Run history (default last 24h); filter by model, status, time range |
 | `list_billings` | Itemized charges and refunds (needs a Billing-access key) |

package/dist/lib/api.js

@@ -200,6 +200,13 @@ export async function fetchUsageStats(startTime, endTime, model) {
         body.model_uuid = model;
     return apiPost('/api/v3/user/usage_stats', body, { timeoutMs: GET_TIMEOUT_MS });
 }
+// The model's page on the docs site (same rule as the dashboard's model page
+// link): provider folder, then the id without a leading "wavespeed-ai/" and
+// with "/" turned into "-".
+export function modelDocsUrl(modelId) {
+    const provider = modelId.split('/')[0];
+    return `https://wavespeed.ai/docs/docs-api/${provider}/${modelId.replace('wavespeed-ai/', '').split('/').join('-')}`;
+}
 export async function fetchPricing(modelId, inputs) {
     // /model/price is the documented pricing endpoint.
     return apiPost('/api/v3/model/price', { model_id: modelId, inputs });

package/dist/server.js

@@ -9,20 +9,19 @@
 // * Inputs are never mutated. The one explicit transform is the `@path`
 //   marker, which uploads the referenced local file and substitutes its
 //   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.
+// * Price quotes are estimates for the inputs given; the model's docs page
+//   is linked next to them for what each input means.
 import { createRequire } from 'node:module';
 import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
 import { z } from 'zod';
 import { Client } from 'wavespeed';
-import { fetchModels, fetchBalance, fetchHistory, fetchBillings, fetchUsageStats, deletePredictions, PREDICTION_STATUSES, fetchPricing, submitPrediction, waitForPrediction, MCP_CLIENT_NAME, TERMINAL_FAILURES, } from './lib/api.js';
+import { fetchModels, fetchBalance, fetchHistory, fetchBillings, fetchUsageStats, deletePredictions, PREDICTION_STATUSES, fetchPricing, modelDocsUrl, submitPrediction, waitForPrediction, MCP_CLIENT_NAME, TERMINAL_FAILURES, } from './lib/api.js';
 import { getApiKey, getBaseUrl } from './lib/config.js';
 import { MAX_DELETE_IDS, prepareDelete, summarizeDelete } from './lib/delete-predictions.js';
 // 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 { uploadWithCache } from './lib/upload-cache.js';
-import { missingPriceVars, isFloorQuote } from './lib/pricing-vars.js';
 import { findUnknownInputs } from './lib/validate-inputs.js';
 const require = createRequire(import.meta.url);
 const { version: PKG_VERSION } = require('../package.json');
@@ -164,6 +163,7 @@ export function createServer() {
             required: schema?.required ?? [],
             properties: schema?.properties ?? {},
             property_order: schema?.['x-order-properties'] ?? [],
+            docs_url: modelDocsUrl(meta.model_id),
         });
     });
     server.tool('run_model', 'Run any WaveSpeed model. input keys come from get_model_schema; unknown keys are rejected before submission. Local files: pass "@./path" string values — they are uploaded and replaced with hosted URLs (bare paths are NOT uploaded). Returns the prediction id right away (done=false); call get_prediction with that id and wait_seconds until done=true to get the output URLs. The task keeps running on the server. Pass wait_seconds here to wait for a quick model in the same call.', {
@@ -208,40 +208,20 @@ export function createServer() {
             ...(done ? {} : { next: STILL_RUNNING_HINT }),
         });
     });
-    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.', {
+    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, and a quote made without them can be lower than the real charge.', {
         model: z.string().describe('Model ID from list_models'),
         input: z.record(z.unknown()).default({}).describe('Inputs the quote should account for'),
     }, { 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.
+        // Resolve @path markers exactly like run_model: some prices are measured
+        // from the file itself, so quoting with the literal "@./a.mp3" string
+        // would be wrong. 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.
-        let unpriced = [];
-        let atFloor = false;
-        try {
-            const models = await fetchModels();
-            const meta = models.find((m) => m.model_id === model);
-            unpriced = missingPriceVars(meta?.formula, priceInput, requestSchema(meta ?? {})?.properties);
-            atFloor = isFloorQuote(meta?.formula, priceInput);
-        }
-        catch {
-            /* no catalog — the generic disclaimer still applies */
-        }
-        return ok({
-            ...data,
-            estimate: true,
-            unpriced_inputs: unpriced,
-            at_base_price: atFloor,
-            disclaimer: PRICE_DISCLAIMER,
-        });
+        return ok({ ...data, estimate: true, disclaimer: PRICE_DISCLAIMER, docs_url: modelDocsUrl(model) });
     });
     server.tool('get_balance', 'Show the WaveSpeed account credit balance, with the link to top up.', {}, { readOnlyHint: true, openWorldHint: true }, async () => {
         return ok({ ...(await fetchBalance()), top_up_url: TOP_UP_URL });

package/package.json

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

package/dist/lib/pricing-vars.js

@@ -1,75 +0,0 @@
-// Pricing formulas are JSONata expressions returned by /api/v3/models, e.g.
-//   {"total_price": $max([5, $min([$ceil(get_duration_v3(audio)), 600])]) / 5 * base_price}
-//   {"total_price": base_price * duration / 5}
-//
-// The pricing endpoint evaluates them server-side and happily accepts a partial
-// (or empty) `inputs` object — a formula whose variables are all missing
-// collapses to `base_price`, which is the floor of the model's price range, not
-// a typical charge. `wavespeed price wavespeed-ai/infinitetalk` with no inputs
-// quotes $0.15 while a 60s audio actually costs $1.80. So we parse the formula
-// to find which inputs move the price, and say which ones the quote is blind to.
-const RESERVED = new Set(['base_price', 'total_price', 'true', 'false', 'null']);
-/**
- * Variable names a pricing formula reads from the model's inputs, in first-seen
- * order. Excludes JSONata builtins (`$max`), helper calls (`get_duration_v3`),
- * string literals, and `base_price`/`total_price`.
- *
- * Returns [] for a flat-priced model (`base_price * 1`) or an unparseable
- * formula — callers should treat [] as "nothing input-dependent to warn about".
- */
-export function extractPriceVars(formula) {
-    if (!formula)
-        return [];
-    // Drop string literals first so "720p" / "total_price" keys never register.
-    const stripped = formula.replace(/"(?:[^"\\]|\\.)*"/g, '""').replace(/'(?:[^'\\]|\\.)*'/g, "''");
-    const vars = [];
-    const seen = new Set();
-    const re = /(\$?)([A-Za-z_][A-Za-z0-9_]*)\s*(\()?/g;
-    let m;
-    while ((m = re.exec(stripped)) !== null) {
-        const [, dollar, name, call] = m;
-        if (dollar)
-            continue; // $max, $ceil, $length — JSONata builtin
-        if (call)
-            continue; // get_duration_v3(...) — server-side helper
-        if (RESERVED.has(name))
-            continue;
-        if (seen.has(name))
-            continue;
-        seen.add(name);
-        vars.push(name);
-    }
-    return vars;
-}
-/**
- * Price-driving variables the quote is genuinely blind to.
- *
- * A driver the caller omitted but that carries a schema default is NOT blind —
- * the server priced it at that default, which is exactly what a run would use.
- * Only drivers with no default (infinitetalk's `audio`, seedance's
- * `reference_videos`) leave the quote uninformed, and those are the ones that
- * silently collapse the formula toward base_price.
- */
-export function missingPriceVars(formula, inputs, schemaProps) {
-    return extractPriceVars(formula).filter((v) => {
-        const val = inputs[v];
-        if (val !== undefined && val !== null && val !== '')
-            return false;
-        const prop = schemaProps?.[v];
-        return !(prop && 'default' in prop);
-    });
-}
-/**
- * True when every price driver is unsupplied — the case where the formula
- * degenerates to `base_price` and the quote is the floor of the model's range
- * rather than a representative charge.
- */
-export function isFloorQuote(formula, inputs) {
-    const vars = extractPriceVars(formula);
-    if (vars.length === 0)
-        return false;
-    return vars.every((v) => {
-        const val = inputs[v];
-        return val === undefined || val === null || val === '';
-    });
-}