google-veo3-1-mcp-server 1.0.0

package/dist/index.js

@@ -0,0 +1,372 @@
+#!/usr/bin/env node
+/**
+ * Google Veo 3.1 MCP Server
+ *
+ * Provides tools for video generation using Google's Veo 3.1 API:
+ * - generate_video: Text-to-Video and Image-to-Video with reference images
+ * - extend_video: Extend existing videos by 7 seconds
+ * - interpolate_frames: Generate video between first and last frames
+ * - get_video_status: Check status of video generation operations
+ */
+import { Server } from '@modelcontextprotocol/sdk/server/index.js';
+import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
+import { CallToolRequestSchema, ListToolsRequestSchema } from '@modelcontextprotocol/sdk/types.js';
+import { config } from 'dotenv';
+import { generateVideo } from './tools/generate.js';
+import { extendVideo } from './tools/extend.js';
+import { interpolateFrames } from './tools/interpolate.js';
+import { getOperationStatus, getApiKey } from './utils/video.js';
+import { debugLog, errorLog, infoLog } from './utils/debug.js';
+import { MODELS, RESOLUTIONS, ASPECT_RATIOS, DURATIONS, REFERENCE_TYPES, PERSON_GENERATION_OPTIONS } from './types/tools.js';
+// Load environment variables
+config();
+// Verify API key is available
+try {
+    getApiKey();
+}
+catch (error) {
+    errorLog('GOOGLE_API_KEY not found in environment variables');
+    process.exit(1);
+}
+// =============================================================================
+// Tool Definitions
+// =============================================================================
+const tools = [
+    {
+        name: 'generate_video',
+        description: `Generate a video using Google Veo 3.1 API.
+
+Supports:
+- Text-to-Video: Generate video from a text prompt
+- Image-to-Video: Animate a static image
+- Reference Images: Use up to 3 asset images or 1 style image for consistency
+
+Models:
+- veo-3.1-generate-preview: High quality (default)
+- veo-3.1-fast-generate-preview: Faster generation
+
+Pricing (per second):
+- 720p/1080p with audio: $0.40 (standard) / $0.20 (fast)
+- 720p/1080p video only: $0.20 (standard) / $0.10 (fast)
+- 4K with audio: $0.60 (standard only)`,
+        inputSchema: {
+            type: 'object',
+            properties: {
+                prompt: {
+                    type: 'string',
+                    description: 'Text prompt describing the video to generate. Required unless image is provided.'
+                },
+                model: {
+                    type: 'string',
+                    enum: MODELS,
+                    description: 'Model to use for generation. Default: veo-3.1-generate-preview'
+                },
+                aspect_ratio: {
+                    type: 'string',
+                    enum: ASPECT_RATIOS,
+                    description: 'Aspect ratio of the video. Default: 16:9'
+                },
+                resolution: {
+                    type: 'string',
+                    enum: RESOLUTIONS,
+                    description: 'Video resolution. 4K only available for standard model. Default: 720p'
+                },
+                duration_seconds: {
+                    type: 'number',
+                    enum: DURATIONS,
+                    description: 'Duration of the video in seconds (4, 6, or 8). Default: 8'
+                },
+                generate_audio: {
+                    type: 'boolean',
+                    description: 'Whether to generate audio. Note: Gemini API preview always generates audio (this parameter is ignored). Use Vertex AI for audio control.'
+                },
+                negative_prompt: {
+                    type: 'string',
+                    description: 'Elements to avoid in the generated video'
+                },
+                image: {
+                    type: 'string',
+                    description: 'Image for Image-to-Video mode. Can be a file path, URL, or base64 string'
+                },
+                reference_images: {
+                    type: 'array',
+                    items: {
+                        type: 'object',
+                        properties: {
+                            image: {
+                                type: 'string',
+                                description: 'Image file path, URL, or base64 string'
+                            },
+                            reference_type: {
+                                type: 'string',
+                                enum: REFERENCE_TYPES,
+                                description: 'Type of reference: "asset" for characters/objects, "style" for visual style'
+                            }
+                        },
+                        required: ['image', 'reference_type']
+                    },
+                    description: 'Reference images for consistency. Max 3 asset images or 1 style image'
+                },
+                person_generation: {
+                    type: 'string',
+                    enum: PERSON_GENERATION_OPTIONS,
+                    description: 'Control person generation: allow_adult (default), dont_allow, allow_all'
+                },
+                seed: {
+                    type: 'number',
+                    description: 'Random seed for reproducibility (0-4294967295)'
+                },
+                output_path: {
+                    type: 'string',
+                    description: 'Path to save the generated video. If not provided, uses OUTPUT_DIR env var'
+                }
+            },
+            required: []
+        }
+    },
+    {
+        name: 'extend_video',
+        description: `Extend an existing video by 7 seconds using Veo 3.1 API.
+
+The extension continues from the last second of the input video.
+
+Requirements:
+- Input video: 1-30 seconds, 24fps, 720p or 1080p
+- Output: Always 7 seconds at 720p
+
+Estimated cost: ~$1.40 per extension`,
+        inputSchema: {
+            type: 'object',
+            properties: {
+                video: {
+                    type: 'string',
+                    description: 'Video to extend. Can be a GCS URI (gs://), file path, URL, or base64 string'
+                },
+                prompt: {
+                    type: 'string',
+                    description: 'Optional prompt to guide the extension'
+                },
+                output_path: {
+                    type: 'string',
+                    description: 'Path to save the extended video'
+                }
+            },
+            required: ['video']
+        }
+    },
+    {
+        name: 'interpolate_frames',
+        description: `Generate a video that smoothly transitions between two keyframes using Veo 3.1 API.
+
+Creates a video that starts at the first frame and ends at the last frame with AI-generated motion in between.
+
+Pricing: Same as generate_video based on duration and audio settings`,
+        inputSchema: {
+            type: 'object',
+            properties: {
+                first_frame: {
+                    type: 'string',
+                    description: 'Starting frame image. Can be a file path, URL, or base64 string'
+                },
+                last_frame: {
+                    type: 'string',
+                    description: 'Ending frame image. Can be a file path, URL, or base64 string'
+                },
+                prompt: {
+                    type: 'string',
+                    description: 'Optional prompt to guide the interpolation'
+                },
+                duration_seconds: {
+                    type: 'number',
+                    enum: DURATIONS,
+                    description: 'Duration of the video in seconds (4, 6, or 8). Default: 8'
+                },
+                generate_audio: {
+                    type: 'boolean',
+                    description: 'Whether to generate audio. Note: Gemini API preview always generates audio (ignored).'
+                },
+                output_path: {
+                    type: 'string',
+                    description: 'Path to save the generated video'
+                }
+            },
+            required: ['first_frame', 'last_frame']
+        }
+    },
+    {
+        name: 'get_video_status',
+        description: `Check the status of a video generation operation.
+
+Returns the current status (done/pending) and video URL if completed.`,
+        inputSchema: {
+            type: 'object',
+            properties: {
+                operation_name: {
+                    type: 'string',
+                    description: 'The operation name returned from generate_video, extend_video, or interpolate_frames'
+                }
+            },
+            required: ['operation_name']
+        }
+    }
+];
+// =============================================================================
+// Server Setup
+// =============================================================================
+const server = new Server({
+    name: 'google-veo3-1-mcp-server',
+    version: '1.0.0'
+}, {
+    capabilities: {
+        tools: {}
+    }
+});
+// =============================================================================
+// Request Handlers
+// =============================================================================
+// List available tools
+server.setRequestHandler(ListToolsRequestSchema, async () => {
+    debugLog('Listing tools');
+    return { tools };
+});
+// Handle tool calls
+server.setRequestHandler(CallToolRequestSchema, async (request) => {
+    const { name, arguments: args } = request.params;
+    debugLog(`Tool called: ${name}`, args);
+    try {
+        switch (name) {
+            case 'generate_video': {
+                const params = args;
+                const result = await generateVideo(params);
+                return {
+                    content: [
+                        {
+                            type: 'text',
+                            text: JSON.stringify(result, null, 2)
+                        }
+                    ]
+                };
+            }
+            case 'extend_video': {
+                const params = args;
+                const result = await extendVideo(params);
+                return {
+                    content: [
+                        {
+                            type: 'text',
+                            text: JSON.stringify(result, null, 2)
+                        }
+                    ]
+                };
+            }
+            case 'interpolate_frames': {
+                const params = args;
+                const result = await interpolateFrames(params);
+                return {
+                    content: [
+                        {
+                            type: 'text',
+                            text: JSON.stringify(result, null, 2)
+                        }
+                    ]
+                };
+            }
+            case 'get_video_status': {
+                const params = args;
+                if (!params.operation_name) {
+                    return {
+                        content: [
+                            {
+                                type: 'text',
+                                text: JSON.stringify({
+                                    success: false,
+                                    error: 'operation_name is required'
+                                }, null, 2)
+                            }
+                        ]
+                    };
+                }
+                try {
+                    const apiKey = getApiKey();
+                    const status = await getOperationStatus(params.operation_name, apiKey);
+                    const result = {
+                        success: true,
+                        operation_name: params.operation_name,
+                        done: status.done ?? false
+                    };
+                    if (status.done && status.response?.generatedVideos?.length) {
+                        result.video_url = status.response.generatedVideos[0].video.uri;
+                    }
+                    if (status.error) {
+                        result.success = false;
+                        result.error = status.error.message;
+                        result.failure_reason = `Error code: ${status.error.code}`;
+                    }
+                    return {
+                        content: [
+                            {
+                                type: 'text',
+                                text: JSON.stringify(result, null, 2)
+                            }
+                        ]
+                    };
+                }
+                catch (error) {
+                    return {
+                        content: [
+                            {
+                                type: 'text',
+                                text: JSON.stringify({
+                                    success: false,
+                                    operation_name: params.operation_name,
+                                    done: false,
+                                    error: error instanceof Error ? error.message : String(error)
+                                }, null, 2)
+                            }
+                        ]
+                    };
+                }
+            }
+            default:
+                return {
+                    content: [
+                        {
+                            type: 'text',
+                            text: JSON.stringify({
+                                success: false,
+                                error: `Unknown tool: ${name}`
+                            }, null, 2)
+                        }
+                    ]
+                };
+        }
+    }
+    catch (error) {
+        errorLog(`Tool execution failed: ${name}`, error);
+        return {
+            content: [
+                {
+                    type: 'text',
+                    text: JSON.stringify({
+                        success: false,
+                        error: error instanceof Error ? error.message : String(error)
+                    }, null, 2)
+                }
+            ]
+        };
+    }
+});
+// =============================================================================
+// Start Server
+// =============================================================================
+async function main() {
+    infoLog('Starting Google Veo 3.1 MCP Server...');
+    const transport = new StdioServerTransport();
+    await server.connect(transport);
+    infoLog('Server connected and ready');
+}
+main().catch((error) => {
+    errorLog('Server failed to start', error);
+    process.exit(1);
+});
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";AAEA;;;;;;;;GAQG;AAEH,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AACnE,OAAO,EAAE,oBAAoB,EAAE,MAAM,2CAA2C,CAAC;AACjF,OAAO,EACL,qBAAqB,EACrB,sBAAsB,EAEvB,MAAM,oCAAoC,CAAC;AAC5C,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAEhC,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACpD,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,EAAE,iBAAiB,EAAE,MAAM,wBAAwB,CAAC;AAC3D,OAAO,EAAE,kBAAkB,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AACjE,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,kBAAkB,CAAC;AAC/D,OAAO,EACL,MAAM,EACN,WAAW,EACX,aAAa,EACb,SAAS,EACT,eAAe,EACf,yBAAyB,EAM1B,MAAM,kBAAkB,CAAC;AAE1B,6BAA6B;AAC7B,MAAM,EAAE,CAAC;AAET,8BAA8B;AAC9B,IAAI,CAAC;IACH,SAAS,EAAE,CAAC;AACd,CAAC;AAAC,OAAO,KAAK,EAAE,CAAC;IACf,QAAQ,CAAC,mDAAmD,CAAC,CAAC;IAC9D,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AAClB,CAAC;AAED,gFAAgF;AAChF,mBAAmB;AACnB,gFAAgF;AAEhF,MAAM,KAAK,GAAW;IACpB;QACE,IAAI,EAAE,gBAAgB;QACtB,WAAW,EAAE;;;;;;;;;;;;;;uCAcsB;QACnC,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,MAAM,EAAE;oBACN,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,kFAAkF;iBAChG;gBACD,KAAK,EAAE;oBACL,IAAI,EAAE,QAAQ;oBACd,IAAI,EAAE,MAA6B;oBACnC,WAAW,EAAE,gEAAgE;iBAC9E;gBACD,YAAY,EAAE;oBACZ,IAAI,EAAE,QAAQ;oBACd,IAAI,EAAE,aAAoC;oBAC1C,WAAW,EAAE,0CAA0C;iBACxD;gBACD,UAAU,EAAE;oBACV,IAAI,EAAE,QAAQ;oBACd,IAAI,EAAE,WAAkC;oBACxC,WAAW,EAAE,uEAAuE;iBACrF;gBACD,gBAAgB,EAAE;oBAChB,IAAI,EAAE,QAAQ;oBACd,IAAI,EAAE,SAAgC;oBACtC,WAAW,EAAE,2DAA2D;iBACzE;gBACD,cAAc,EAAE;oBACd,IAAI,EAAE,SAAS;oBACf,WAAW,EAAE,0IAA0I;iBACxJ;gBACD,eAAe,EAAE;oBACf,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,0CAA0C;iBACxD;gBACD,KAAK,EAAE;oBACL,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,0EAA0E;iBACxF;gBACD,gBAAgB,EAAE;oBAChB,IAAI,EAAE,OAAO;oBACb,KAAK,EAAE;wBACL,IAAI,EAAE,QAAQ;wBACd,UAAU,EAAE;4BACV,KAAK,EAAE;gCACL,IAAI,EAAE,QAAQ;gCACd,WAAW,EAAE,wCAAwC;6BACtD;4BACD,cAAc,EAAE;gCACd,IAAI,EAAE,QAAQ;gCACd,IAAI,EAAE,eAAsC;gCAC5C,WAAW,EAAE,6EAA6E;6BAC3F;yBACF;wBACD,QAAQ,EAAE,CAAC,OAAO,EAAE,gBAAgB,CAAC;qBACtC;oBACD,WAAW,EAAE,uEAAuE;iBACrF;gBACD,iBAAiB,EAAE;oBACjB,IAAI,EAAE,QAAQ;oBACd,IAAI,EAAE,yBAAgD;oBACtD,WAAW,EAAE,yEAAyE;iBACvF;gBACD,IAAI,EAAE;oBACJ,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,gDAAgD;iBAC9D;gBACD,WAAW,EAAE;oBACX,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,4EAA4E;iBAC1F;aACF;YACD,QAAQ,EAAE,EAAE;SACb;KACF;IACD;QACE,IAAI,EAAE,cAAc;QACpB,WAAW,EAAE;;;;;;;;qCAQoB;QACjC,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,KAAK,EAAE;oBACL,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,6EAA6E;iBAC3F;gBACD,MAAM,EAAE;oBACN,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,wCAAwC;iBACtD;gBACD,WAAW,EAAE;oBACX,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,iCAAiC;iBAC/C;aACF;YACD,QAAQ,EAAE,CAAC,OAAO,CAAC;SACpB;KACF;IACD;QACE,IAAI,EAAE,oBAAoB;QAC1B,WAAW,EAAE;;;;qEAIoD;QACjE,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,WAAW,EAAE;oBACX,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,iEAAiE;iBAC/E;gBACD,UAAU,EAAE;oBACV,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,+DAA+D;iBAC7E;gBACD,MAAM,EAAE;oBACN,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,4CAA4C;iBAC1D;gBACD,gBAAgB,EAAE;oBAChB,IAAI,EAAE,QAAQ;oBACd,IAAI,EAAE,SAAgC;oBACtC,WAAW,EAAE,2DAA2D;iBACzE;gBACD,cAAc,EAAE;oBACd,IAAI,EAAE,SAAS;oBACf,WAAW,EAAE,uFAAuF;iBACrG;gBACD,WAAW,EAAE;oBACX,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,kCAAkC;iBAChD;aACF;YACD,QAAQ,EAAE,CAAC,aAAa,EAAE,YAAY,CAAC;SACxC;KACF;IACD;QACE,IAAI,EAAE,kBAAkB;QACxB,WAAW,EAAE;;sEAEqD;QAClE,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,cAAc,EAAE;oBACd,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,sFAAsF;iBACpG;aACF;YACD,QAAQ,EAAE,CAAC,gBAAgB,CAAC;SAC7B;KACF;CACF,CAAC;AAEF,gFAAgF;AAChF,eAAe;AACf,gFAAgF;AAEhF,MAAM,MAAM,GAAG,IAAI,MAAM,CACvB;IACE,IAAI,EAAE,0BAA0B;IAChC,OAAO,EAAE,OAAO;CACjB,EACD;IACE,YAAY,EAAE;QACZ,KAAK,EAAE,EAAE;KACV;CACF,CACF,CAAC;AAEF,gFAAgF;AAChF,mBAAmB;AACnB,gFAAgF;AAEhF,uBAAuB;AACvB,MAAM,CAAC,iBAAiB,CAAC,sBAAsB,EAAE,KAAK,IAAI,EAAE;IAC1D,QAAQ,CAAC,eAAe,CAAC,CAAC;IAC1B,OAAO,EAAE,KAAK,EAAE,CAAC;AACnB,CAAC,CAAC,CAAC;AAEH,oBAAoB;AACpB,MAAM,CAAC,iBAAiB,CAAC,qBAAqB,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;IAChE,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,GAAG,OAAO,CAAC,MAAM,CAAC;IAEjD,QAAQ,CAAC,gBAAgB,IAAI,EAAE,EAAE,IAAI,CAAC,CAAC;IAEvC,IAAI,CAAC;QACH,QAAQ,IAAI,EAAE,CAAC;YACb,KAAK,gBAAgB,CAAC,CAAC,CAAC;gBACtB,MAAM,MAAM,GAAG,IAAsC,CAAC;gBACtD,MAAM,MAAM,GAAG,MAAM,aAAa,CAAC,MAAM,CAAC,CAAC;gBAE3C,OAAO;oBACL,OAAO,EAAE;wBACP;4BACE,IAAI,EAAE,MAAM;4BACZ,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;yBACtC;qBACF;iBACF,CAAC;YACJ,CAAC;YAED,KAAK,cAAc,CAAC,CAAC,CAAC;gBACpB,MAAM,MAAM,GAAG,IAAoC,CAAC;gBACpD,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,MAAM,CAAC,CAAC;gBAEzC,OAAO;oBACL,OAAO,EAAE;wBACP;4BACE,IAAI,EAAE,MAAM;4BACZ,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;yBACtC;qBACF;iBACF,CAAC;YACJ,CAAC;YAED,KAAK,oBAAoB,CAAC,CAAC,CAAC;gBAC1B,MAAM,MAAM,GAAG,IAA0C,CAAC;gBAC1D,MAAM,MAAM,GAAG,MAAM,iBAAiB,CAAC,MAAM,CAAC,CAAC;gBAE/C,OAAO;oBACL,OAAO,EAAE;wBACP;4BACE,IAAI,EAAE,MAAM;4BACZ,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;yBACtC;qBACF;iBACF,CAAC;YACJ,CAAC;YAED,KAAK,kBAAkB,CAAC,CAAC,CAAC;gBACxB,MAAM,MAAM,GAAG,IAAuC,CAAC;gBAEvD,IAAI,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC;oBAC3B,OAAO;wBACL,OAAO,EAAE;4BACP;gCACE,IAAI,EAAE,MAAM;gCACZ,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oCACnB,OAAO,EAAE,KAAK;oCACd,KAAK,EAAE,4BAA4B;iCACpC,EAAE,IAAI,EAAE,CAAC,CAAC;6BACZ;yBACF;qBACF,CAAC;gBACJ,CAAC;gBAED,IAAI,CAAC;oBACH,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;oBAC3B,MAAM,MAAM,GAAG,MAAM,kBAAkB,CAAC,MAAM,CAAC,cAAc,EAAE,MAAM,CAAC,CAAC;oBAEvE,MAAM,MAAM,GAAsB;wBAChC,OAAO,EAAE,IAAI;wBACb,cAAc,EAAE,MAAM,CAAC,cAAc;wBACrC,IAAI,EAAE,MAAM,CAAC,IAAI,IAAI,KAAK;qBAC3B,CAAC;oBAEF,IAAI,MAAM,CAAC,IAAI,IAAI,MAAM,CAAC,QAAQ,EAAE,eAAe,EAAE,MAAM,EAAE,CAAC;wBAC5D,MAAM,CAAC,SAAS,GAAG,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC;oBAClE,CAAC;oBAED,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;wBACjB,MAAM,CAAC,OAAO,GAAG,KAAK,CAAC;wBACvB,MAAM,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC;wBACpC,MAAM,CAAC,cAAc,GAAG,eAAe,MAAM,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;oBAC7D,CAAC;oBAED,OAAO;wBACL,OAAO,EAAE;4BACP;gCACE,IAAI,EAAE,MAAM;gCACZ,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;6BACtC;yBACF;qBACF,CAAC;gBACJ,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,OAAO;wBACL,OAAO,EAAE;4BACP;gCACE,IAAI,EAAE,MAAM;gCACZ,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oCACnB,OAAO,EAAE,KAAK;oCACd,cAAc,EAAE,MAAM,CAAC,cAAc;oCACrC,IAAI,EAAE,KAAK;oCACX,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;iCAC9D,EAAE,IAAI,EAAE,CAAC,CAAC;6BACZ;yBACF;qBACF,CAAC;gBACJ,CAAC;YACH,CAAC;YAED;gBACE,OAAO;oBACL,OAAO,EAAE;wBACP;4BACE,IAAI,EAAE,MAAM;4BACZ,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;gCACnB,OAAO,EAAE,KAAK;gCACd,KAAK,EAAE,iBAAiB,IAAI,EAAE;6BAC/B,EAAE,IAAI,EAAE,CAAC,CAAC;yBACZ;qBACF;iBACF,CAAC;QACN,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,QAAQ,CAAC,0BAA0B,IAAI,EAAE,EAAE,KAAK,CAAC,CAAC;QAElD,OAAO;YACL,OAAO,EAAE;gBACP;oBACE,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;wBACnB,OAAO,EAAE,KAAK;wBACd,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;qBAC9D,EAAE,IAAI,EAAE,CAAC,CAAC;iBACZ;aACF;SACF,CAAC;IACJ,CAAC;AACH,CAAC,CAAC,CAAC;AAEH,gFAAgF;AAChF,eAAe;AACf,gFAAgF;AAEhF,KAAK,UAAU,IAAI;IACjB,OAAO,CAAC,uCAAuC,CAAC,CAAC;IAEjD,MAAM,SAAS,GAAG,IAAI,oBAAoB,EAAE,CAAC;IAC7C,MAAM,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IAEhC,OAAO,CAAC,4BAA4B,CAAC,CAAC;AACxC,CAAC;AAED,IAAI,EAAE,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;IACrB,QAAQ,CAAC,wBAAwB,EAAE,KAAK,CAAC,CAAC;IAC1C,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AAClB,CAAC,CAAC,CAAC"}

