grok-imagine-image-mcp-server 1.0.0

package/dist/tools/edit.js

@@ -0,0 +1,236 @@
+/**
+ * Edit image tool - Edit images using xAI Grok Imagine
+ * Uses /v1/images/edits endpoint
+ */
+import * as fs from 'fs/promises';
+import * as path from 'path';
+import { McpError, ErrorCode } from '@modelcontextprotocol/sdk/types.js';
+import { saveBase64Image, downloadAndSaveImage, generateThumbnailData, isThumbnailEnabled, createThumbnailContent, } from '../utils/image.js';
+import { normalizeAndValidatePath, getDisplayPath, generateUniqueFilePath, } from '../utils/path.js';
+import { debugLog } from '../utils/debug.js';
+import { RESOLUTIONS, MODELS } from '../types/tools.js';
+// Edit endpoint is different from generation endpoint
+const XAI_EDIT_ENDPOINT = 'https://api.x.ai/v1/images/edits';
+export async function editImage(apiKey, params) {
+    debugLog('Edit image called with params:', {
+        ...params,
+        image_base64: params.image_base64 ? '[REDACTED]' : undefined,
+    });
+    const { prompt, image_path, image_base64, image_url, output_path = 'edited_image.jpg', model = 'grok-imagine-image', n = 1, resolution = '1k', response_format = 'b64_json', return_base64 = false, include_thumbnail, } = params;
+    // Validate that at least one image source is provided
+    if (!image_path && !image_base64 && !image_url) {
+        throw new McpError(ErrorCode.InvalidParams, 'One of image_path, image_base64, or image_url is required');
+    }
+    // Normalize and validate output path
+    let normalizedPath = await normalizeAndValidatePath(output_path);
+    normalizedPath = await generateUniqueFilePath(normalizedPath);
+    // Validation
+    if (!prompt || prompt.trim().length === 0) {
+        throw new McpError(ErrorCode.InvalidParams, 'Prompt is required and cannot be empty');
+    }
+    if (!MODELS.includes(model)) {
+        throw new McpError(ErrorCode.InvalidParams, `Invalid model: ${model}. Must be one of: ${MODELS.join(', ')}`);
+    }
+    // Only grok-imagine-image supports image editing
+    if (model !== 'grok-imagine-image') {
+        throw new McpError(ErrorCode.InvalidParams, `Image editing is only supported by grok-imagine-image model. Got: ${model}`);
+    }
+    if (n < 1 || n > 10) {
+        throw new McpError(ErrorCode.InvalidParams, `Invalid n: ${n}. Must be between 1 and 10`);
+    }
+    if (!RESOLUTIONS.includes(resolution)) {
+        throw new McpError(ErrorCode.InvalidParams, `Invalid resolution: ${resolution}. Must be one of: ${RESOLUTIONS.join(', ')}`);
+    }
+    try {
+        // Prepare image URL
+        let imageDataUrl;
+        if (image_base64) {
+            imageDataUrl = `data:image/jpeg;base64,${image_base64}`;
+            debugLog('Using provided base64 image');
+        }
+        else if (image_path) {
+            // Read image from file and convert to data URL
+            const absolutePath = path.isAbsolute(image_path)
+                ? image_path
+                : path.join(process.cwd(), image_path);
+            debugLog(`Reading image from: ${absolutePath}`);
+            try {
+                const imageBuffer = await fs.readFile(absolutePath);
+                const base64 = imageBuffer.toString('base64');
+                imageDataUrl = `data:image/jpeg;base64,${base64}`;
+                debugLog(`Image loaded: ${imageBuffer.length} bytes`);
+            }
+            catch (error) {
+                throw new McpError(ErrorCode.InvalidParams, `Failed to read image file: ${error.message}`);
+            }
+        }
+        else if (image_url) {
+            // Use URL directly or download and convert
+            if (image_url.startsWith('data:')) {
+                imageDataUrl = image_url;
+            }
+            else {
+                // Download image from URL and convert to data URL
+                debugLog(`Downloading image from: ${image_url}`);
+                try {
+                    const response = await fetch(image_url);
+                    if (!response.ok) {
+                        throw new Error(`HTTP ${response.status}`);
+                    }
+                    const arrayBuffer = await response.arrayBuffer();
+                    const base64 = Buffer.from(arrayBuffer).toString('base64');
+                    imageDataUrl = `data:image/jpeg;base64,${base64}`;
+                    debugLog(`Image downloaded: ${arrayBuffer.byteLength} bytes`);
+                }
+                catch (error) {
+                    throw new McpError(ErrorCode.InvalidParams, `Failed to download image from URL: ${error.message}`);
+                }
+            }
+        }
+        else {
+            throw new McpError(ErrorCode.InvalidParams, 'No image source provided');
+        }
+        debugLog('Calling xAI Edit API...');
+        // Build request body according to /v1/images/edits spec
+        const requestBody = {
+            model,
+            prompt,
+            image: {
+                url: imageDataUrl,
+            },
+            n,
+            resolution,
+            response_format,
+        };
+        debugLog('Request body:', {
+            ...requestBody,
+            image: { url: '[DATA_URL]' },
+        });
+        // Call xAI Edit API
+        const response = await fetch(XAI_EDIT_ENDPOINT, {
+            method: 'POST',
+            headers: {
+                'Content-Type': 'application/json',
+                Authorization: `Bearer ${apiKey}`,
+            },
+            body: JSON.stringify(requestBody),
+        });
+        if (!response.ok) {
+            const errorData = await response.json().catch(() => ({}));
+            const errorMessage = errorData.error?.message || `HTTP ${response.status}: ${response.statusText}`;
+            debugLog('API error:', errorData);
+            if (response.status === 401) {
+                throw new McpError(ErrorCode.InvalidRequest, 'Authentication failed. Please check your XAI_API_KEY environment variable.');
+            }
+            else if (response.status === 403) {
+                throw new McpError(ErrorCode.InvalidRequest, 'Access denied. Please check your API key permissions.');
+            }
+            else if (response.status === 400) {
+                throw new McpError(ErrorCode.InvalidRequest, `Bad request: ${errorMessage}`);
+            }
+            else if (response.status === 429) {
+                throw new McpError(ErrorCode.InvalidRequest, 'Rate limit exceeded. Please wait and try again.');
+            }
+            else {
+                throw new McpError(ErrorCode.InternalError, `API error (${response.status}): ${errorMessage}`);
+            }
+        }
+        const data = (await response.json());
+        debugLog('API response received');
+        if (!data.data || data.data.length === 0) {
+            throw new McpError(ErrorCode.InternalError, 'No image data returned from API');
+        }
+        debugLog(`Received ${data.data.length} image(s) from API`);
+        // Process and save all generated images
+        const savedPaths = [];
+        const base64Results = [];
+        const revisedPrompts = [];
+        for (let i = 0; i < data.data.length; i++) {
+            const imgData = data.data[i];
+            // Collect revised prompts
+            if (imgData.revised_prompt) {
+                revisedPrompts.push(imgData.revised_prompt);
+            }
+            // Generate numbered filename for multiple images
+            let imagePath = normalizedPath;
+            if (n > 1) {
+                const pathParts = normalizedPath.split('.');
+                const ext = pathParts.pop();
+                const basePath = pathParts.join('.');
+                imagePath = `${basePath}_${i + 1}.${ext}`;
+            }
+            if (imgData.b64_json) {
+                // Save base64 image
+                await saveBase64Image(imgData.b64_json, imagePath);
+                savedPaths.push(imagePath);
+                if (return_base64) {
+                    base64Results.push(imgData.b64_json);
+                }
+                debugLog(`Image ${i + 1}: Saved from b64_json to ${imagePath}`);
+            }
+            else if (imgData.url) {
+                // Download and save URL image
+                await downloadAndSaveImage(imgData.url, imagePath);
+                savedPaths.push(imagePath);
+                debugLog(`Image ${i + 1}: Downloaded from URL to ${imagePath}`);
+            }
+            else {
+                throw new McpError(ErrorCode.InternalError, `No image data (url or b64_json) in response for image ${i + 1}`);
+            }
+        }
+        // Determine if thumbnails should be included
+        const shouldIncludeThumbnail = include_thumbnail !== undefined ? include_thumbnail : isThumbnailEnabled();
+        // Build result message
+        let resultText;
+        if (n === 1) {
+            const displayPath = getDisplayPath(savedPaths[0]);
+            resultText = `Image edited successfully: ${displayPath}`;
+        }
+        else {
+            resultText = `${n} images edited successfully:\n`;
+            savedPaths.forEach((filePath, idx) => {
+                resultText += `  ${idx + 1}. ${getDisplayPath(filePath)}\n`;
+            });
+        }
+        // Add parameters info
+        resultText += `\nParameters:`;
+        resultText += `\n  - Model: ${model}`;
+        resultText += `\n  - Resolution: ${resolution}`;
+        // Add revised prompt if available
+        if (revisedPrompts.length > 0) {
+            resultText += `\n\nRevised prompt: ${revisedPrompts[0]}`;
+        }
+        // Add base64 data if requested
+        if (return_base64 && base64Results.length > 0) {
+            resultText += '\n\nBase64 data included in response.';
+        }
+        // Return with thumbnails if enabled
+        if (shouldIncludeThumbnail) {
+            debugLog('[Thumbnail] Including thumbnails in response');
+            const content = [{ type: 'text', text: resultText }];
+            // Generate and add thumbnails for each image
+            for (const imagePath of savedPaths) {
+                try {
+                    const thumbnailData = await generateThumbnailData(imagePath);
+                    const thumbnailContent = createThumbnailContent(thumbnailData);
+                    content.push(thumbnailContent);
+                    debugLog(`[Thumbnail] Added thumbnail for: ${imagePath}`);
+                }
+                catch (error) {
+                    debugLog(`[WARNING] Failed to generate thumbnail for ${imagePath}:`, error.message);
+                    // Continue without thumbnail for this image
+                }
+            }
+            return { content };
+        }
+        return resultText;
+    }
+    catch (error) {
+        debugLog('Error editing image:', error);
+        if (error instanceof McpError) {
+            throw error;
+        }
+        throw new McpError(ErrorCode.InternalError, `Failed to edit image: ${error.message}`);
+    }
+}
+//# sourceMappingURL=edit.js.map

