mcp-image 0.7.0 → 0.8.1

package/README.md

@@ -1,47 +1,62 @@
-# 🍌 MCP Image Generator
-
-> Powered by Gemini 3 Pro Image - Nano Banana Pro 🍌
-
-A powerful MCP (Model Context Protocol) server that enables AI assistants to generate and edit images using Google's Gemini 3 Pro Image (Nano Banana Pro 🍌). Seamlessly integrate advanced image generation capabilities into Codex, Cursor, Claude Code, and other MCP-compatible AI tools.
-
-## ✨ Features
-
-- **AI-Powered Image Generation**: Create images from text prompts using Gemini 3 Pro Image (Nano Banana Pro)
-- **Intelligent Prompt Enhancement**: Automatically optimizes your prompts using Gemini 2.5 Flash for superior image quality
-  - Adds photographic and artistic details
-  - Enriches lighting, composition, and atmosphere descriptions
-  - Preserves your intent while maximizing generation quality
-- **Image Editing**: Transform existing images with natural language instructions
-  - Context-aware editing that preserves original style
-  - Maintains visual consistency with source image
-- **High-Resolution Output**: Support for 2K and 4K image generation
-  - Standard quality for fast generation
-  - 2K resolution for enhanced detail
-  - 4K resolution for professional-grade images with superior text rendering
-- **Flexible Aspect Ratios**: Multiple aspect ratio options (1:1, 16:9, 9:16, 21:9, and more)
-- **Advanced Options**:
-  - Multi-image blending for composite scenes
-  - Character consistency across generations
-  - World knowledge integration for accurate context
-- **Multiple Output Formats**: PNG, JPEG, WebP support
-- **File Output**: Images are saved as files for easy access and integration
+# MCP Image Generator 🍌
+
+> AI image generation and editing MCP server for Cursor, Claude Code, Codex, and any MCP-compatible tool — powered by Nano Banana 2 and Nano Banana Pro (Google Gemini).
+
+[![npm version](https://badge.fury.io/js/mcp-image.svg)](https://www.npmjs.com/package/mcp-image)
+[![npm downloads](https://img.shields.io/npm/dm/mcp-image.svg)](https://www.npmjs.com/package/mcp-image)
+[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
+
+An MCP server that turns simple text prompts into high-quality images. Unlike a simple API wrapper, this server automatically enhances your prompt and configures sensible defaults for generation — you don't need to learn prompt engineering or tune settings. Just describe what you want.
+
+## How It Works
+
+```
+You: "cat on a roof"
+        ↓
+  Your AI assistant infers context
+  (purpose, style, mood, resolution...)
+        ↓
+  MCP optimizes your prompt
+  (adds lighting, composition, atmosphere, artistic details)
+        ↓
+  Image generation with smart defaults
+  (grounding, consistency, resolution — all configured automatically)
+        ↓
+  High-quality image, zero effort
+```
+
+Your AI assistant interprets your intent — the style, purpose, and context behind your request. The MCP focuses on output quality by refining the prompt to meet a structured visual clarity standard and selecting appropriate generation settings. You just describe what you want.
 
-## 🎨 Agent Skill: Image Generation Prompt Guide
+The prompt optimizer uses a **Subject–Context–Style** framework (powered by Gemini 2.5 Flash) to fill in missing visual details — subject characteristics, environment, lighting, camera work — while preserving your original intent. It doesn't blindly add details: prompts that already meet the quality standard are left largely intact.
 
-This project also provides a standalone **Agent Skill** that teaches AI assistants to write better image generation prompts — no MCP server or API key required.
+**Example — what the optimizer does to a short prompt:**
 
-### What it does
+> **Input:** "cat on a roof"
+>
+> **After optimization:** "A sleek, midnight black cat, perched with poised elegance on the apex of a weathered, terracotta tile roof. Its emerald eyes, narrowed slightly, reflect the warm glow of a setting sun. Each individual tile is distinct, showing subtle variations in color and texture, with patches of moss clinging to the crevices. The cat's fur is sharply defined, catching the golden hour light, highlighting its sleek contours. In the background, the silhouettes of distant, old-world city buildings with ornate spires are softly blurred, bathed in a gradient of fiery orange, soft pink, and deep violet twilight. A gentle, ethereal mist begins to rise from the alleyways below, adding a touch of mystery. The composition is a medium shot, taken from a slightly low angle, emphasizing the cat's commanding presence against the vast sky. Photorealistic style, captured with a prime lens, wide aperture to create a beautiful bokeh, enhancing the depth of field."
+
+## Features
+
+- **Built-in Prompt Optimization**: Your simple prompt is automatically enriched with photographic and artistic details — lighting, composition, atmosphere — using Gemini 2.5 Flash. No prompt engineering skills required.
+- **Three Quality Tiers**: Choose between fast iteration, balanced quality, or maximum fidelity with Nano Banana 2 (Gemini 3.1 Flash Image) and Nano Banana Pro (Gemini 3 Pro Image). [See Quality Presets](#quality-presets).
+- **Image Editing**: Transform existing images with natural language instructions (image-to-image) while preserving original style and visual consistency.
+- **High-Resolution Output**: Up to 4K image generation for professional-grade output with superior text rendering and fine details.
+- **Flexible Aspect Ratios**: From square (1:1) to ultra-wide (21:9) and ultra-tall (1:8) formats.
+- **Character Consistency**: Maintain consistent character appearance across multiple generations — ideal for storyboards, product shots, and visual series.
+- **Advanced Capabilities**:
+  - Google Search grounding for real-time factual accuracy
+  - World knowledge for photorealistic depictions of historical figures, landmarks, and factual scenarios
+  - Multi-image blending for composite scenes
+  - Purpose-aware generation (e.g., "cookbook cover" produces different results than "social media post")
+- **Multiple Output Formats**: PNG, JPEG, WebP support.
 
-> **Note:** This skill does not generate images itself — it teaches your AI assistant to write better prompts. Your AI tool must already have built-in image generation capabilities (e.g., Cursor's image generation feature).
+## Agent Skill: Image Generation Prompt Guide
 
-A reference guide that AI assistants use to improve image generation prompts based on the **Subject-Context-Style** framework. Works with any image model (Gemini, DALL-E, Flux, Stable Diffusion, etc.).
+This project also provides a standalone **[Agent Skill](https://agentskills.io)** (`SKILL.md`) that teaches AI assistants to write better image generation prompts — no MCP server or API key required.
 
-Covers:
+> **Note:** This skill does not generate images itself. It teaches your AI assistant to write better prompts for tools that already have built-in image generation (e.g., Cursor's native image generation).
 
-- **Prompt structure** — How to build prompts around Subject, Context, and Style
-- **Visual details** — Lighting, textures, camera angles, atmosphere, text in images
-- **Advanced features** — Character consistency, multi-element composition, factual accuracy, purpose-specific output
-- **Image editing** — How to describe edits while keeping the original look intact
+Based on the **Subject-Context-Style** framework, covering prompt structure, visual details (lighting, textures, camera angles), advanced techniques (character consistency, composition), and image editing. Works with any image model (Gemini, GPT Image, Flux, Stable Diffusion, Midjourney, etc.).
 
 ### Install
 
@@ -62,25 +77,25 @@ npx mcp-image skills install --path ~/.codex/skills
 npx mcp-image skills install --path ~/.claude/skills
 ```
 
-### When to use the Skill vs the MCP server
+### When to Use the Skill vs the MCP Server
 
 | | MCP Server | Agent Skill |
 |---|---|---|
 | **Use when** | Your AI tool does not have built-in image generation | Your AI tool already generates images natively |
 | **Requires** | Gemini API key | Nothing |
-| **What it does** | Generates images via API | Teaches the AI to write better prompts |
-| **Works with** | MCP-compatible tools | Any tool supporting the [Agent Skills](https://agentskills.io) standard |
+| **What it does** | Generates images via Gemini API with automatic prompt optimization | Teaches the AI to write better prompts |
+| **Works with** | MCP-compatible tools (Cursor, Claude Code, Codex, etc.) | Any tool supporting the [Agent Skills](https://agentskills.io) open standard |
 
 ---
 
-## 🔧 Prerequisites
+## Prerequisites
 
 - **Node.js** 20 or higher
 - **Gemini API Key** - Get yours at [Google AI Studio](https://aistudio.google.com/apikey)
-- **Codex**, **Cursor**, or **Claude Code** (file I/O capable AI tools)
+- An MCP-compatible AI tool: **Cursor**, **Claude Code**, **Codex**, or others
 - Basic terminal/command line knowledge
 
-## 🚀 Quick Start
+## Quick Start
 
 ### 1. Get Your Gemini API Key
 
@@ -140,34 +155,53 @@ claude mcp add mcp-image --scope user --env GEMINI_API_KEY=your-api-key --env IM
 
 ⚠️ **Security Note**: Never commit your API key to version control. Keep it secure and use environment-specific configuration.
 
-📁 **Path Requirements**: 
+📁 **Path Requirements**:
 - `IMAGE_OUTPUT_DIR` must be an absolute path (e.g., `/Users/username/images`, not `./images`)
 - Defaults to `./output` in the current working directory if not specified
 - Directory will be created automatically if it doesn't exist
 
-### Optional: Skip Prompt Enhancement
+## Quality Presets
 
-Set `SKIP_PROMPT_ENHANCEMENT=true` to disable automatic prompt optimization and send your prompts directly to the image generator. Useful when you need full control over the exact prompt wording.
+Choose the right balance of speed, quality, and cost:
+
+| Preset | Model | Best for | Speed |
+|--------|-------|----------|-------|
+| `fast` (default) | Nano Banana 2 (Gemini 3.1 Flash Image) | Quick iterations, drafts, high-volume generation | ~30–40s |
+| `balanced` | Nano Banana 2 + Thinking | Production images, good quality with reasonable speed | Medium |
+| `quality` | Nano Banana Pro (Gemini 3 Pro Image) | Final deliverables, maximum fidelity, critical visuals | Slow |
+
+Set the default via `IMAGE_QUALITY` environment variable:
+
+```
+IMAGE_QUALITY=fast       # (default) Fastest generation
+IMAGE_QUALITY=balanced   # Enhanced thinking for better quality
+IMAGE_QUALITY=quality    # Maximum quality output
+```
+
+To override per-request, just tell your AI assistant (e.g., "generate in high quality" or "use balanced quality"). The assistant will pass the appropriate `quality` parameter automatically.
 
 **Codex:**
 ```toml
 [mcp_servers.mcp-image.env]
 GEMINI_API_KEY = "your_gemini_api_key_here"
-SKIP_PROMPT_ENHANCEMENT = "true"
-IMAGE_OUTPUT_DIR = "/absolute/path/to/images"
+IMAGE_QUALITY = "balanced"
 ```
 
 **Cursor:**
-Add `"SKIP_PROMPT_ENHANCEMENT": "true"` to the env section in your config.
+Add `"IMAGE_QUALITY": "balanced"` to the env section in your config.
 
 **Claude Code:**
 ```bash
-claude mcp add mcp-image --env GEMINI_API_KEY=your-api-key --env SKIP_PROMPT_ENHANCEMENT=true --env IMAGE_OUTPUT_DIR=/absolute/path/to/images -- npx -y mcp-image
+claude mcp add mcp-image --env GEMINI_API_KEY=your-api-key --env IMAGE_QUALITY=balanced --env IMAGE_OUTPUT_DIR=/absolute/path/to/images -- npx -y mcp-image
 ```
 
-## 📖 Usage Examples
+### Skip Prompt Enhancement
+
+Set `SKIP_PROMPT_ENHANCEMENT=true` to disable automatic prompt optimization and send your prompts directly to the image generator. Useful when you need full control over the exact prompt wording.
+
+## Usage Examples
 
-Once configured, your AI assistant can generate images using natural language:
+Once configured, just describe what you want in natural language:
 
 ### Basic Image Generation
 
@@ -175,7 +209,7 @@ Once configured, your AI assistant can generate images using natural language:
 "Generate a serene mountain landscape at sunset with a lake reflection"
 ```
 
-The system automatically enhances this to include rich details about lighting, materials, composition, and atmosphere for optimal results.
+Your prompt is automatically enhanced with rich details about lighting, materials, composition, and atmosphere.
 
 ### Image Editing
 
@@ -192,7 +226,7 @@ The system automatically enhances this to include rich details about lighting, m
 (with maintainCharacterConsistency: true)
 ```
 
-**High-Resolution 4K Generation:**
+**High-Resolution 4K with Text Rendering:**
 ```
 "Generate a professional product photo of a smartphone with clear text on the screen"
 (with imageSize: "4K")
@@ -204,28 +238,29 @@ The system automatically enhances this to include rich details about lighting, m
 (with aspectRatio: "21:9")
 ```
 
-## 🔧 API Reference
+## API Reference
 
 ### `generate_image` Tool
 
-The MCP server exposes a single tool for all image operations. Internally, it uses a two-stage process:
-1. **Prompt Optimization**: Gemini 2.5 Flash analyzes and enriches your prompt
-2. **Image Generation**: Gemini 3 Pro Image creates the final image
+The server uses a two-stage process with separate models for each stage:
+1. **Prompt Optimization** (Gemini 2.5 Flash): Refines your prompt using the Subject–Context–Style framework. Skippable via `SKIP_PROMPT_ENHANCEMENT`.
+2. **Image Generation** (Nano Banana 2 or Pro): Creates the final image. Model varies by quality preset.
 
 #### Parameters
 
 | Parameter | Type | Required | Description |
 |-----------|------|----------|-------------|
 | `prompt` | string | ✅ | Text description or editing instruction |
-| `inputImagePath` | string | - | Absolute path to input image for editing |
+| `quality` | string | - | Quality preset: `fast` (default), `balanced`, `quality`. Overrides `IMAGE_QUALITY` env var for this request |
+| `inputImagePath` | string | - | Absolute path to input image for image-to-image editing |
 | `fileName` | string | - | Custom filename for output (auto-generated if not specified) |
-| `aspectRatio` | string | - | Aspect ratio for the generated image. Supported values: `1:1` (square, default), `2:3`, `3:2`, `3:4`, `4:3`, `4:5`, `5:4`, `9:16`, `16:9`, `21:9` |
-| `imageSize` | string | - | Image resolution for high-quality output. Specify `2K` or `4K` for higher resolution images with better text rendering and fine details. Leave unspecified for standard quality. Supported values: `2K`, `4K` |
+| `aspectRatio` | string | - | `1:1` (default), `2:3`, `3:2`, `3:4`, `4:3`, `4:5`, `5:4`, `9:16`, `16:9`, `21:9`, `1:4`, `1:8`, `4:1`, `8:1` |
+| `imageSize` | string | - | `1K`, `2K`, `4K`. Leave unspecified for standard quality |
 | `blendImages` | boolean | - | Enable multi-image blending for combining multiple visual elements naturally |
 | `maintainCharacterConsistency` | boolean | - | Maintain character appearance consistency across different poses and scenes |
-| `useWorldKnowledge` | boolean | - | Use real-world knowledge for accurate context (recommended for historical figures, landmarks, or factual scenarios) |
-| `useGoogleSearch` | boolean | - | Enable Google Search grounding to access real-time web information for factually accurate image generation. Use when prompt requires current or time-sensitive data that may have changed since the model's knowledge cutoff. Leave disabled for creative, fictional, historical, or timeless content. |
-| `purpose` | string | - | Intended use for the image (e.g., "cookbook cover", "social media post", "presentation slide"). Helps tailor visual style, quality level, and details to match the purpose. |
+| `useWorldKnowledge` | boolean | - | Use real-world knowledge for accurate context (historical figures, landmarks, factual scenarios) |
+| `useGoogleSearch` | boolean | - | Enable Google Search grounding for real-time factual accuracy |
+| `purpose` | string | - | Intended use (e.g., "cookbook cover", "social media post"). Helps tailor visual style and details |
 
 #### Response
 
@@ -238,14 +273,14 @@ The MCP server exposes a single tool for all image operations. Internally, it us
     "mimeType": "image/png"
   },
   "metadata": {
-    "model": "gemini-3-pro-image-preview",
+    "model": "gemini-3.1-flash-image-preview",
     "processingTime": 5000,
-    "timestamp": "2024-01-01T12:00:00.000Z"
+    "timestamp": "2026-01-01T12:00:00.000Z"
   }
 }
 ```
 
-## 🛠️ Troubleshooting
+## Troubleshooting
 
 ### Common Issues
 
@@ -265,27 +300,31 @@ The MCP server exposes a single tool for all image operations. Internally, it us
 
 ### Performance Tips
 
-- Image generation: 30-60 seconds typical (includes prompt optimization)
-- Image editing: 15-45 seconds typical (includes context analysis)
-- High-resolution generation (2K/4K): May take longer but provides superior quality
-- Simple prompts work great - the AI automatically adds professional details
+- `fast` preset: ~30–40 seconds typical (includes prompt optimization)
+- `balanced` preset: Slightly longer due to enhanced thinking
+- `quality` preset: Slower but highest fidelity output
+- High-resolution (2K/4K): Additional processing time for superior detail
+- Simple prompts work great — the optimizer automatically adds professional details
 - Complex prompts are preserved and further enhanced
-- Consider enabling `useWorldKnowledge` for historical or factual subjects
+- Consider `useWorldKnowledge` for historical or factual subjects
 - Use `imageSize: "4K"` when text clarity and fine details are critical
 
-## 💰 Usage Notes
+## Usage Notes
 
-- This MCP server uses the paid Gemini API for both prompt optimization and image generation
-  - Gemini 2.5 Flash for intelligent prompt enhancement (minimal token usage)
-  - Gemini 3 Pro Image for actual image generation
+- This MCP server uses the paid Gemini API:
+  - **Prompt optimization**: Gemini 2.5 Flash (minimal token usage)
+  - **Image generation**: Model depends on quality preset
+    - `fast` / `balanced`: Nano Banana 2 — Gemini 3.1 Flash Image (lower cost)
+    - `quality`: Nano Banana Pro — Gemini 3 Pro Image (higher cost)
+  - `balanced` uses additional thinking tokens (slightly higher cost than `fast`)
 - Check current pricing and rate limits at [Google AI Studio](https://aistudio.google.com/)
 - Monitor your API usage to avoid unexpected charges
 - The prompt optimization step adds minimal cost while significantly improving output quality
 
-## 📄 License
+## License
 
 MIT License - see [LICENSE](LICENSE) for details.
 
 ---
 
-**Need help?** [Open an issue](https://github.com/shinpr/mcp-image/issues) or check the [troubleshooting section](#-troubleshooting) above.
+**Need help?** [Open an issue](https://github.com/shinpr/mcp-image/issues) or check the [troubleshooting section](#troubleshooting) above.

package/bin/install-skills.js

@@ -1,8 +1,11 @@
 #!/usr/bin/env node
-'use strict'
 
-const { cpSync, existsSync, mkdirSync } = require('node:fs')
-const { dirname, resolve } = require('node:path')
+import { cpSync, existsSync, mkdirSync } from 'node:fs'
+import { dirname, resolve } from 'node:path'
+import { fileURLToPath } from 'node:url'
+
+const __filename = fileURLToPath(import.meta.url)
+const __dirname = dirname(__filename)
 
 const SKILLS_SOURCE = resolve(__dirname, '..', 'skills', 'image-generation')
 const SKILL_DIR_NAME = 'image-generation'
@@ -77,7 +80,7 @@ function install(targetPath) {
   console.log(`Installed skills to: ${targetPath}`)
 }
 
-function run(args) {
+export function run(args) {
   if (args.length === 0) {
     printHelp()
     process.exit(0)
@@ -110,5 +113,3 @@ function run(args) {
   console.log('Installed files:')
   console.log('  - image-generation/SKILL.md')
 }
-
-module.exports = { run }

package/dist/api/geminiClient.d.ts

@@ -3,9 +3,10 @@
  * Integrates with Google's Gemini AI API using the official SDK
  * Supports automatic URL Context processing and feature parameters
  */
-import type { Result } from '../types/result';
-import type { Config } from '../utils/config';
-import { GeminiAPIError, NetworkError } from '../utils/errors';
+import type { ImageQuality } from '../types/mcp.js';
+import type { Result } from '../types/result.js';
+import type { Config } from '../utils/config.js';
+import { GeminiAPIError, NetworkError } from '../utils/errors.js';
 /**
  * Metadata for generated images
  */
@@ -27,6 +28,7 @@ export interface GeminiApiParams {
     aspectRatio?: string;
     imageSize?: string;
     useGoogleSearch?: boolean;
+    quality?: ImageQuality;
 }
 /**
  * Result of image generation

package/dist/api/geminiClient.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"geminiClient.d.ts","sourceRoot":"","sources":["../../src/api/geminiClient.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAGH,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,iBAAiB,CAAA;AAE7C,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,iBAAiB,CAAA;AAC7C,OAAO,EAAE,cAAc,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAA;AAkH9D;;GAEG;AACH,MAAM,WAAW,wBAAwB;IACvC,KAAK,EAAE,MAAM,CAAA;IACb,MAAM,EAAE,MAAM,CAAA;IACd,QAAQ,EAAE,MAAM,CAAA;IAChB,SAAS,EAAE,IAAI,CAAA;IACf,kBAAkB,EAAE,OAAO,CAAA;IAE3B,YAAY,CAAC,EAAE,MAAM,CAAA;IACrB,UAAU,CAAC,EAAE,MAAM,CAAA;CACpB;AAED;;GAEG;AACH,MAAM,WAAW,eAAe;IAC9B,MAAM,EAAE,MAAM,CAAA;IACd,UAAU,CAAC,EAAE,MAAM,CAAA;IACnB,WAAW,CAAC,EAAE,MAAM,CAAA;IACpB,SAAS,CAAC,EAAE,MAAM,CAAA;IAClB,eAAe,CAAC,EAAE,OAAO,CAAA;CAC1B;AAED;;GAEG;AACH,MAAM,WAAW,oBAAoB;IACnC,SAAS,EAAE,MAAM,CAAA;IACjB,QAAQ,EAAE,wBAAwB,CAAA;CACnC;AAED;;GAEG;AACH,MAAM,WAAW,YAAY;IAC3B,aAAa,CACX,MAAM,EAAE,eAAe,GACtB,OAAO,CAAC,MAAM,CAAC,oBAAoB,EAAE,cAAc,GAAG,YAAY,CAAC,CAAC,CAAA;CACxE;AA2UD;;;;GAIG;AACH,wBAAgB,kBAAkB,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAAC,YAAY,EAAE,cAAc,CAAC,CAevF"}
+{"version":3,"file":"geminiClient.d.ts","sourceRoot":"","sources":["../../src/api/geminiClient.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAGH,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAA;AAEnD,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAA;AAEhD,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAA;AAChD,OAAO,EAAE,cAAc,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAA;AAsHjE;;GAEG;AACH,MAAM,WAAW,wBAAwB;IACvC,KAAK,EAAE,MAAM,CAAA;IACb,MAAM,EAAE,MAAM,CAAA;IACd,QAAQ,EAAE,MAAM,CAAA;IAChB,SAAS,EAAE,IAAI,CAAA;IACf,kBAAkB,EAAE,OAAO,CAAA;IAE3B,YAAY,CAAC,EAAE,MAAM,CAAA;IACrB,UAAU,CAAC,EAAE,MAAM,CAAA;CACpB;AAED;;GAEG;AACH,MAAM,WAAW,eAAe;IAC9B,MAAM,EAAE,MAAM,CAAA;IACd,UAAU,CAAC,EAAE,MAAM,CAAA;IACnB,WAAW,CAAC,EAAE,MAAM,CAAA;IACpB,SAAS,CAAC,EAAE,MAAM,CAAA;IAClB,eAAe,CAAC,EAAE,OAAO,CAAA;IACzB,OAAO,CAAC,EAAE,YAAY,CAAA;CACvB;AAED;;GAEG;AACH,MAAM,WAAW,oBAAoB;IACnC,SAAS,EAAE,MAAM,CAAA;IACjB,QAAQ,EAAE,wBAAwB,CAAA;CACnC;AAED;;GAEG;AACH,MAAM,WAAW,YAAY;IAC3B,aAAa,CACX,MAAM,EAAE,eAAe,GACtB,OAAO,CAAC,MAAM,CAAC,oBAAoB,EAAE,cAAc,GAAG,YAAY,CAAC,CAAC,CAAA;CACxE;AAoVD;;;;GAIG;AACH,wBAAgB,kBAAkB,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAAC,YAAY,EAAE,cAAc,CAAC,CAevF"}

package/dist/api/geminiClient.js

@@ -1,14 +1,12 @@
-"use strict";
 /**
  * Gemini API client for image generation
  * Integrates with Google's Gemini AI API using the official SDK
  * Supports automatic URL Context processing and feature parameters
  */
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.createGeminiClient = createGeminiClient;
-const genai_1 = require("@google/genai");
-const result_1 = require("../types/result");
-const errors_1 = require("../utils/errors");
+import { GoogleGenAI } from '@google/genai';
+import { GEMINI_MODELS } from '../types/mcp.js';
+import { Err, Ok } from '../types/result.js';
+import { GeminiAPIError, NetworkError } from '../utils/errors.js';
 /**
  * Safely analyze response structure for debugging (removes sensitive data)
  */
@@ -70,9 +68,9 @@ function isGeminiResponse(obj) {
  * Implementation of Gemini API client
  */
 class GeminiClientImpl {
-    constructor(genai) {
+    constructor(genai, defaultQuality = 'fast') {
         this.genai = genai;
-        this.modelName = 'gemini-3-pro-image-preview';
+        this.defaultQuality = defaultQuality;
     }
     async generateImage(params) {
         try {
@@ -105,6 +103,10 @@ class GeminiClientImpl {
                     ],
                 });
             }
+            // Determine effective quality
+            const effectiveQuality = params.quality ?? this.defaultQuality;
+            // Select model based on quality preset
+            const modelName = effectiveQuality === 'quality' ? GEMINI_MODELS.PRO : GEMINI_MODELS.FLASH;
             // Construct config object for generateContent
             const imageConfig = {};
             if (params.aspectRatio) {
@@ -113,19 +115,20 @@ class GeminiClientImpl {
             if (params.imageSize) {
                 imageConfig['imageSize'] = params.imageSize;
             }
-            const config = Object.keys(imageConfig).length > 0
-                ? {
-                    imageConfig,
-                    responseModalities: ['IMAGE'],
-                }
-                : {
-                    responseModalities: ['IMAGE'],
-                };
-            // Construct tools array for Google Search grounding
-            const tools = params.useGoogleSearch ? [{ googleSearch: {} }] : undefined;
-            // Generate content using Gemini API (@google/genai v1.17.0+)
+            // Build config with optional thinkingConfig
+            const thinkingConfig = effectiveQuality === 'balanced' ? { thinkingConfig: { thinkingLevel: 'high' } } : {};
+            const config = {
+                ...(Object.keys(imageConfig).length > 0 && { imageConfig }),
+                responseModalities: ['IMAGE'],
+                ...thinkingConfig,
+            };
+            // Construct tools array for Google Search grounding (web + image search)
+            const tools = params.useGoogleSearch
+                ? [{ googleSearch: { searchTypes: { webSearch: {}, imageSearch: {} } } }]
+                : undefined;
+            // Generate content using Gemini API
             const rawResponse = await this.genai.models.generateContent({
-                model: this.modelName,
+                model: modelName,
                 contents: requestContent,
                 config,
                 ...(tools && { tools }),
@@ -137,14 +140,14 @@ class GeminiClientImpl {
                 const asRecord = rawResponse;
                 if (asRecord['error']) {
                     const error = asRecord['error'];
-                    return (0, result_1.Err)(new errors_1.GeminiAPIError(`Gemini API Error: ${error['message'] || 'Unknown error'}`, {
+                    return Err(new GeminiAPIError(`Gemini API Error: ${error['message'] || 'Unknown error'}`, {
                         code: error['code'],
                         status: error['status'],
                         details: error['details'] || responseStructure,
                         stage: 'api_error',
                     }));
                 }
-                return (0, result_1.Err)(new errors_1.GeminiAPIError('Invalid response structure from Gemini API', {
+                return Err(new GeminiAPIError('Invalid response structure from Gemini API', {
                     message: 'The API returned an unexpected response format',
                     responseStructure: responseStructure,
                     stage: 'response_validation',
@@ -160,7 +163,7 @@ class GeminiClientImpl {
             if (responseAsRecord['promptFeedback']) {
                 const promptFeedback = responseAsRecord['promptFeedback'];
                 if (promptFeedback['blockReason'] === 'SAFETY') {
-                    return (0, result_1.Err)(new errors_1.GeminiAPIError('Image generation blocked for safety reasons', {
+                    return Err(new GeminiAPIError('Image generation blocked for safety reasons', {
                         stage: 'prompt_analysis',
                         blockReason: promptFeedback['blockReason'],
                         suggestion: 'Rephrase your prompt to avoid potentially sensitive content',
@@ -168,7 +171,7 @@ class GeminiClientImpl {
                 }
                 if (promptFeedback['blockReason'] === 'OTHER' ||
                     promptFeedback['blockReason'] === 'PROHIBITED_CONTENT') {
-                    return (0, result_1.Err)(new errors_1.GeminiAPIError('Image generation blocked due to prohibited content', {
+                    return Err(new GeminiAPIError('Image generation blocked due to prohibited content', {
                         stage: 'prompt_analysis',
                         blockReason: promptFeedback['blockReason'],
                         suggestion: 'Remove any prohibited content from your prompt and try again',
@@ -177,7 +180,7 @@ class GeminiClientImpl {
             }
             // Check for candidates
             if (!responseData.candidates || responseData.candidates.length === 0) {
-                return (0, result_1.Err)(new errors_1.GeminiAPIError('No image generated: Content may have been filtered', {
+                return Err(new GeminiAPIError('No image generated: Content may have been filtered', {
                     stage: 'generation',
                     candidatesCount: 0,
                     suggestion: 'Try rephrasing your prompt to avoid potentially sensitive content',
@@ -185,7 +188,7 @@ class GeminiClientImpl {
             }
             const candidate = responseData.candidates[0];
             if (!candidate || !candidate.content || !candidate.content.parts) {
-                return (0, result_1.Err)(new errors_1.GeminiAPIError('No valid content in response', {
+                return Err(new GeminiAPIError('No valid content in response', {
                     stage: 'candidate_extraction',
                     suggestion: 'The API response was incomplete. Please try again',
                 }));
@@ -195,7 +198,7 @@ class GeminiClientImpl {
             if (candidate.finishReason) {
                 const finishReason = candidate.finishReason;
                 if (finishReason === 'IMAGE_SAFETY') {
-                    return (0, result_1.Err)(new errors_1.GeminiAPIError('Image generation stopped for safety reasons', {
+                    return Err(new GeminiAPIError('Image generation stopped for safety reasons', {
                         finishReason,
                         stage: 'generation_stopped',
                         suggestion: 'Modify your prompt to avoid potentially sensitive content',
@@ -214,7 +217,7 @@ class GeminiClientImpl {
                     }));
                 }
                 if (finishReason === 'MAX_TOKENS') {
-                    return (0, result_1.Err)(new errors_1.GeminiAPIError('Maximum token limit reached during generation', {
+                    return Err(new GeminiAPIError('Maximum token limit reached during generation', {
                         finishReason,
                         stage: 'generation_stopped',
                         suggestion: 'Try using a shorter or simpler prompt',
@@ -222,7 +225,7 @@ class GeminiClientImpl {
                 }
             }
             if (parts.length === 0) {
-                return (0, result_1.Err)(new errors_1.GeminiAPIError('No content parts in response', {
+                return Err(new GeminiAPIError('No content parts in response', {
                     stage: 'content_extraction',
                     suggestion: 'The generation was incomplete. Please try again',
                 }));
@@ -233,7 +236,7 @@ class GeminiClientImpl {
             if (!imagePart?.inlineData) {
                 // If there's text, it's likely an error message from Gemini
                 const errorMessage = textPart?.text || 'Image generation failed';
-                return (0, result_1.Err)(new errors_1.GeminiAPIError('Image generation failed due to content filtering', {
+                return Err(new GeminiAPIError('Image generation failed due to content filtering', {
                     reason: errorMessage,
                     stage: 'image_extraction',
                     suggestion: 'The prompt was blocked by safety filters. Try rephrasing your prompt to avoid potentially sensitive content.',
@@ -244,7 +247,7 @@ class GeminiClientImpl {
             const mimeType = imagePart.inlineData.mimeType || 'image/png';
             // Create metadata
             const metadata = {
-                model: this.modelName,
+                model: modelName,
                 prompt: params.prompt,
                 mimeType,
                 timestamp: new Date(),
@@ -252,7 +255,7 @@ class GeminiClientImpl {
                 ...(responseData.modelVersion && { modelVersion: responseData.modelVersion }),
                 ...(responseData.responseId && { responseId: responseData.responseId }),
             };
-            return (0, result_1.Ok)({
+            return Ok({
                 imageData: imageBuffer,
                 metadata,
             });
@@ -265,14 +268,14 @@ class GeminiClientImpl {
         const errorMessage = error instanceof Error ? error.message : 'Unknown error';
         // Check if it's a network error
         if (this.isNetworkError(error)) {
-            return (0, result_1.Err)(new errors_1.NetworkError(`Network error during image generation: ${errorMessage}`, 'Check your internet connection and try again', error instanceof Error ? error : undefined));
+            return Err(new NetworkError(`Network error during image generation: ${errorMessage}`, 'Check your internet connection and try again', error instanceof Error ? error : undefined));
         }
         // Check if it's an API-specific error
         if (this.isAPIError(error)) {
-            return (0, result_1.Err)(new errors_1.GeminiAPIError(`Failed to generate image: ${errorMessage}`, this.getAPIErrorSuggestion(errorMessage), this.extractStatusCode(error)));
+            return Err(new GeminiAPIError(`Failed to generate image: ${errorMessage}`, this.getAPIErrorSuggestion(errorMessage), this.extractStatusCode(error)));
         }
         // Generic API error
-        return (0, result_1.Err)(new errors_1.GeminiAPIError(`Failed to generate image with prompt "${prompt}": ${errorMessage}`, 'Check your API key, quota, and prompt validity. Try again with a different prompt'));
+        return Err(new GeminiAPIError(`Failed to generate image with prompt "${prompt}": ${errorMessage}`, 'Check your API key, quota, and prompt validity. Try again with a different prompt'));
     }
     isNetworkError(error) {
         if (error instanceof Error) {
@@ -313,16 +316,16 @@ class GeminiClientImpl {
  * @param config Configuration containing API key and other settings
  * @returns Result containing the client or an error
  */
-function createGeminiClient(config) {
+export function createGeminiClient(config) {
     try {
-        const genai = new genai_1.GoogleGenAI({
+        const genai = new GoogleGenAI({
             apiKey: config.geminiApiKey,
         });
-        return (0, result_1.Ok)(new GeminiClientImpl(genai));
+        return Ok(new GeminiClientImpl(genai, config.imageQuality));
     }
     catch (error) {
         const errorMessage = error instanceof Error ? error.message : 'Unknown error';
-        return (0, result_1.Err)(new errors_1.GeminiAPIError(`Failed to initialize Gemini client: ${errorMessage}`, 'Verify your GEMINI_API_KEY is valid and the @google/genai package is properly installed'));
+        return Err(new GeminiAPIError(`Failed to initialize Gemini client: ${errorMessage}`, 'Verify your GEMINI_API_KEY is valid and the @google/genai package is properly installed'));
     }
 }
 //# sourceMappingURL=geminiClient.js.map