@fre4x/gemini 1.0.25 → 1.0.26

package/dist/index.js

@@ -55417,7 +55417,7 @@ var LIST_MODELS_TOOL = {
 };
 var GENERATE_IMAGE_TOOL = {
   name: "generate_image",
-  description: "Generate an image using Imagen and return it as base64",
+  description: "Generate an image using Imagen, return as base64 and optionally save to disk",
   inputSchema: {
     type: "object",
     properties: {
@@ -55431,6 +55431,10 @@ var GENERATE_IMAGE_TOOL = {
         type: "string",
         description: "Imagen model (default: imagen-3.0-generate-002)",
         default: "imagen-3.0-generate-002"
+      },
+      output_dir: {
+        type: "string",
+        description: "Directory to save image file (e.g. /tmp or ./output). Omit to skip saving."
       }
     },
     required: ["prompt"]
@@ -55454,13 +55458,17 @@ var GENERATE_VIDEO_TOOL = {
 };
 var GET_VIDEO_STATUS_TOOL = {
   name: "get_video_status",
-  description: "Poll the status of an async video generation started by generate_video. Returns done=true and video URLs when complete.",
+  description: "Poll status of async video generation. Returns done=true and video URLs when complete. Optionally downloads and saves to disk.",
   inputSchema: {
     type: "object",
     properties: {
       operation_name: {
         type: "string",
         description: "Operation name returned by generate_video"
+      },
+      output_dir: {
+        type: "string",
+        description: "Directory to save video file when done (e.g. /tmp or ./output). Omit to skip saving."
       }
     },
     required: ["operation_name"]
@@ -55575,15 +55583,25 @@ Mock response \u2014 no API call made.` }] };
       return { content: [{ type: "text", text: lines.join("\n").trim() || "No models found." }] };
     }
     if (name === "generate_image") {
-      const { prompt, aspect_ratio = "1:1", model = "imagen-3.0-generate-002" } = z2.object({
+      const { prompt, aspect_ratio = "1:1", model = "imagen-3.0-generate-002", output_dir } = z2.object({
         prompt: z2.string(),
         aspect_ratio: z2.string().optional(),
-        model: z2.string().optional()
+        model: z2.string().optional(),
+        output_dir: z2.string().optional()
       }).parse(args);
       if (IS_MOCK) {
-        return {
-          content: [{ type: "image", data: "iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAYAAAAfFcSJAAAADUlEQVR42mNk+M9QDwADhgGAWjR9awAAAABJRU5ErkJggg==", mimeType: "image/png" }]
-        };
+        const mockBase64 = "iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAYAAAAfFcSJAAAADUlEQVR42mNk+M9QDwADhgGAWjR9awAAAABJRU5ErkJggg==";
+        const content2 = [
+          { type: "image", data: mockBase64, mimeType: "image/png" }
+        ];
+        if (output_dir) {
+          const fname = `image-${Date.now()}.png`;
+          const fpath = `${output_dir.replace(/\/$/, "")}/${fname}`;
+          await fs.mkdir(output_dir, { recursive: true });
+          await fs.writeFile(fpath, Buffer.from(mockBase64, "base64"));
+          content2.push({ type: "text", text: `[Mock] Saved to: ${fpath}` });
+        }
+        return { content: content2 };
       }
       const response = await ai.models.generateImages({
         model,
@@ -55606,7 +55624,22 @@ Check Imagen access at https://ai.google.dev/` }],
           isError: true
         };
       }
-      return { content: images };
+      const savedPaths = [];
+      if (output_dir) {
+        await fs.mkdir(output_dir, { recursive: true });
+        for (const img of images) {
+          const fname = `image-${Date.now()}-${savedPaths.length + 1}.png`;
+          const fpath = `${output_dir.replace(/\/$/, "")}/${fname}`;
+          await fs.writeFile(fpath, Buffer.from(img.data, "base64"));
+          savedPaths.push(fpath);
+        }
+      }
+      const content = [...images];
+      if (savedPaths.length > 0) {
+        content.push({ type: "text", text: `Saved to:
+${savedPaths.join("\n")}` });
+      }
+      return { content };
     }
     if (name === "generate_video") {
       const { prompt, model = "veo-2.0-generate-001" } = z2.object({ prompt: z2.string(), model: z2.string().optional() }).parse(args);
@@ -55630,11 +55663,17 @@ Poll every ~30 seconds until done=true.` }]
       };
     }
     if (name === "get_video_status") {
-      const { operation_name } = z2.object({ operation_name: z2.string() }).parse(args);
+      const { operation_name, output_dir } = z2.object({ operation_name: z2.string(), output_dir: z2.string().optional() }).parse(args);
       if (IS_MOCK) {
-        return { content: [{ type: "text", text: `[Mock] Operation: ${operation_name}
-done: true
-Video URL: https://example.com/mock-video.mp4` }] };
+        const lines2 = [`[Mock] Operation: ${operation_name}`, "done: true", "Video URL: https://example.com/mock-video.mp4"];
+        if (output_dir) {
+          const fname = `video-${Date.now()}.mp4`;
+          const fpath = `${output_dir.replace(/\/$/, "")}/${fname}`;
+          await fs.mkdir(output_dir, { recursive: true });
+          await fs.writeFile(fpath, "[mock video data]");
+          lines2.push(`Saved to: ${fpath}`);
+        }
+        return { content: [{ type: "text", text: lines2.join("\n") }] };
       }
       const resp = await fetch(
         `https://generativelanguage.googleapis.com/v1beta/${operation_name}?key=${API_KEY}`
@@ -55651,16 +55690,31 @@ Video URL: https://example.com/mock-video.mp4` }] };
         };
       }
       const videos = (op.response?.generatedVideos ?? []).map((v) => v.video?.uri ?? "").filter(Boolean);
-      return {
-        content: [{
-          type: "text",
-          text: videos.length > 0 ? `Video generation complete!
+      if (videos.length === 0) {
+        return { content: [{ type: "text", text: `Complete but no video URLs found:
+${JSON.stringify(op.response, null, 2)}` }] };
+      }
+      const savedPaths = [];
+      if (output_dir) {
+        await fs.mkdir(output_dir, { recursive: true });
+        for (const url2 of videos) {
+          const fname = `video-${Date.now()}-${savedPaths.length + 1}.mp4`;
+          const fpath = `${output_dir.replace(/\/$/, "")}/${fname}`;
+          const videoResp = await fetch(url2);
+          if (!videoResp.ok) throw new Error(`Failed to download video: ${videoResp.status}`);
+          const buf = await videoResp.arrayBuffer();
+          await fs.writeFile(fpath, Buffer.from(buf));
+          savedPaths.push(fpath);
+        }
+      }
+      const lines = [`Video generation complete!
 
 Download URLs:
-${videos.join("\n")}` : `Complete but no video URLs found:
-${JSON.stringify(op.response, null, 2)}`
-        }]
-      };
+${videos.join("\n")}`];
+      if (savedPaths.length > 0) lines.push(`
+Saved to:
+${savedPaths.join("\n")}`);
+      return { content: [{ type: "text", text: lines.join("") }] };
     }
     throw new Error(`Tool not found: ${name}`);
   } catch (error48) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@fre4x/gemini",
-  "version": "1.0.25",
+  "version": "1.0.26",
   "description": "A Gemini MCP server providing text completion, multimodal analysis, and image/video generation.",
   "type": "module",
   "main": "dist/index.js",