package/dist/tools/edit.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"edit.js","sourceRoot":"","sources":["../../src/tools/edit.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,KAAK,EAAE,MAAM,aAAa,CAAC;AAClC,OAAO,KAAK,IAAI,MAAM,MAAM,CAAC;AAC7B,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,MAAM,oCAAoC,CAAC;AACzE,OAAO,EACL,eAAe,EACf,oBAAoB,EACpB,qBAAqB,EACrB,kBAAkB,EAClB,sBAAsB,GACvB,MAAM,mBAAmB,CAAC;AAC3B,OAAO,EACL,wBAAwB,EACxB,cAAc,EACd,sBAAsB,GACvB,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAE7C,OAAO,EAAE,WAAW,EAAE,MAAM,EAAE,MAAM,mBAAmB,CAAC;AAExD,sDAAsD;AACtD,MAAM,iBAAiB,GAAG,kCAAkC,CAAC;AAE7D,MAAM,CAAC,KAAK,UAAU,SAAS,CAC7B,MAAc,EACd,MAAuB;IAYvB,QAAQ,CAAC,gCAAgC,EAAE;QACzC,GAAG,MAAM;QACT,YAAY,EAAE,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,SAAS;KAC7D,CAAC,CAAC;IAEH,MAAM,EACJ,MAAM,EACN,UAAU,EACV,YAAY,EACZ,SAAS,EACT,WAAW,GAAG,kBAAkB,EAChC,KAAK,GAAG,oBAAoB,EAC5B,CAAC,GAAG,CAAC,EACL,UAAU,GAAG,IAAI,EACjB,eAAe,GAAG,UAAU,EAC5B,aAAa,GAAG,KAAK,EACrB,iBAAiB,GAClB,GAAG,MAAM,CAAC;IAEX,sDAAsD;IACtD,IAAI,CAAC,UAAU,IAAI,CAAC,YAAY,IAAI,CAAC,SAAS,EAAE,CAAC;QAC/C,MAAM,IAAI,QAAQ,CAChB,SAAS,CAAC,aAAa,EACvB,2DAA2D,CAC5D,CAAC;IACJ,CAAC;IAED,qCAAqC;IACrC,IAAI,cAAc,GAAG,MAAM,wBAAwB,CAAC,WAAW,CAAC,CAAC;IACjE,cAAc,GAAG,MAAM,sBAAsB,CAAC,cAAc,CAAC,CAAC;IAE9D,aAAa;IACb,IAAI,CAAC,MAAM,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC1C,MAAM,IAAI,QAAQ,CAChB,SAAS,CAAC,aAAa,EACvB,wCAAwC,CACzC,CAAC;IACJ,CAAC;IAED,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAY,CAAC,EAAE,CAAC;QACnC,MAAM,IAAI,QAAQ,CAChB,SAAS,CAAC,aAAa,EACvB,kBAAkB,KAAK,qBAAqB,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAChE,CAAC;IACJ,CAAC;IAED,iDAAiD;IACjD,IAAI,KAAK,KAAK,oBAAoB,EAAE,CAAC;QACnC,MAAM,IAAI,QAAQ,CAChB,SAAS,CAAC,aAAa,EACvB,qEAAqE,KAAK,EAAE,CAC7E,CAAC;IACJ,CAAC;IAED,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC;QACpB,MAAM,IAAI,QAAQ,CAChB,SAAS,CAAC,aAAa,EACvB,cAAc,CAAC,4BAA4B,CAC5C,CAAC;IACJ,CAAC;IAED,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,UAAiB,CAAC,EAAE,CAAC;QAC7C,MAAM,IAAI,QAAQ,CAChB,SAAS,CAAC,aAAa,EACvB,uBAAuB,UAAU,qBAAqB,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAC/E,CAAC;IACJ,CAAC;IAED,IAAI,CAAC;QACH,oBAAoB;QACpB,IAAI,YAAoB,CAAC;QAEzB,IAAI,YAAY,EAAE,CAAC;YACjB,YAAY,GAAG,0BAA0B,YAAY,EAAE,CAAC;YACxD,QAAQ,CAAC,6BAA6B,CAAC,CAAC;QAC1C,CAAC;aAAM,IAAI,UAAU,EAAE,CAAC;YACtB,+CAA+C;YAC/C,MAAM,YAAY,GAAG,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC;gBAC9C,CAAC,CAAC,UAAU;gBACZ,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,UAAU,CAAC,CAAC;YAEzC,QAAQ,CAAC,uBAAuB,YAAY,EAAE,CAAC,CAAC;YAEhD,IAAI,CAAC;gBACH,MAAM,WAAW,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;gBACpD,MAAM,MAAM,GAAG,WAAW,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;gBAC9C,YAAY,GAAG,0BAA0B,MAAM,EAAE,CAAC;gBAClD,QAAQ,CAAC,iBAAiB,WAAW,CAAC,MAAM,QAAQ,CAAC,CAAC;YACxD,CAAC;YAAC,OAAO,KAAU,EAAE,CAAC;gBACpB,MAAM,IAAI,QAAQ,CAChB,SAAS,CAAC,aAAa,EACvB,8BAA8B,KAAK,CAAC,OAAO,EAAE,CAC9C,CAAC;YACJ,CAAC;QACH,CAAC;aAAM,IAAI,SAAS,EAAE,CAAC;YACrB,2CAA2C;YAC3C,IAAI,SAAS,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;gBAClC,YAAY,GAAG,SAAS,CAAC;YAC3B,CAAC;iBAAM,CAAC;gBACN,kDAAkD;gBAClD,QAAQ,CAAC,2BAA2B,SAAS,EAAE,CAAC,CAAC;gBAEjD,IAAI,CAAC;oBACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,SAAS,CAAC,CAAC;oBACxC,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;wBACjB,MAAM,IAAI,KAAK,CAAC,QAAQ,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;oBAC7C,CAAC;oBACD,MAAM,WAAW,GAAG,MAAM,QAAQ,CAAC,WAAW,EAAE,CAAC;oBACjD,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;oBAC3D,YAAY,GAAG,0BAA0B,MAAM,EAAE,CAAC;oBAClD,QAAQ,CAAC,qBAAqB,WAAW,CAAC,UAAU,QAAQ,CAAC,CAAC;gBAChE,CAAC;gBAAC,OAAO,KAAU,EAAE,CAAC;oBACpB,MAAM,IAAI,QAAQ,CAChB,SAAS,CAAC,aAAa,EACvB,sCAAsC,KAAK,CAAC,OAAO,EAAE,CACtD,CAAC;gBACJ,CAAC;YACH,CAAC;QACH,CAAC;aAAM,CAAC;YACN,MAAM,IAAI,QAAQ,CAAC,SAAS,CAAC,aAAa,EAAE,0BAA0B,CAAC,CAAC;QAC1E,CAAC;QAED,QAAQ,CAAC,yBAAyB,CAAC,CAAC;QAEpC,wDAAwD;QACxD,MAAM,WAAW,GAAwB;YACvC,KAAK;YACL,MAAM;YACN,KAAK,EAAE;gBACL,GAAG,EAAE,YAAY;aAClB;YACD,CAAC;YACD,UAAU;YACV,eAAe;SAChB,CAAC;QAEF,QAAQ,CAAC,eAAe,EAAE;YACxB,GAAG,WAAW;YACd,KAAK,EAAE,EAAE,GAAG,EAAE,YAAY,EAAE;SAC7B,CAAC,CAAC;QAEH,oBAAoB;QACpB,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,iBAAiB,EAAE;YAC9C,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;gBAClC,aAAa,EAAE,UAAU,MAAM,EAAE;aAClC;YACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC;SAClC,CAAC,CAAC;QAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,SAAS,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;YAC1D,MAAM,YAAY,GAChB,SAAS,CAAC,KAAK,EAAE,OAAO,IAAI,QAAQ,QAAQ,CAAC,MAAM,KAAK,QAAQ,CAAC,UAAU,EAAE,CAAC;YAEhF,QAAQ,CAAC,YAAY,EAAE,SAAS,CAAC,CAAC;YAElC,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;gBAC5B,MAAM,IAAI,QAAQ,CAChB,SAAS,CAAC,cAAc,EACxB,4EAA4E,CAC7E,CAAC;YACJ,CAAC;iBAAM,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;gBACnC,MAAM,IAAI,QAAQ,CAChB,SAAS,CAAC,cAAc,EACxB,uDAAuD,CACxD,CAAC;YACJ,CAAC;iBAAM,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;gBACnC,MAAM,IAAI,QAAQ,CAAC,SAAS,CAAC,cAAc,EAAE,gBAAgB,YAAY,EAAE,CAAC,CAAC;YAC/E,CAAC;iBAAM,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;gBACnC,MAAM,IAAI,QAAQ,CAChB,SAAS,CAAC,cAAc,EACxB,iDAAiD,CAClD,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACN,MAAM,IAAI,QAAQ,CAChB,SAAS,CAAC,aAAa,EACvB,cAAc,QAAQ,CAAC,MAAM,MAAM,YAAY,EAAE,CAClD,CAAC;YACJ,CAAC;QACH,CAAC;QAED,MAAM,IAAI,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAqB,CAAC;QAEzD,QAAQ,CAAC,uBAAuB,CAAC,CAAC;QAElC,IAAI,CAAC,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACzC,MAAM,IAAI,QAAQ,CAAC,SAAS,CAAC,aAAa,EAAE,iCAAiC,CAAC,CAAC;QACjF,CAAC;QAED,QAAQ,CAAC,YAAY,IAAI,CAAC,IAAI,CAAC,MAAM,oBAAoB,CAAC,CAAC;QAE3D,wCAAwC;QACxC,MAAM,UAAU,GAAa,EAAE,CAAC;QAChC,MAAM,aAAa,GAAa,EAAE,CAAC;QACnC,MAAM,cAAc,GAAa,EAAE,CAAC;QAEpC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YAC1C,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAE7B,0BAA0B;YAC1B,IAAI,OAAO,CAAC,cAAc,EAAE,CAAC;gBAC3B,cAAc,CAAC,IAAI,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;YAC9C,CAAC;YAED,iDAAiD;YACjD,IAAI,SAAS,GAAG,cAAc,CAAC;YAC/B,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC;gBACV,MAAM,SAAS,GAAG,cAAc,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;gBAC5C,MAAM,GAAG,GAAG,SAAS,CAAC,GAAG,EAAE,CAAC;gBAC5B,MAAM,QAAQ,GAAG,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;gBACrC,SAAS,GAAG,GAAG,QAAQ,IAAI,CAAC,GAAG,CAAC,IAAI,GAAG,EAAE,CAAC;YAC5C,CAAC;YAED,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC;gBACrB,oBAAoB;gBACpB,MAAM,eAAe,CAAC,OAAO,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;gBACnD,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;gBAC3B,IAAI,aAAa,EAAE,CAAC;oBAClB,aAAa,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;gBACvC,CAAC;gBACD,QAAQ,CAAC,SAAS,CAAC,GAAG,CAAC,4BAA4B,SAAS,EAAE,CAAC,CAAC;YAClE,CAAC;iBAAM,IAAI,OAAO,CAAC,GAAG,EAAE,CAAC;gBACvB,8BAA8B;gBAC9B,MAAM,oBAAoB,CAAC,OAAO,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;gBACnD,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;gBAC3B,QAAQ,CAAC,SAAS,CAAC,GAAG,CAAC,4BAA4B,SAAS,EAAE,CAAC,CAAC;YAClE,CAAC;iBAAM,CAAC;gBACN,MAAM,IAAI,QAAQ,CAChB,SAAS,CAAC,aAAa,EACvB,yDAAyD,CAAC,GAAG,CAAC,EAAE,CACjE,CAAC;YACJ,CAAC;QACH,CAAC;QAED,6CAA6C;QAC7C,MAAM,sBAAsB,GAC1B,iBAAiB,KAAK,SAAS,CAAC,CAAC,CAAC,iBAAiB,CAAC,CAAC,CAAC,kBAAkB,EAAE,CAAC;QAE7E,uBAAuB;QACvB,IAAI,UAAkB,CAAC;QACvB,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;YACZ,MAAM,WAAW,GAAG,cAAc,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC;YAClD,UAAU,GAAG,8BAA8B,WAAW,EAAE,CAAC;QAC3D,CAAC;aAAM,CAAC;YACN,UAAU,GAAG,GAAG,CAAC,gCAAgC,CAAC;YAClD,UAAU,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,GAAG,EAAE,EAAE;gBACnC,UAAU,IAAI,KAAK,GAAG,GAAG,CAAC,KAAK,cAAc,CAAC,QAAQ,CAAC,IAAI,CAAC;YAC9D,CAAC,CAAC,CAAC;QACL,CAAC;QAED,sBAAsB;QACtB,UAAU,IAAI,eAAe,CAAC;QAC9B,UAAU,IAAI,gBAAgB,KAAK,EAAE,CAAC;QACtC,UAAU,IAAI,qBAAqB,UAAU,EAAE,CAAC;QAEhD,kCAAkC;QAClC,IAAI,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC9B,UAAU,IAAI,uBAAuB,cAAc,CAAC,CAAC,CAAC,EAAE,CAAC;QAC3D,CAAC;QAED,+BAA+B;QAC/B,IAAI,aAAa,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC9C,UAAU,IAAI,uCAAuC,CAAC;QACxD,CAAC;QAED,oCAAoC;QACpC,IAAI,sBAAsB,EAAE,CAAC;YAC3B,QAAQ,CAAC,8CAA8C,CAAC,CAAC;YACzD,MAAM,OAAO,GAKR,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC,CAAC;YAE1C,6CAA6C;YAC7C,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE,CAAC;gBACnC,IAAI,CAAC;oBACH,MAAM,aAAa,GAAG,MAAM,qBAAqB,CAAC,SAAS,CAAC,CAAC;oBAC7D,MAAM,gBAAgB,GAAG,sBAAsB,CAAC,aAAa,CAAC,CAAC;oBAC/D,OAAO,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;oBAC/B,QAAQ,CAAC,oCAAoC,SAAS,EAAE,CAAC,CAAC;gBAC5D,CAAC;gBAAC,OAAO,KAAU,EAAE,CAAC;oBACpB,QAAQ,CACN,8CAA8C,SAAS,GAAG,EAC1D,KAAK,CAAC,OAAO,CACd,CAAC;oBACF,4CAA4C;gBAC9C,CAAC;YACH,CAAC;YAED,OAAO,EAAE,OAAO,EAAE,CAAC;QACrB,CAAC;QAED,OAAO,UAAU,CAAC;IACpB,CAAC;IAAC,OAAO,KAAU,EAAE,CAAC;QACpB,QAAQ,CAAC,sBAAsB,EAAE,KAAK,CAAC,CAAC;QAExC,IAAI,KAAK,YAAY,QAAQ,EAAE,CAAC;YAC9B,MAAM,KAAK,CAAC;QACd,CAAC;QAED,MAAM,IAAI,QAAQ,CAChB,SAAS,CAAC,aAAa,EACvB,yBAAyB,KAAK,CAAC,OAAO,EAAE,CACzC,CAAC;IACJ,CAAC;AACH,CAAC"}

