modelscope-image-mcp 1.0.1 → 1.0.2

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "modelscope-image-mcp",
-  "version": "1.0.1",
+  "version": "1.0.2",
   "description": "ModelScope (魔搭) 文生图 / 图生图 MCP 服务：免费调用 Qwen-Image、Z-Image、FLUX.1 等模型",
   "type": "module",
   "main": "server.js",

package/server.js

@@ -125,7 +125,17 @@ async function waitTask(taskId, timeoutMs = TASK_TIMEOUT_MS) {
   throw new Error(`任务超时（${Math.round(timeoutMs / 1000)}s），可调大 MODELSCOPE_TIMEOUT_MS`);
 }
 
-async function downloadImage(url, dir) {
+/** 把模型 ID 压成适合做文件名的短标识：black-forest-labs/FLUX.1-Krea-dev -> flux-1-krea-dev */
+function modelSlug(model) {
+  return String(model)
+    .split("/")
+    .pop()
+    .toLowerCase()
+    .replace(/[^a-z0-9]+/g, "-")
+    .replace(/^-+|-+$/g, "");
+}
+
+async function downloadImage(url, dir, tag = "") {
   const res = await fetch(url);
   if (!res.ok) throw new Error(`下载图片失败：HTTP ${res.status} ${url}`);
 
@@ -140,7 +150,8 @@ async function downloadImage(url, dir) {
     /* 忽略，用默认后缀 */
   }
 
-  const file = path.join(dir, `ms-${timestamp()}-${Math.floor(Math.random() * 1e4)}${ext}`);
+  const prefix = tag ? `ms-${tag}` : "ms";
+  const file = path.join(dir, `${prefix}-${timestamp()}-${Math.floor(Math.random() * 1e4)}${ext}`);
   await fs.promises.writeFile(file, buf);
   return file;
 }
@@ -177,9 +188,10 @@ async function runGeneration({ model, prompt, negativePrompt, size, seed, image,
   const urls = (Array.isArray(raw) ? raw : [raw]).filter(Boolean);
 
   const dir = outputDir || OUTPUT_DIR;
+  const tag = modelSlug(model);
   const files = [];
   for (const url of urls) {
-    files.push(await downloadImage(url, dir));
+    files.push(await downloadImage(url, dir, tag));
   }
 
   return { taskId, files, urls, model };
@@ -330,7 +342,7 @@ async function handleMessage(msg) {
       reply(id, {
         protocolVersion: params?.protocolVersion || "2024-11-05",
         capabilities: { tools: {} },
-        serverInfo: { name: "modelscope-image-mcp", version: "1.0.1" },
+        serverInfo: { name: "modelscope-image-mcp", version: "1.0.2" },
       });
       return;