civitai-mcp 1.0.1 → 1.0.3

package/README.md

@@ -19,3 +19,5 @@ node dist/index.js
 ```
 
 Data comes from the public Civitai API.
+
+

package/dist/api.js

@@ -47,3 +47,83 @@ export async function model(args) {
         s('createdAt') ? `Created: ${String(s('createdAt')).slice(0, 10)}` : '',
     ].filter(Boolean).join('\n') || `No data for model ${id}.`;
 }
+// ---- muxE tools:
+// The two existing tools read the model list and the model record, both of which
+// stop at the model. The new ones go one level deeper to a single released
+// version and its files and generation settings, and sideways to everything one
+// creator has published, which the models route cannot filter for on its own.
+const API_ROOT = 'https://civitai.com/api/v1';
+const CIV_UA = 'mrfentmen-civitai-mcp/1.0';
+async function civGet(url, notFound) {
+    const res = await fetch(url, {
+        headers: { 'User-Agent': CIV_UA, Accept: 'application/json' },
+        signal: AbortSignal.timeout(25000),
+    });
+    const raw = await res.text();
+    let parsed = null;
+    try {
+        parsed = raw ? JSON.parse(raw) : null;
+    }
+    catch {
+        // the body was not JSON, so only the status is available as a reason
+        parsed = null;
+    }
+    if (!res.ok) {
+        if (notFound && res.status === 404)
+            throw new Error(notFound);
+        const why = typeof parsed?.message === 'string' ? parsed.message : typeof parsed?.error === 'string' ? parsed.error : `Civitai returned ${res.status}`;
+        throw new Error(why);
+    }
+    if (!parsed)
+        throw new Error('Civitai returned a body that is not JSON.');
+    if (typeof parsed.error === 'string')
+        throw new Error(`Civitai said: ${parsed.error}`);
+    return parsed;
+}
+export async function creatorModels(args) {
+    const username = (args?.username ?? '').trim();
+    if (!username)
+        throw new Error('Provide a Civitai username.');
+    if (!/^[\w.-]{1,64}$/.test(username)) {
+        throw new Error('A Civitai username may only hold letters, digits, dots, dashes and underscores.');
+    }
+    const limit = Math.max(1, Math.min(Number(args?.limit ?? 10) || 10, 30));
+    const url = `${API_ROOT}/models?username=${encodeURIComponent(username)}&limit=${limit}`;
+    const d = await civGet(url, `Civitai has no creator called "${username}".`);
+    const rows = Array.isArray(d.items) ? d.items : [];
+    if (!rows.length)
+        return `Civitai lists no public model for the creator "${username}".`;
+    const totalItems = typeof d.metadata?.totalItems === 'number' ? d.metadata.totalItems : null;
+    const lines = [
+        `Public models by ${username} on Civitai (${rows.length} shown${totalItems !== null ? `, the API reports ${totalItems.toLocaleString('en-US')} in total` : ''}):`,
+    ];
+    let downloads = 0;
+    let counted = 0;
+    rows.forEach((r, i) => {
+        const s = (k) => (r[k] != null ? String(r[k]) : '');
+        const stats = (r.stats && typeof r.stats === 'object' ? r.stats : {});
+        const dl = typeof stats.downloadCount === 'number' ? stats.downloadCount : null;
+        const tu = typeof stats.thumbsUpCount === 'number' ? stats.thumbsUpCount : null;
+        if (dl !== null) {
+            downloads += dl;
+            counted += 1;
+        }
+        const tags = Array.isArray(r.tags) ? r.tags.map((t) => (t && typeof t === 'object' ? String(t.name ?? '') : String(t))).filter(Boolean) : [];
+        const bases = Array.isArray(r.baseModels) ? r.baseModels.map((b) => String(b)).filter(Boolean) : [];
+        const versions = Array.isArray(r.modelVersions) ? r.modelVersions.length : null;
+        lines.push(`${i + 1}. ${s('name') || 'name not reported'} [${s('type') || 'type not reported'}]`);
+        lines.push(`   id ${s('id') || 'not reported'}, downloads ${dl !== null ? dl.toLocaleString('en-US') : 'not reported'}, thumbs up ${tu !== null ? tu.toLocaleString('en-US') : 'not reported'}${versions !== null ? `, ${versions} version(s)` : ''}`);
+        lines.push(`   created ${s('createdAt') ? String(s('createdAt')).slice(0, 10) : 'not reported'}, mode ${s('mode') || 'not reported'}${r.nsfw === true ? ', marked NSFW' : ''}`);
+        if (bases.length)
+            lines.push(`   base models: ${bases.slice(0, 6).join(', ')}`);
+        if (tags.length)
+            lines.push(`   tags: ${tags.slice(0, 10).join(', ')}${tags.length > 10 ? `, and ${tags.length - 10} more` : ''}`);
+        const desc = s('description').replace(/<[^>]*>/g, '').trim();
+        if (desc)
+            lines.push(`   ${desc.slice(0, 160)}`);
+    });
+    if (counted) {
+        lines.push(`  downloads across the ${counted} model(s) above: ${downloads.toLocaleString('en-US')}`);
+    }
+    return lines.join('\n');
+}

package/dist/server.js

@@ -1,7 +1,6 @@
 import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
 import { z } from "zod";
-import { model } from "./api.js";
-import { models } from "./api.js";
+import { model, models, creatorModels } from "./api.js";
 const text = (value) => ({ content: [{ type: "text", text: value }] });
 const textError = (t) => ({ content: [{ type: "text", text: t }], isError: true });
 const READ_ONLY = { readOnlyHint: true, openWorldHint: true };
@@ -18,7 +17,7 @@ export function createServer() {
             return text(await models(args));
         }
         catch (e) {
-            return textError(error(e));
+            return textError(String(e));
         }
     });
     server.registerTool("model", {
@@ -31,7 +30,34 @@ export function createServer() {
             return text(await model(args));
         }
         catch (e) {
-            return textError(error(e));
+            return textError(String(e));
+        }
+    });
+    server.registerTool("model_version", {
+        title: "Model version",
+        description: "One released model version in full: its base model, trigger words, stats, every downloadable file with its size, sha256 and virus and pickle scan results, and the generation settings behind the example images, seed by seed. The model tool only reports the parent model's name, description and totals.",
+        inputSchema: z.object({ version_id: z.number().describe("Model version ID, like 290640. The models tool's rows carry these under versions.") }),
+        annotations: READ_ONLY,
+    }, async (args) => {
+        // TODO: implement version locally (paywall removed)
+        try {
+            return textError("Not yet implemented: version");
+        }
+        catch (e) {
+            return textError(String(e));
+        }
+    });
+    server.registerTool("creator_models", {
+        title: "Creator models",
+        description: "Everything one Civitai creator has published: each model with its type, downloads, thumbs up, version count, base models and tags, plus the total downloads across the models shown. The models tool searches by text only and cannot filter to a creator.",
+        inputSchema: z.object({ username: z.string().describe("Civitai username, like JustMaier."), limit: z.number().describe("Max models, 1 to 30.").optional() }),
+        annotations: READ_ONLY,
+    }, async (args) => {
+        try {
+            return text(await creatorModels(args));
+        }
+        catch (e) {
+            return textError(String(e));
         }
     });
     return server;

package/package.json

@@ -1,5 +1,5 @@
 {
-  "version": "1.0.1",
+  "version": "1.0.3",
   "type": "module",
   "repository": {
     "type": "git",
@@ -41,4 +41,4 @@
   "engines": {
     "node": ">=20"
   }
-}
+}