package/dist/tools/generate.d.ts

@@ -0,0 +1,13 @@
+/**
+ * Generate image tool - Create images from text prompts using xAI Grok Imagine
+ */
+import type { GenerateImageParams } from '../types/tools.js';
+export declare function generateImage(apiKey: string, params: GenerateImageParams): Promise<string | {
+    content: Array<{
+        type: string;
+        text?: string;
+        data?: string;
+        mimeType?: string;
+    }>;
+}>;
+//# sourceMappingURL=generate.d.ts.map

package/dist/tools/generate.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"generate.d.ts","sourceRoot":"","sources":["../../src/tools/generate.ts"],"names":[],"mappings":"AAAA;;GAEG;AAgBH,OAAO,KAAK,EAAE,mBAAmB,EAAoB,MAAM,mBAAmB,CAAC;AAU/E,wBAAsB,aAAa,CACjC,MAAM,EAAE,MAAM,EACd,MAAM,EAAE,mBAAmB,GAC1B,OAAO,CACN,MAAM,GACN;IACE,OAAO,EAAE,KAAK,CAAC;QACb,IAAI,EAAE,MAAM,CAAC;QACb,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,QAAQ,CAAC,EAAE,MAAM,CAAC;KACnB,CAAC,CAAC;CACJ,CACJ,CAgQA"}