package/dist/tools/extend.d.ts

@@ -0,0 +1,14 @@
+/**
+ * Video Extension Tool for Veo 3.1
+ * Extends an existing video by 7 seconds
+ */
+import { type ExtendVideoParams, type VideoGenerationResult } from '../types/tools.js';
+/**
+ * Extend a video by 7 seconds using Veo 3.1 API
+ *
+ * Requirements:
+ * - Input video: 1-30 seconds, 24fps, 720p or 1080p
+ * - Output: 7 seconds extension at 720p
+ */
+export declare function extendVideo(params: ExtendVideoParams): Promise<VideoGenerationResult>;
+//# sourceMappingURL=extend.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"extend.d.ts","sourceRoot":"","sources":["../../src/tools/extend.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EAQL,KAAK,iBAAiB,EAGtB,KAAK,qBAAqB,EAE3B,MAAM,mBAAmB,CAAC;AAW3B;;;;;;GAMG;AACH,wBAAsB,WAAW,CAAC,MAAM,EAAE,iBAAiB,GAAG,OAAO,CAAC,qBAAqB,CAAC,CAqK3F"}

package/dist/tools/extend.js

@@ -0,0 +1,153 @@
+/**
+ * Video Extension Tool for Veo 3.1
+ * Extends an existing video by 7 seconds
+ */
+import { GEMINI_API_BASE_URL, DEFAULT_MODEL, EXTENSION_DURATION, EXTENSION_RESOLUTION, DEFAULT_POLL_INTERVAL, DEFAULT_MAX_POLL_ATTEMPTS, calculateCost } from '../types/tools.js';
+import { getApiKey, pollVideoResult, downloadAndSaveVideo, extractVideoFromResponse, readVideoAsBase64 } from '../utils/video.js';
+import { generateDefaultOutputPath, getDisplayPath } from '../utils/path.js';
+import { debugLog, errorLog } from '../utils/debug.js';
+/**
+ * Extend a video by 7 seconds using Veo 3.1 API
+ *
+ * Requirements:
+ * - Input video: 1-30 seconds, 24fps, 720p or 1080p
+ * - Output: 7 seconds extension at 720p
+ */
+export async function extendVideo(params) {
+    const apiKey = getApiKey();
+    // Validate required parameter
+    if (!params.video) {
+        return {
+            success: false,
+            error: 'Video parameter is required for video extension'
+        };
+    }
+    const model = DEFAULT_MODEL;
+    debugLog('Extending video with parameters', {
+        model,
+        hasPrompt: !!params.prompt,
+        videoSource: params.video.startsWith('gs://') ? 'GCS' : 'file/base64'
+    });
+    try {
+        // Build request instance
+        const instance = {};
+        if (params.prompt) {
+            instance.prompt = params.prompt;
+        }
+        // Handle video input
+        if (params.video.startsWith('gs://')) {
+            instance.video = { uri: params.video };
+        }
+        else {
+            const base64Data = await readVideoAsBase64(params.video);
+            instance.video = { bytesBase64Encoded: base64Data };
+        }
+        // Build request - extension has fixed parameters
+        const request = {
+            instances: [instance],
+            parameters: {
+            // Extension is always 7 seconds at 720p
+            // These may be ignored by the API but we include them for completeness
+            }
+        };
+        debugLog('API Request', request);
+        // Make API call
+        const url = `${GEMINI_API_BASE_URL}/models/${model}:predictLongRunning?key=${apiKey}`;
+        const response = await fetch(url, {
+            method: 'POST',
+            headers: {
+                'Content-Type': 'application/json'
+            },
+            body: JSON.stringify(request)
+        });
+        if (!response.ok) {
+            const errorText = await response.text();
+            errorLog('API request failed', { status: response.status, error: errorText });
+            if (response.status === 401) {
+                return {
+                    success: false,
+                    error: 'Authentication failed. Please check your GOOGLE_API_KEY.'
+                };
+            }
+            if (response.status === 403) {
+                return {
+                    success: false,
+                    error: 'Access denied. Your API key may not have access to Veo 3.1.'
+                };
+            }
+            if (response.status === 429) {
+                return {
+                    success: false,
+                    error: 'Rate limit exceeded. Please try again later.'
+                };
+            }
+            if (response.status === 400) {
+                return {
+                    success: false,
+                    error: `Invalid video for extension. Requirements: 1-30 seconds, 24fps, 720p or 1080p. Details: ${errorText}`
+                };
+            }
+            return {
+                success: false,
+                error: `API request failed: ${response.status} ${response.statusText} - ${errorText}`
+            };
+        }
+        const operationResponse = await response.json();
+        debugLog('Initial operation response', operationResponse);
+        const operationName = operationResponse.name;
+        // Calculate estimated cost (7 seconds at 720p, no audio for extensions)
+        const estimatedCost = calculateCost(model, EXTENSION_RESOLUTION, EXTENSION_DURATION, false);
+        // Poll for completion
+        const pollInterval = parseInt(process.env.VIDEO_POLL_INTERVAL || '') || DEFAULT_POLL_INTERVAL;
+        const maxAttempts = parseInt(process.env.VIDEO_MAX_POLL_ATTEMPTS || '') || DEFAULT_MAX_POLL_ATTEMPTS;
+        const finalResponse = await pollVideoResult(operationName, apiKey, pollInterval, maxAttempts);
+        // Check for errors
+        if (finalResponse.error) {
+            return {
+                success: false,
+                operation_name: operationName,
+                error: finalResponse.error.message,
+                failure_reason: `Error code: ${finalResponse.error.code}`
+            };
+        }
+        // Extract video data
+        const videoData = extractVideoFromResponse(finalResponse);
+        if (!videoData) {
+            return {
+                success: false,
+                operation_name: operationName,
+                error: 'No video was generated. The content may have been filtered by safety policies.',
+                estimated_cost: estimatedCost
+            };
+        }
+        // Download and save if output path specified
+        let videoPath;
+        let videoUrl = videoData.url;
+        if (params.output_path || process.env.OUTPUT_DIR) {
+            const outputPath = params.output_path || generateDefaultOutputPath(process.env.OUTPUT_DIR || './output', 'veo3_extended');
+            try {
+                videoPath = await downloadAndSaveVideo(videoData.url, videoData.base64, outputPath);
+                debugLog('Extended video saved', { path: videoPath });
+            }
+            catch (downloadError) {
+                errorLog('Failed to save extended video', downloadError);
+            }
+        }
+        return {
+            success: true,
+            operation_name: operationName,
+            video_url: videoUrl,
+            video_path: videoPath ? getDisplayPath(videoPath) : undefined,
+            duration_seconds: EXTENSION_DURATION,
+            estimated_cost: estimatedCost
+        };
+    }
+    catch (error) {
+        errorLog('Video extension failed', error);
+        return {
+            success: false,
+            error: error instanceof Error ? error.message : String(error)
+        };
+    }
+}
+//# sourceMappingURL=extend.js.map

