yzsd-image 1.0.2 → 1.0.4

package/README.md

@@ -22,7 +22,7 @@ YZSD AI 图片生成 MCP 服务器。通过 Model Context Protocol 使用 AI 生
     "yzsd-image": {
       "type": "stdio",
       "command": "npx",
-      "args": ["-y", "yzsd-image"],
+      "args": ["-y", "yzsd-image@latest"],
       "env": {
         "YZSD_API_KEY": "your-api-key-here"
       }
@@ -70,7 +70,7 @@ npm install -g yzsd-image
     "yzsd-image": {
       "type": "stdio",
       "command": "npx",
-      "args": ["-y", "yzsd-image"],
+      "args": ["-y", "yzsd-image@latest"],
       "env": {
         "YZSD_API_KEY": "your-api-key-here"
       }
@@ -88,7 +88,7 @@ npm install -g yzsd-image
   "yzsd-image": {
     "type": "stdio",
     "command": "npx",
-    "args": ["-y", "yzsd-image"],
+    "args": ["-y", "yzsd-image@latest"],
     "env": {
       "YZSD_API_KEY": "your-api-key-here"
     }

package/dist/index.js

@@ -6,6 +6,7 @@ import axios from "axios";
 import * as fs from "fs/promises";
 import * as path from "path";
 const API_BASE_URL = "https://token.1001xr.asia:8443";
+const MODEL_PATH = "/v1beta/models/gemini-3-pro-image-preview:generateContent";
 const VALID_ASPECT_RATIOS = [
     "1:1", "16:9", "9:16", "4:3", "3:4",
     "3:2", "2:3", "21:9", "5:4", "4:5"
@@ -139,18 +140,24 @@ class YZSDImageServer {
         if (!VALID_SIZES.includes(size)) {
             throw new Error(`Invalid size. Must be one of: ${VALID_SIZES.join(", ")}`);
         }
-        const response = await axios.post(`${API_BASE_URL}/v1/images/generate`, {
-            prompt,
-            aspect_ratio: aspectRatio,
-            size,
-        }, {
+        const payload = {
+            contents: [{ parts: [{ text: prompt }] }],
+            generationConfig: {
+                responseModalities: ["IMAGE"],
+                imageConfig: {
+                    aspectRatio: aspectRatio,
+                    image_size: size,
+                },
+            },
+        };
+        const response = await axios.post(`${API_BASE_URL}${MODEL_PATH}`, payload, {
             headers: {
                 "Authorization": `Bearer ${key}`,
                 "Content-Type": "application/json",
             },
             timeout: size === "4K" ? 120000 : size === "2K" ? 60000 : 30000,
         });
-        const imageData = response.data.image || response.data.data?.[0]?.b64_json;
+        const imageData = response.data.candidates?.[0]?.content?.parts?.[0]?.inlineData?.data;
         if (!imageData) {
             throw new Error("No image data in response");
         }
@@ -175,18 +182,34 @@ class YZSDImageServer {
         }
         const imageBuffer = await fs.readFile(inputPath);
         const base64Image = imageBuffer.toString("base64");
-        const response = await axios.post(`${API_BASE_URL}/v1/images/edit`, {
-            image: base64Image,
-            prompt,
-            aspect_ratio: aspectRatio,
-        }, {
+        const mimeType = inputPath.toLowerCase().endsWith('.png') ? 'image/png' : 'image/jpeg';
+        const payload = {
+            contents: [{
+                    parts: [
+                        {
+                            inlineData: {
+                                mimeType: mimeType,
+                                data: base64Image,
+                            },
+                        },
+                        { text: prompt },
+                    ],
+                }],
+            generationConfig: {
+                responseModalities: ["IMAGE"],
+                imageConfig: {
+                    aspectRatio: aspectRatio,
+                },
+            },
+        };
+        const response = await axios.post(`${API_BASE_URL}${MODEL_PATH}`, payload, {
             headers: {
                 "Authorization": `Bearer ${key}`,
                 "Content-Type": "application/json",
             },
             timeout: 120000,
         });
-        const imageData = response.data.image || response.data.data?.[0]?.b64_json;
+        const imageData = response.data.candidates?.[0]?.content?.parts?.[0]?.inlineData?.data;
         if (!imageData) {
             throw new Error("No image data in response");
         }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "yzsd-image",
-  "version": "1.0.2",
+  "version": "1.0.4",
   "description": "YZSD AI 图片生成 MCP 服务器 - 支持文生图和图生图编辑",
   "type": "module",
   "main": "dist/index.js",