package/dist/tools/generate.js

@@ -0,0 +1,183 @@
+/**
+ * Generate image tool - Create images from text prompts using xAI Grok Imagine
+ */
+import { McpError, ErrorCode } from '@modelcontextprotocol/sdk/types.js';
+import { saveBase64Image, downloadAndSaveImage, generateThumbnailData, isThumbnailEnabled, createThumbnailContent, } from '../utils/image.js';
+import { normalizeAndValidatePath, getDisplayPath, generateUniqueFilePath, } from '../utils/path.js';
+import { debugLog } from '../utils/debug.js';
+import { ASPECT_RATIOS, RESOLUTIONS, QUALITIES, MODELS, } from '../types/tools.js';
+const XAI_API_ENDPOINT = 'https://api.x.ai/v1/images/generations';
+export async function generateImage(apiKey, params) {
+    debugLog('Generate image called with params:', params);
+    const { prompt, output_path = 'generated_image.jpg', model = 'grok-imagine-image', n = 1, aspect_ratio = '1:1', resolution = '1k', quality, response_format = 'b64_json', return_base64 = false, include_thumbnail, } = params;
+    // Normalize and validate output path
+    let normalizedPath = await normalizeAndValidatePath(output_path);
+    normalizedPath = await generateUniqueFilePath(normalizedPath);
+    // Validation
+    if (!prompt || prompt.trim().length === 0) {
+        throw new McpError(ErrorCode.InvalidParams, 'Prompt is required and cannot be empty');
+    }
+    if (!MODELS.includes(model)) {
+        throw new McpError(ErrorCode.InvalidParams, `Invalid model: ${model}. Must be one of: ${MODELS.join(', ')}`);
+    }
+    if (n < 1 || n > 10) {
+        throw new McpError(ErrorCode.InvalidParams, `Invalid n: ${n}. Must be between 1 and 10`);
+    }
+    // Validate aspect ratio
+    if (!ASPECT_RATIOS.includes(aspect_ratio)) {
+        throw new McpError(ErrorCode.InvalidParams, `Invalid aspect_ratio: ${aspect_ratio}. Must be one of: ${ASPECT_RATIOS.join(', ')}`);
+    }
+    if (!RESOLUTIONS.includes(resolution)) {
+        throw new McpError(ErrorCode.InvalidParams, `Invalid resolution: ${resolution}. Must be one of: ${RESOLUTIONS.join(', ')}`);
+    }
+    if (quality && !QUALITIES.includes(quality)) {
+        throw new McpError(ErrorCode.InvalidParams, `Invalid quality: ${quality}. Must be one of: ${QUALITIES.join(', ')}`);
+    }
+    try {
+        debugLog('Calling xAI API...');
+        // Build request body
+        const requestBody = {
+            model,
+            prompt,
+            n,
+            aspect_ratio,
+            resolution,
+            response_format,
+        };
+        // Quality is currently no-op but include if specified
+        if (quality) {
+            requestBody.quality = quality;
+        }
+        debugLog('Request body:', requestBody);
+        // Call xAI API
+        const response = await fetch(XAI_API_ENDPOINT, {
+            method: 'POST',
+            headers: {
+                'Content-Type': 'application/json',
+                Authorization: `Bearer ${apiKey}`,
+            },
+            body: JSON.stringify(requestBody),
+        });
+        if (!response.ok) {
+            const errorData = await response.json().catch(() => ({}));
+            const errorMessage = errorData.error?.message || `HTTP ${response.status}: ${response.statusText}`;
+            debugLog('API error:', errorData);
+            if (response.status === 401) {
+                throw new McpError(ErrorCode.InvalidRequest, 'Authentication failed. Please check your XAI_API_KEY environment variable.');
+            }
+            else if (response.status === 403) {
+                throw new McpError(ErrorCode.InvalidRequest, 'Access denied. Please check your API key permissions.');
+            }
+            else if (response.status === 400) {
+                throw new McpError(ErrorCode.InvalidRequest, `Bad request: ${errorMessage}`);
+            }
+            else if (response.status === 429) {
+                throw new McpError(ErrorCode.InvalidRequest, 'Rate limit exceeded. Please wait and try again.');
+            }
+            else {
+                throw new McpError(ErrorCode.InternalError, `API error (${response.status}): ${errorMessage}`);
+            }
+        }
+        const data = (await response.json());
+        debugLog('API response received');
+        if (!data.data || data.data.length === 0) {
+            throw new McpError(ErrorCode.InternalError, 'No image data returned from API');
+        }
+        debugLog(`Received ${data.data.length} image(s) from API`);
+        // Process and save all generated images
+        const savedPaths = [];
+        const base64Results = [];
+        const revisedPrompts = [];
+        for (let i = 0; i < data.data.length; i++) {
+            const imageData = data.data[i];
+            // Collect revised prompts
+            if (imageData.revised_prompt) {
+                revisedPrompts.push(imageData.revised_prompt);
+            }
+            // Generate numbered filename for multiple images
+            let imagePath = normalizedPath;
+            if (n > 1) {
+                const pathParts = normalizedPath.split('.');
+                const ext = pathParts.pop();
+                const basePath = pathParts.join('.');
+                imagePath = `${basePath}_${i + 1}.${ext}`;
+            }
+            if (imageData.b64_json) {
+                // Save base64 image
+                await saveBase64Image(imageData.b64_json, imagePath);
+                savedPaths.push(imagePath);
+                if (return_base64) {
+                    base64Results.push(imageData.b64_json);
+                }
+                debugLog(`Image ${i + 1}: Saved from b64_json to ${imagePath}`);
+            }
+            else if (imageData.url) {
+                // Download and save URL image
+                await downloadAndSaveImage(imageData.url, imagePath);
+                savedPaths.push(imagePath);
+                debugLog(`Image ${i + 1}: Downloaded from URL to ${imagePath}`);
+            }
+            else {
+                throw new McpError(ErrorCode.InternalError, `No image data (url or b64_json) in response for image ${i + 1}`);
+            }
+        }
+        // Determine if thumbnails should be included
+        const shouldIncludeThumbnail = include_thumbnail !== undefined ? include_thumbnail : isThumbnailEnabled();
+        // Build result message
+        let resultText;
+        if (n === 1) {
+            const displayPath = getDisplayPath(savedPaths[0]);
+            resultText = `Image generated successfully: ${displayPath}`;
+        }
+        else {
+            resultText = `${n} images generated successfully:\n`;
+            savedPaths.forEach((filePath, idx) => {
+                resultText += `  ${idx + 1}. ${getDisplayPath(filePath)}\n`;
+            });
+        }
+        // Add parameters info
+        resultText += `\nParameters:`;
+        resultText += `\n  - Model: ${model}`;
+        resultText += `\n  - Aspect ratio: ${aspect_ratio}`;
+        resultText += `\n  - Resolution: ${resolution}`;
+        if (quality) {
+            resultText += `\n  - Quality: ${quality}`;
+        }
+        // Add revised prompt if available
+        if (revisedPrompts.length > 0) {
+            resultText += `\n\nRevised prompt: ${revisedPrompts[0]}`;
+        }
+        // Add base64 data if requested
+        if (return_base64 && base64Results.length > 0) {
+            resultText += '\n\nBase64 data included in response.';
+        }
+        // Return with thumbnails if enabled
+        if (shouldIncludeThumbnail) {
+            debugLog('[Thumbnail] Including thumbnails in response');
+            const content = [{ type: 'text', text: resultText }];
+            // Generate and add thumbnails for each image
+            for (const imagePath of savedPaths) {
+                try {
+                    const thumbnailData = await generateThumbnailData(imagePath);
+                    const thumbnailContent = createThumbnailContent(thumbnailData);
+                    content.push(thumbnailContent);
+                    debugLog(`[Thumbnail] Added thumbnail for: ${imagePath}`);
+                }
+                catch (error) {
+                    debugLog(`[WARNING] Failed to generate thumbnail for ${imagePath}:`, error.message);
+                    // Continue without thumbnail for this image
+                }
+            }
+            return { content };
+        }
+        return resultText;
+    }
+    catch (error) {
+        debugLog('Error generating image:', error);
+        if (error instanceof McpError) {
+            throw error;
+        }
+        throw new McpError(ErrorCode.InternalError, `Failed to generate image: ${error.message}`);
+    }
+}
+//# sourceMappingURL=generate.js.map

