mcp-duojie-image 0.5.2 → 0.5.4

package/dist/index.js

@@ -23,7 +23,7 @@ const server = new McpServer({
     version: "0.1.0",
 });
 server.registerTool("generate_image", {
-    description: "Generate image async. Ask user to choose model before calling.",
+    description: "Generate image async. If user didn't specify model and no model was chosen in this conversation, ask once. Reuse that model later unless user changes it. Return after tool call; don't wait/check.",
     inputSchema: z.object({
         prompt: z.string().min(1).describe("Prompt"),
         output_dir: z.string().describe("Absolute output dir"),
@@ -32,7 +32,7 @@ server.registerTool("generate_image", {
         size: z.enum(["512px", "1K", "2K", "4K"]).optional().describe("Default 1K"),
         aspect_ratio: z.string().optional().describe("e.g. 16:9"),
         format: z.string().optional().describe("Default png"),
-        model: z.enum(["gemini-3.1-flash-image", "gpt-image-2"]).describe("Required"),
+        model: z.enum(["gemini-3.1-flash-image", "gpt-image-2"]).describe("Required; ask once per conversation if unknown"),
         overwrite: z.boolean().optional().describe("Default false"),
         negative_prompt: z.string().optional().describe("Avoid"),
         silent: z.boolean().optional().describe("No notification"),
@@ -92,8 +92,9 @@ server.registerTool("generate_image", {
         content: [{
                 type: "text",
                 text: JSON.stringify({
-                    status: "pending",
-                    message: `已开始：${args.output_filename}`,
+                    success: true,
+                    status: "submitted",
+                    message: `已提交异步生成：${args.output_filename}；无需等待或检查结果`,
                 }),
             }],
     };

package/dist/provider.js

@@ -46,18 +46,43 @@ function mapSize(size) {
 function isOpenAIImageModel(model) {
     return model === "gpt-image-2";
 }
+function parseAspectRatio(aspectRatio) {
+    const match = aspectRatio?.match(/^(\d+):(\d+)$/);
+    if (!match)
+        return [1, 1];
+    return [Number(match[1]), Number(match[2])];
+}
+function mapOpenAI4KSize(aspectRatio) {
+    const [ratioWidth, ratioHeight] = parseAspectRatio(aspectRatio);
+    const maxPixels = 3840 * 2160;
+    const maxEdge = 3840;
+    let width = Math.floor(Math.sqrt((maxPixels * ratioWidth) / ratioHeight));
+    let height = Math.floor((width * ratioHeight) / ratioWidth);
+    if (width > maxEdge) {
+        width = maxEdge;
+        height = Math.floor((width * ratioHeight) / ratioWidth);
+    }
+    if (height > maxEdge) {
+        height = maxEdge;
+        width = Math.floor((height * ratioWidth) / ratioHeight);
+    }
+    width = Math.floor(width / 8) * 8;
+    height = Math.floor(height / 8) * 8;
+    return `${width}x${height}`;
+}
 function mapOpenAISize(size, aspectRatio) {
     const squareBySize = {
         "512px": "512x512",
         "1K": "1024x1024",
         "2K": "2048x2048",
-        "4K": "4096x4096",
     };
+    if (size === "4K")
+        return mapOpenAI4KSize(aspectRatio);
     if (aspectRatio === "16:9" || aspectRatio === "3:2" || aspectRatio === "4:3") {
-        return size === "2K" ? "2048x1536" : size === "4K" ? "4096x3072" : "1536x1024";
+        return size === "2K" ? "2048x1536" : "1536x1024";
     }
     if (aspectRatio === "9:16" || aspectRatio === "2:3" || aspectRatio === "3:4") {
-        return size === "2K" ? "1536x2048" : size === "4K" ? "3072x4096" : "1024x1536";
+        return size === "2K" ? "1536x2048" : "1024x1536";
     }
     return squareBySize[size] ?? "1024x1024";
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "mcp-duojie-image",
-  "version": "0.5.2",
+  "version": "0.5.4",
   "description": "MCP server for image generation via duojie.games",
   "type": "module",
   "main": "dist/index.js",