package/dist/tools/extend.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"extend.js","sourceRoot":"","sources":["../../src/tools/extend.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EACL,mBAAmB,EACnB,aAAa,EACb,kBAAkB,EAClB,oBAAoB,EACpB,qBAAqB,EACrB,yBAAyB,EACzB,aAAa,EAMd,MAAM,mBAAmB,CAAC;AAC3B,OAAO,EACL,SAAS,EACT,eAAe,EACf,oBAAoB,EACpB,wBAAwB,EACxB,iBAAiB,EAClB,MAAM,mBAAmB,CAAC;AAC3B,OAAO,EAAE,yBAAyB,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC;AAC7E,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAEvD;;;;;;GAMG;AACH,MAAM,CAAC,KAAK,UAAU,WAAW,CAAC,MAAyB;IACzD,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;IAE3B,8BAA8B;IAC9B,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;QAClB,OAAO;YACL,OAAO,EAAE,KAAK;YACd,KAAK,EAAE,iDAAiD;SACzD,CAAC;IACJ,CAAC;IAED,MAAM,KAAK,GAAU,aAAa,CAAC;IAEnC,QAAQ,CAAC,iCAAiC,EAAE;QAC1C,KAAK;QACL,SAAS,EAAE,CAAC,CAAC,MAAM,CAAC,MAAM;QAC1B,WAAW,EAAE,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,aAAa;KACtE,CAAC,CAAC;IAEH,IAAI,CAAC;QACH,yBAAyB;QACzB,MAAM,QAAQ,GAAuC,EAAE,CAAC;QAExD,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;YAClB,QAAQ,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;QAClC,CAAC;QAED,qBAAqB;QACrB,IAAI,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;YACrC,QAAQ,CAAC,KAAK,GAAG,EAAE,GAAG,EAAE,MAAM,CAAC,KAAK,EAAE,CAAC;QACzC,CAAC;aAAM,CAAC;YACN,MAAM,UAAU,GAAG,MAAM,iBAAiB,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YACzD,QAAQ,CAAC,KAAK,GAAG,EAAE,kBAAkB,EAAE,UAAU,EAAE,CAAC;QACtD,CAAC;QAED,iDAAiD;QACjD,MAAM,OAAO,GAAuB;YAClC,SAAS,EAAE,CAAC,QAAQ,CAAC;YACrB,UAAU,EAAE;YACV,wCAAwC;YACxC,uEAAuE;aACxE;SACF,CAAC;QAEF,QAAQ,CAAC,aAAa,EAAE,OAAO,CAAC,CAAC;QAEjC,gBAAgB;QAChB,MAAM,GAAG,GAAG,GAAG,mBAAmB,WAAW,KAAK,2BAA2B,MAAM,EAAE,CAAC;QAEtF,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE;YAChC,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;aACnC;YACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC;SAC9B,CAAC,CAAC;QAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,SAAS,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;YACxC,QAAQ,CAAC,oBAAoB,EAAE,EAAE,MAAM,EAAE,QAAQ,CAAC,MAAM,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAC;YAE9E,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;gBAC5B,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,0DAA0D;iBAClE,CAAC;YACJ,CAAC;YAED,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;gBAC5B,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,6DAA6D;iBACrE,CAAC;YACJ,CAAC;YAED,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;gBAC5B,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,8CAA8C;iBACtD,CAAC;YACJ,CAAC;YAED,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;gBAC5B,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,2FAA2F,SAAS,EAAE;iBAC9G,CAAC;YACJ,CAAC;YAED,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,uBAAuB,QAAQ,CAAC,MAAM,IAAI,QAAQ,CAAC,UAAU,MAAM,SAAS,EAAE;aACtF,CAAC;QACJ,CAAC;QAED,MAAM,iBAAiB,GAAG,MAAM,QAAQ,CAAC,IAAI,EAA0B,CAAC;QACxE,QAAQ,CAAC,4BAA4B,EAAE,iBAAiB,CAAC,CAAC;QAE1D,MAAM,aAAa,GAAG,iBAAiB,CAAC,IAAI,CAAC;QAE7C,wEAAwE;QACxE,MAAM,aAAa,GAAG,aAAa,CAAC,KAAK,EAAE,oBAAoB,EAAE,kBAAkB,EAAE,KAAK,CAAC,CAAC;QAE5F,sBAAsB;QACtB,MAAM,YAAY,GAAG,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,mBAAmB,IAAI,EAAE,CAAC,IAAI,qBAAqB,CAAC;QAC9F,MAAM,WAAW,GAAG,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,uBAAuB,IAAI,EAAE,CAAC,IAAI,yBAAyB,CAAC;QAErG,MAAM,aAAa,GAAG,MAAM,eAAe,CAAC,aAAa,EAAE,MAAM,EAAE,YAAY,EAAE,WAAW,CAAC,CAAC;QAE9F,mBAAmB;QACnB,IAAI,aAAa,CAAC,KAAK,EAAE,CAAC;YACxB,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,cAAc,EAAE,aAAa;gBAC7B,KAAK,EAAE,aAAa,CAAC,KAAK,CAAC,OAAO;gBAClC,cAAc,EAAE,eAAe,aAAa,CAAC,KAAK,CAAC,IAAI,EAAE;aAC1D,CAAC;QACJ,CAAC;QAED,qBAAqB;QACrB,MAAM,SAAS,GAAG,wBAAwB,CAAC,aAAa,CAAC,CAAC;QAE1D,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,cAAc,EAAE,aAAa;gBAC7B,KAAK,EAAE,gFAAgF;gBACvF,cAAc,EAAE,aAAa;aAC9B,CAAC;QACJ,CAAC;QAED,6CAA6C;QAC7C,IAAI,SAA6B,CAAC;QAClC,IAAI,QAAQ,GAAG,SAAS,CAAC,GAAG,CAAC;QAE7B,IAAI,MAAM,CAAC,WAAW,IAAI,OAAO,CAAC,GAAG,CAAC,UAAU,EAAE,CAAC;YACjD,MAAM,UAAU,GAAG,MAAM,CAAC,WAAW,IAAI,yBAAyB,CAChE,OAAO,CAAC,GAAG,CAAC,UAAU,IAAI,UAAU,EACpC,eAAe,CAChB,CAAC;YAEF,IAAI,CAAC;gBACH,SAAS,GAAG,MAAM,oBAAoB,CAAC,SAAS,CAAC,GAAG,EAAE,SAAS,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;gBACpF,QAAQ,CAAC,sBAAsB,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,CAAC;YACxD,CAAC;YAAC,OAAO,aAAa,EAAE,CAAC;gBACvB,QAAQ,CAAC,+BAA+B,EAAE,aAAa,CAAC,CAAC;YAC3D,CAAC;QACH,CAAC;QAED,OAAO;YACL,OAAO,EAAE,IAAI;YACb,cAAc,EAAE,aAAa;YAC7B,SAAS,EAAE,QAAQ;YACnB,UAAU,EAAE,SAAS,CAAC,CAAC,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS;YAC7D,gBAAgB,EAAE,kBAAkB;YACpC,cAAc,EAAE,aAAa;SAC9B,CAAC;IAEJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,QAAQ,CAAC,wBAAwB,EAAE,KAAK,CAAC,CAAC;QAC1C,OAAO;YACL,OAAO,EAAE,KAAK;YACd,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;SAC9D,CAAC;IACJ,CAAC;AACH,CAAC"}

package/dist/tools/generate.d.ts

@@ -0,0 +1,10 @@
+/**
+ * Video Generation Tool for Veo 3.1
+ * Supports Text-to-Video, Image-to-Video, and Reference Images
+ */
+import { type GenerateVideoParams, type VideoGenerationResult } from '../types/tools.js';
+/**
+ * Generate a video using Veo 3.1 API
+ */
+export declare function generateVideo(params: GenerateVideoParams): Promise<VideoGenerationResult>;
+//# 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;;;GAGG;AAEH,OAAO,EAcL,KAAK,mBAAmB,EAGxB,KAAK,qBAAqB,EAG3B,MAAM,mBAAmB,CAAC;AAW3B;;GAEG;AACH,wBAAsB,aAAa,CAAC,MAAM,EAAE,mBAAmB,GAAG,OAAO,CAAC,qBAAqB,CAAC,CA2O/F"}