package/dist/tools/generate.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"generate.js","sourceRoot":"","sources":["../../src/tools/generate.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,MAAM,oCAAoC,CAAC;AACzE,OAAO,EACL,eAAe,EACf,oBAAoB,EACpB,qBAAqB,EACrB,kBAAkB,EAClB,sBAAsB,GACvB,MAAM,mBAAmB,CAAC;AAC3B,OAAO,EACL,wBAAwB,EACxB,cAAc,EACd,sBAAsB,GACvB,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAE7C,OAAO,EACL,aAAa,EACb,WAAW,EACX,SAAS,EACT,MAAM,GACP,MAAM,mBAAmB,CAAC;AAE3B,MAAM,gBAAgB,GAAG,wCAAwC,CAAC;AAElE,MAAM,CAAC,KAAK,UAAU,aAAa,CACjC,MAAc,EACd,MAA2B;IAY3B,QAAQ,CAAC,oCAAoC,EAAE,MAAM,CAAC,CAAC;IAEvD,MAAM,EACJ,MAAM,EACN,WAAW,GAAG,qBAAqB,EACnC,KAAK,GAAG,oBAAoB,EAC5B,CAAC,GAAG,CAAC,EACL,YAAY,GAAG,KAAK,EACpB,UAAU,GAAG,IAAI,EACjB,OAAO,EACP,eAAe,GAAG,UAAU,EAC5B,aAAa,GAAG,KAAK,EACrB,iBAAiB,GAClB,GAAG,MAAM,CAAC;IAEX,qCAAqC;IACrC,IAAI,cAAc,GAAG,MAAM,wBAAwB,CAAC,WAAW,CAAC,CAAC;IACjE,cAAc,GAAG,MAAM,sBAAsB,CAAC,cAAc,CAAC,CAAC;IAE9D,aAAa;IACb,IAAI,CAAC,MAAM,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC1C,MAAM,IAAI,QAAQ,CAChB,SAAS,CAAC,aAAa,EACvB,wCAAwC,CACzC,CAAC;IACJ,CAAC;IAED,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAY,CAAC,EAAE,CAAC;QACnC,MAAM,IAAI,QAAQ,CAChB,SAAS,CAAC,aAAa,EACvB,kBAAkB,KAAK,qBAAqB,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAChE,CAAC;IACJ,CAAC;IAED,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC;QACpB,MAAM,IAAI,QAAQ,CAChB,SAAS,CAAC,aAAa,EACvB,cAAc,CAAC,4BAA4B,CAC5C,CAAC;IACJ,CAAC;IAED,wBAAwB;IACxB,IAAI,CAAC,aAAa,CAAC,QAAQ,CAAC,YAAmB,CAAC,EAAE,CAAC;QACjD,MAAM,IAAI,QAAQ,CAChB,SAAS,CAAC,aAAa,EACvB,yBAAyB,YAAY,qBAAqB,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CACrF,CAAC;IACJ,CAAC;IAED,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,UAAiB,CAAC,EAAE,CAAC;QAC7C,MAAM,IAAI,QAAQ,CAChB,SAAS,CAAC,aAAa,EACvB,uBAAuB,UAAU,qBAAqB,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAC/E,CAAC;IACJ,CAAC;IAED,IAAI,OAAO,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,OAAc,CAAC,EAAE,CAAC;QACnD,MAAM,IAAI,QAAQ,CAChB,SAAS,CAAC,aAAa,EACvB,oBAAoB,OAAO,qBAAqB,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CACvE,CAAC;IACJ,CAAC;IAED,IAAI,CAAC;QACH,QAAQ,CAAC,oBAAoB,CAAC,CAAC;QAE/B,qBAAqB;QACrB,MAAM,WAAW,GAAwB;YACvC,KAAK;YACL,MAAM;YACN,CAAC;YACD,YAAY;YACZ,UAAU;YACV,eAAe;SAChB,CAAC;QAEF,sDAAsD;QACtD,IAAI,OAAO,EAAE,CAAC;YACZ,WAAW,CAAC,OAAO,GAAG,OAAO,CAAC;QAChC,CAAC;QAED,QAAQ,CAAC,eAAe,EAAE,WAAW,CAAC,CAAC;QAEvC,eAAe;QACf,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,gBAAgB,EAAE;YAC7C,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;gBAClC,aAAa,EAAE,UAAU,MAAM,EAAE;aAClC;YACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC;SAClC,CAAC,CAAC;QAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,SAAS,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;YAC1D,MAAM,YAAY,GAChB,SAAS,CAAC,KAAK,EAAE,OAAO,IAAI,QAAQ,QAAQ,CAAC,MAAM,KAAK,QAAQ,CAAC,UAAU,EAAE,CAAC;YAEhF,QAAQ,CAAC,YAAY,EAAE,SAAS,CAAC,CAAC;YAElC,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;gBAC5B,MAAM,IAAI,QAAQ,CAChB,SAAS,CAAC,cAAc,EACxB,4EAA4E,CAC7E,CAAC;YACJ,CAAC;iBAAM,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;gBACnC,MAAM,IAAI,QAAQ,CAChB,SAAS,CAAC,cAAc,EACxB,uDAAuD,CACxD,CAAC;YACJ,CAAC;iBAAM,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;gBACnC,MAAM,IAAI,QAAQ,CAAC,SAAS,CAAC,cAAc,EAAE,gBAAgB,YAAY,EAAE,CAAC,CAAC;YAC/E,CAAC;iBAAM,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;gBACnC,MAAM,IAAI,QAAQ,CAChB,SAAS,CAAC,cAAc,EACxB,iDAAiD,CAClD,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACN,MAAM,IAAI,QAAQ,CAChB,SAAS,CAAC,aAAa,EACvB,cAAc,QAAQ,CAAC,MAAM,MAAM,YAAY,EAAE,CAClD,CAAC;YACJ,CAAC;QACH,CAAC;QAED,MAAM,IAAI,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAqB,CAAC;QAEzD,QAAQ,CAAC,uBAAuB,CAAC,CAAC;QAElC,IAAI,CAAC,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACzC,MAAM,IAAI,QAAQ,CAAC,SAAS,CAAC,aAAa,EAAE,iCAAiC,CAAC,CAAC;QACjF,CAAC;QAED,QAAQ,CAAC,YAAY,IAAI,CAAC,IAAI,CAAC,MAAM,oBAAoB,CAAC,CAAC;QAE3D,wCAAwC;QACxC,MAAM,UAAU,GAAa,EAAE,CAAC;QAChC,MAAM,aAAa,GAAa,EAAE,CAAC;QACnC,MAAM,cAAc,GAAa,EAAE,CAAC;QAEpC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YAC1C,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAE/B,0BAA0B;YAC1B,IAAI,SAAS,CAAC,cAAc,EAAE,CAAC;gBAC7B,cAAc,CAAC,IAAI,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;YAChD,CAAC;YAED,iDAAiD;YACjD,IAAI,SAAS,GAAG,cAAc,CAAC;YAC/B,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC;gBACV,MAAM,SAAS,GAAG,cAAc,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;gBAC5C,MAAM,GAAG,GAAG,SAAS,CAAC,GAAG,EAAE,CAAC;gBAC5B,MAAM,QAAQ,GAAG,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;gBACrC,SAAS,GAAG,GAAG,QAAQ,IAAI,CAAC,GAAG,CAAC,IAAI,GAAG,EAAE,CAAC;YAC5C,CAAC;YAED,IAAI,SAAS,CAAC,QAAQ,EAAE,CAAC;gBACvB,oBAAoB;gBACpB,MAAM,eAAe,CAAC,SAAS,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;gBACrD,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;gBAC3B,IAAI,aAAa,EAAE,CAAC;oBAClB,aAAa,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;gBACzC,CAAC;gBACD,QAAQ,CAAC,SAAS,CAAC,GAAG,CAAC,4BAA4B,SAAS,EAAE,CAAC,CAAC;YAClE,CAAC;iBAAM,IAAI,SAAS,CAAC,GAAG,EAAE,CAAC;gBACzB,8BAA8B;gBAC9B,MAAM,oBAAoB,CAAC,SAAS,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;gBACrD,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;gBAC3B,QAAQ,CAAC,SAAS,CAAC,GAAG,CAAC,4BAA4B,SAAS,EAAE,CAAC,CAAC;YAClE,CAAC;iBAAM,CAAC;gBACN,MAAM,IAAI,QAAQ,CAChB,SAAS,CAAC,aAAa,EACvB,yDAAyD,CAAC,GAAG,CAAC,EAAE,CACjE,CAAC;YACJ,CAAC;QACH,CAAC;QAED,6CAA6C;QAC7C,MAAM,sBAAsB,GAC1B,iBAAiB,KAAK,SAAS,CAAC,CAAC,CAAC,iBAAiB,CAAC,CAAC,CAAC,kBAAkB,EAAE,CAAC;QAE7E,uBAAuB;QACvB,IAAI,UAAkB,CAAC;QACvB,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;YACZ,MAAM,WAAW,GAAG,cAAc,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC;YAClD,UAAU,GAAG,iCAAiC,WAAW,EAAE,CAAC;QAC9D,CAAC;aAAM,CAAC;YACN,UAAU,GAAG,GAAG,CAAC,mCAAmC,CAAC;YACrD,UAAU,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,GAAG,EAAE,EAAE;gBACnC,UAAU,IAAI,KAAK,GAAG,GAAG,CAAC,KAAK,cAAc,CAAC,QAAQ,CAAC,IAAI,CAAC;YAC9D,CAAC,CAAC,CAAC;QACL,CAAC;QAED,sBAAsB;QACtB,UAAU,IAAI,eAAe,CAAC;QAC9B,UAAU,IAAI,gBAAgB,KAAK,EAAE,CAAC;QACtC,UAAU,IAAI,uBAAuB,YAAY,EAAE,CAAC;QACpD,UAAU,IAAI,qBAAqB,UAAU,EAAE,CAAC;QAChD,IAAI,OAAO,EAAE,CAAC;YACZ,UAAU,IAAI,kBAAkB,OAAO,EAAE,CAAC;QAC5C,CAAC;QAED,kCAAkC;QAClC,IAAI,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC9B,UAAU,IAAI,uBAAuB,cAAc,CAAC,CAAC,CAAC,EAAE,CAAC;QAC3D,CAAC;QAED,+BAA+B;QAC/B,IAAI,aAAa,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC9C,UAAU,IAAI,uCAAuC,CAAC;QACxD,CAAC;QAED,oCAAoC;QACpC,IAAI,sBAAsB,EAAE,CAAC;YAC3B,QAAQ,CAAC,8CAA8C,CAAC,CAAC;YACzD,MAAM,OAAO,GAKR,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC,CAAC;YAE1C,6CAA6C;YAC7C,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE,CAAC;gBACnC,IAAI,CAAC;oBACH,MAAM,aAAa,GAAG,MAAM,qBAAqB,CAAC,SAAS,CAAC,CAAC;oBAC7D,MAAM,gBAAgB,GAAG,sBAAsB,CAAC,aAAa,CAAC,CAAC;oBAC/D,OAAO,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;oBAC/B,QAAQ,CAAC,oCAAoC,SAAS,EAAE,CAAC,CAAC;gBAC5D,CAAC;gBAAC,OAAO,KAAU,EAAE,CAAC;oBACpB,QAAQ,CACN,8CAA8C,SAAS,GAAG,EAC1D,KAAK,CAAC,OAAO,CACd,CAAC;oBACF,4CAA4C;gBAC9C,CAAC;YACH,CAAC;YAED,OAAO,EAAE,OAAO,EAAE,CAAC;QACrB,CAAC;QAED,OAAO,UAAU,CAAC;IACpB,CAAC;IAAC,OAAO,KAAU,EAAE,CAAC;QACpB,QAAQ,CAAC,yBAAyB,EAAE,KAAK,CAAC,CAAC;QAE3C,IAAI,KAAK,YAAY,QAAQ,EAAE,CAAC;YAC9B,MAAM,KAAK,CAAC;QACd,CAAC;QAED,MAAM,IAAI,QAAQ,CAChB,SAAS,CAAC,aAAa,EACvB,6BAA6B,KAAK,CAAC,OAAO,EAAE,CAC7C,CAAC;IACJ,CAAC;AACH,CAAC"}

