@coconutmilk/qwen-image-edit-mcp-server 1.0.0

package/index.js

@@ -0,0 +1,157 @@
+#!/usr/bin/env node
+
+import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
+import { z } from "zod";
+
+// ========================================
+// 1. 创建MCP Server
+// ========================================
+const server = new McpServer({
+  name: "qwen-image-edit",
+  version: "1.0.0"
+});
+
+// ========================================
+// 2. 提交异步任务
+// ========================================
+async function submitTask(images, text, options) {
+  const content = [];
+  for (const imageUrl of images) {
+    content.push({ image: imageUrl });
+  }
+  content.push({ text: text });
+
+  const response = await fetch(
+    "https://dashscope.aliyuncs.com/api/v1/services/aigc/multimodal-generation/generation",
+    {
+      method: "POST",
+      headers: {
+        "Content-Type": "application/json",
+        "Authorization": `Bearer ${process.env.DASHSCOPE_API_KEY}`,
+        "X-DashScope-Async": "enable"
+      },
+      body: JSON.stringify({
+        model: "qwen-image-edit-plus-2025-12-15",
+        input: {
+          messages: [{ role: "user", content: content }]
+        },
+        parameters: {
+          n: options.n || 1,
+          size: options.size || "1024*1024",
+          prompt_extend: options.prompt_extend !== false,
+          watermark: options.watermark || false,
+          negative_prompt: options.negative_prompt || " "
+        }
+      })
+    }
+  );
+
+  const result = await response.json();
+  if (result.code) {
+    throw new Error(`提交任务失败: ${result.code} - ${result.message}`);
+  }
+  return result.output.task_id;
+}
+
+// ========================================
+// 3. 轮询任务结果
+// ========================================
+async function pollResult(taskId) {
+  const maxRetries = 60;
+  const interval = 3000;
+
+  for (let i = 0; i < maxRetries; i++) {
+    const response = await fetch(
+      `https://dashscope.aliyuncs.com/api/v1/tasks/${taskId}`,
+      {
+        method: "GET",
+        headers: {
+          "Authorization": `Bearer ${process.env.DASHSCOPE_API_KEY}`
+        }
+      }
+    );
+
+    const result = await response.json();
+    const status = result.output.task_status;
+
+    if (status === "SUCCEEDED") {
+      return result.output.results;
+    } else if (status === "FAILED") {
+      throw new Error(`任务失败: ${result.output.message || "未知错误"}`);
+    }
+
+    await new Promise(resolve => setTimeout(resolve, interval));
+  }
+
+  throw new Error("任务超时，超过3分钟未完成");
+}
+
+// ========================================
+// 4. 注册工具（新版 registerTool 写法）
+// ========================================
+server.registerTool(
+  // 第一个参数：工具名
+  "edit_image",
+  // 第二个参数：config对象
+  {
+    title: "图片编辑工具",
+    description: "使用千问图像编辑模型编辑图片。支持单图编辑和多图融合，可修改文字、增删物体、改变动作、迁移风格等。",
+    inputSchema: {
+      images: z.array(z.string()).min(1).max(6)
+        .describe("图片URL数组，1-6张图片"),
+      text: z.string()
+        .describe("编辑指令，描述你想怎么编辑图片，比如'把背景换成海滩'"),
+      n: z.number().min(1).max(6).optional()
+        .describe("生成图片数量，默认1"),
+      size: z.string().optional()
+        .describe("输出图片尺寸，如'1024*1024'、'1024*1536'，默认1024*1024"),
+      negative_prompt: z.string().optional()
+        .describe("负向提示词，描述不想出现的内容"),
+      prompt_extend: z.boolean().optional()
+        .describe("是否启用提示词智能改写，默认true"),
+      watermark: z.boolean().optional()
+        .describe("是否添加水印，默认false")
+    },
+    annotations: {
+      title: "图片编辑工具",
+      readOnlyHint: false,
+      destructiveHint: false,
+      openWorldHint: true
+    }
+  },
+  // 第三个参数：handler函数
+  async ({ images, text, n, size, negative_prompt, prompt_extend, watermark }) => {
+    try {
+      const taskId = await submitTask(images, text, {
+        n, size, negative_prompt, prompt_extend, watermark
+      });
+      const results = await pollResult(taskId);
+
+      const content = [];
+      content.push({
+        type: "text",
+        text: `图片编辑完成，共生成 ${results.length} 张图片：`
+      });
+      for (let i = 0; i < results.length; i++) {
+        content.push({
+          type: "text",
+          text: `图片${i + 1}: ${results[i].url}`
+        });
+      }
+      return { content };
+
+    } catch (error) {
+      return {
+        content: [{ type: "text", text: `编辑失败: ${error.message}` }],
+        isError: true
+      };
+    }
+  }
+);
+
+// ========================================
+// 5. 启动
+// ========================================
+const transport = new StdioServerTransport();
+await server.connect(transport);

package/package.json

@@ -0,0 +1,12 @@
+{
+  "name": "@coconutmilk/qwen-image-edit-mcp-server",
+  "version": "1.0.0",
+  "type": "module",
+  "bin": {
+    "qwen-image-edit-mcp-server": "./index.js"
+  },
+  "dependencies": {
+    "@modelcontextprotocol/sdk": "^1.0.0",
+    "zod": "^3.23.0"
+  }
+}