mcp-duojie-image 0.4.0 → 0.5.0

package/dist/config.js

@@ -12,7 +12,7 @@ export function loadConfig(overrides) {
     return {
         apiKey,
         baseUrl: overrides?.baseUrl ?? process.env.DUOJIE_BASE_URL ?? "https://api.duojie.games",
-        model: overrides?.model ?? process.env.DUOJIE_IMAGE_MODEL ?? "gemini-3-pro-image-preview",
+        model: overrides?.model ?? process.env.DUOJIE_IMAGE_MODEL ?? "gemini-3.1-flash-image",
         timeoutMs: 300_000,
     };
 }

package/dist/provider.js

@@ -24,35 +24,46 @@ async function fetchWithRetry(url, init, timeoutMs, retries = 0) {
     }
     throw new AppError("E_PROVIDER_BAD_REQUEST", `Network error: ${lastErr}`);
 }
-async function buildMessages(opts) {
+async function buildParts(opts) {
     const { prompt, referenceImages, negativePrompt } = opts;
     const fullPrompt = negativePrompt ? `${prompt}\n\nAvoid: ${negativePrompt}` : prompt;
-    if (!referenceImages || referenceImages.length === 0) {
-        return [{ role: "user", content: fullPrompt }];
+    const parts = [];
+    if (referenceImages && referenceImages.length > 0) {
+        const imageParts = await Promise.all(referenceImages.map(async (imgPath) => {
+            const data = await resizeIfNeeded(imgPath);
+            return { inlineData: { mimeType: "image/png", data: data.toString("base64") } };
+        }));
+        parts.push(...imageParts);
     }
-    const parts = await Promise.all(referenceImages.map(async (imgPath) => {
-        const data = await resizeIfNeeded(imgPath);
-        const b64 = data.toString("base64");
-        return { type: "image_url", image_url: { url: `data:image/png;base64,${b64}` } };
-    }));
-    parts.push({ type: "text", text: fullPrompt });
-    return [{ role: "user", content: parts }];
+    parts.push({ text: fullPrompt });
+    return parts;
+}
+function mapSize(size) {
+    if (size === "512px")
+        return "512";
+    return size; // "1K", "2K", "4K" are used as-is
 }
 export async function generateImage(config, opts) {
-    const messages = await buildMessages(opts);
+    const parts = await buildParts(opts);
     const model = opts.model ?? config.model;
     const imageConfig = {
-        image_size: opts.size ?? "1K",
+        imageSize: mapSize(opts.size ?? "1K"),
     };
     if (opts.aspectRatio) {
-        imageConfig.aspect_ratio = opts.aspectRatio;
+        imageConfig.aspectRatio = opts.aspectRatio;
     }
-    const body = JSON.stringify({ model, messages, image_config: imageConfig });
-    const url = `${config.baseUrl}/v1/chat/completions`;
+    const body = JSON.stringify({
+        contents: [{ role: "user", parts }],
+        generationConfig: {
+            responseModalities: ["IMAGE"],
+            imageConfig,
+        },
+    });
+    const url = `${config.baseUrl}/v1beta/models/${model}:generateContent`;
     const res = await fetchWithRetry(url, {
         method: "POST",
         headers: {
-            "Authorization": `Bearer ${config.apiKey}`,
+            "x-goog-api-key": config.apiKey,
             "Content-Type": "application/json",
         },
         body,
@@ -68,14 +79,9 @@ export async function generateImage(config, opts) {
         throw new AppError("E_PROVIDER_BAD_REQUEST", `Upstream error ${res.status}: ${text.slice(0, 200)}`);
     }
     const data = (await res.json());
-    const imageUrl = data.choices?.[0]?.message?.images?.[0]?.image_url?.url;
-    if (!imageUrl) {
+    const imagePart = data.candidates?.[0]?.content?.parts?.find((p) => p.inlineData);
+    if (!imagePart?.inlineData?.data) {
         throw new AppError("E_PROVIDER_BAD_REQUEST", "No image in response");
     }
-    // data:image/png;base64,<data>
-    const comma = imageUrl.indexOf(",");
-    if (comma === -1) {
-        throw new AppError("E_PROVIDER_BAD_REQUEST", "Unexpected image_url format");
-    }
-    return Buffer.from(imageUrl.slice(comma + 1), "base64");
+    return Buffer.from(imagePart.inlineData.data, "base64");
 }

package/package.json

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