package/dist/types/batch.d.ts

@@ -0,0 +1,137 @@
+/**
+ * Batch processing type definitions
+ */
+import type { AspectRatio, Resolution, Model } from './tools.js';
+/**
+ * Individual job configuration in batch
+ */
+export interface BatchJobConfig {
+    /** Text prompt describing the image to generate */
+    prompt: string;
+    /** Output file path (optional, auto-generated if not specified) */
+    output_path?: string;
+    /** Model to use (default: grok-imagine-image) */
+    model?: Model;
+    /** Aspect ratio (model-dependent, default: 1:1) */
+    aspect_ratio?: AspectRatio;
+    /** Resolution (default: 1k) */
+    resolution?: Resolution;
+    /** Number of images to generate for this prompt (1-10, default: 1) */
+    n?: number;
+    /** Path to source image for editing */
+    image_path?: string;
+    /** Base64 encoded source image for editing */
+    image_base64?: string;
+    /** URL of source image for editing */
+    image_url?: string;
+}
+/**
+ * Retry policy configuration
+ */
+export interface RetryPolicy {
+    /** Maximum number of retry attempts (0-5, default: 2) */
+    max_retries?: number;
+    /** Delay between retries in milliseconds (100-60000, default: 1000) */
+    retry_delay_ms?: number;
+    /** Error patterns to trigger retry (case-insensitive) */
+    retry_on_errors?: string[];
+}
+/**
+ * Batch configuration file structure
+ */
+export interface BatchConfig {
+    /** Array of job configurations (required, 1-100 jobs) */
+    jobs: BatchJobConfig[];
+    /** Default output directory for generated images */
+    output_dir?: string;
+    /** Maximum concurrent jobs (1-10, default: 2) */
+    max_concurrent?: number;
+    /** Timeout in milliseconds (default: 600000 = 10 minutes) */
+    timeout?: number;
+    /** Retry policy for failed jobs */
+    retry_policy?: RetryPolicy;
+    /** Default model for all jobs */
+    default_model?: Model;
+    /** Default resolution for all jobs */
+    default_resolution?: Resolution;
+    /** Default aspect ratio for all jobs */
+    default_aspect_ratio?: AspectRatio;
+}
+/**
+ * Result of a single batch job
+ */
+export interface BatchJobResult {
+    /** Job index (1-based) */
+    index: number;
+    /** Original prompt */
+    prompt: string;
+    /** Job status */
+    status: 'completed' | 'failed' | 'cancelled';
+    /** Output file path(s) */
+    output_paths?: string[];
+    /** Error message if failed */
+    error?: string;
+    /** Job duration in milliseconds */
+    duration_ms?: number;
+    /** Revised prompt from API */
+    revised_prompt?: string;
+    /** Whether this was an edit job */
+    is_edit?: boolean;
+}
+/**
+ * Overall batch execution result
+ */
+export interface BatchResult {
+    /** Total number of jobs */
+    total: number;
+    /** Number of successful jobs */
+    succeeded: number;
+    /** Number of failed jobs */
+    failed: number;
+    /** Number of cancelled jobs (timeout) */
+    cancelled: number;
+    /** Individual job results */
+    results: BatchJobResult[];
+    /** Batch start timestamp (ISO) */
+    started_at: string;
+    /** Batch finish timestamp (ISO) */
+    finished_at: string;
+    /** Total batch duration in milliseconds */
+    total_duration_ms: number;
+    /** Estimated total cost in USD */
+    estimated_cost?: number;
+}
+/**
+ * Batch execution options (from CLI)
+ */
+export interface BatchExecutionOptions {
+    /** Override output directory */
+    outputDir?: string;
+    /** Output format */
+    format?: 'text' | 'json';
+    /** Timeout in milliseconds */
+    timeout?: number;
+    /** Maximum concurrent jobs */
+    maxConcurrent?: number;
+    /** Estimate cost only without executing */
+    estimateOnly?: boolean;
+    /** Allow any output path (for CI/CD) */
+    allowAnyPath?: boolean;
+}
+/**
+ * Cost estimation result
+ */
+export interface CostEstimate {
+    totalJobs: number;
+    totalImages: number;
+    estimatedCostMin: number;
+    estimatedCostMax: number;
+    breakdown: {
+        model: string;
+        count: number;
+        images: number;
+        costMin: number;
+        costMax: number;
+    }[];
+}
+//# sourceMappingURL=batch.d.ts.map

package/dist/types/batch.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"batch.d.ts","sourceRoot":"","sources":["../../src/types/batch.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,KAAK,EAAE,WAAW,EAAE,UAAU,EAAE,KAAK,EAAE,MAAM,YAAY,CAAC;AAEjE;;GAEG;AACH,MAAM,WAAW,cAAc;IAC7B,mDAAmD;IACnD,MAAM,EAAE,MAAM,CAAC;IACf,mEAAmE;IACnE,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,iDAAiD;IACjD,KAAK,CAAC,EAAE,KAAK,CAAC;IACd,mDAAmD;IACnD,YAAY,CAAC,EAAE,WAAW,CAAC;IAC3B,+BAA+B;IAC/B,UAAU,CAAC,EAAE,UAAU,CAAC;IACxB,sEAAsE;IACtE,CAAC,CAAC,EAAE,MAAM,CAAC;IAGX,uCAAuC;IACvC,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,8CAA8C;IAC9C,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,sCAAsC;IACtC,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED;;GAEG;AACH,MAAM,WAAW,WAAW;IAC1B,yDAAyD;IACzD,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,uEAAuE;IACvE,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,yDAAyD;IACzD,eAAe,CAAC,EAAE,MAAM,EAAE,CAAC;CAC5B;AAED;;GAEG;AACH,MAAM,WAAW,WAAW;IAC1B,yDAAyD;IACzD,IAAI,EAAE,cAAc,EAAE,CAAC;IACvB,oDAAoD;IACpD,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,iDAAiD;IACjD,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,6DAA6D;IAC7D,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,mCAAmC;IACnC,YAAY,CAAC,EAAE,WAAW,CAAC;IAC3B,iCAAiC;IACjC,aAAa,CAAC,EAAE,KAAK,CAAC;IACtB,sCAAsC;IACtC,kBAAkB,CAAC,EAAE,UAAU,CAAC;IAChC,wCAAwC;IACxC,oBAAoB,CAAC,EAAE,WAAW,CAAC;CACpC;AAED;;GAEG;AACH,MAAM,WAAW,cAAc;IAC7B,0BAA0B;IAC1B,KAAK,EAAE,MAAM,CAAC;IACd,sBAAsB;IACtB,MAAM,EAAE,MAAM,CAAC;IACf,iBAAiB;IACjB,MAAM,EAAE,WAAW,GAAG,QAAQ,GAAG,WAAW,CAAC;IAC7C,0BAA0B;IAC1B,YAAY,CAAC,EAAE,MAAM,EAAE,CAAC;IACxB,8BAA8B;IAC9B,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,mCAAmC;IACnC,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,8BAA8B;IAC9B,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,mCAAmC;IACnC,OAAO,CAAC,EAAE,OAAO,CAAC;CACnB;AAED;;GAEG;AACH,MAAM,WAAW,WAAW;IAC1B,2BAA2B;IAC3B,KAAK,EAAE,MAAM,CAAC;IACd,gCAAgC;IAChC,SAAS,EAAE,MAAM,CAAC;IAClB,4BAA4B;IAC5B,MAAM,EAAE,MAAM,CAAC;IACf,yCAAyC;IACzC,SAAS,EAAE,MAAM,CAAC;IAClB,6BAA6B;IAC7B,OAAO,EAAE,cAAc,EAAE,CAAC;IAC1B,kCAAkC;IAClC,UAAU,EAAE,MAAM,CAAC;IACnB,mCAAmC;IACnC,WAAW,EAAE,MAAM,CAAC;IACpB,2CAA2C;IAC3C,iBAAiB,EAAE,MAAM,CAAC;IAC1B,kCAAkC;IAClC,cAAc,CAAC,EAAE,MAAM,CAAC;CACzB;AAED;;GAEG;AACH,MAAM,WAAW,qBAAqB;IACpC,gCAAgC;IAChC,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,oBAAoB;IACpB,MAAM,CAAC,EAAE,MAAM,GAAG,MAAM,CAAC;IACzB,8BAA8B;IAC9B,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,8BAA8B;IAC9B,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,2CAA2C;IAC3C,YAAY,CAAC,EAAE,OAAO,CAAC;IACvB,wCAAwC;IACxC,YAAY,CAAC,EAAE,OAAO,CAAC;CACxB;AAED;;GAEG;AACH,MAAM,WAAW,YAAY;IAC3B,SAAS,EAAE,MAAM,CAAC;IAClB,WAAW,EAAE,MAAM,CAAC;IACpB,gBAAgB,EAAE,MAAM,CAAC;IACzB,gBAAgB,EAAE,MAAM,CAAC;IACzB,SAAS,EAAE;QACT,KAAK,EAAE,MAAM,CAAC;QACd,KAAK,EAAE,MAAM,CAAC;QACd,MAAM,EAAE,MAAM,CAAC;QACf,OAAO,EAAE,MAAM,CAAC;QAChB,OAAO,EAAE,MAAM,CAAC;KACjB,EAAE,CAAC;CACL"}

package/dist/types/batch.js

@@ -0,0 +1,5 @@
+/**
+ * Batch processing type definitions
+ */
+export {};
+//# sourceMappingURL=batch.js.map

package/dist/types/batch.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"batch.js","sourceRoot":"","sources":["../../src/types/batch.ts"],"names":[],"mappings":"AAAA;;GAEG"}