@mixio-pro/kalaasetu-mcp 1.0.12 → 1.0.13

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mixio-pro/kalaasetu-mcp",
-  "version": "1.0.12",
+  "version": "1.0.13",
   "description": "A powerful Model Context Protocol server providing AI tools for content generation and analysis",
   "type": "module",
   "module": "src/index.ts",

package/src/tools/image-to-video.ts

@@ -1,4 +1,3 @@
-import * as fs from "fs";
 import { GoogleAuth } from "google-auth-library";
 import { exec } from "child_process";
 import * as path from "path";
@@ -99,7 +98,9 @@ export const imageToVideo = {
     duration_seconds: z
       .string()
       .optional()
-      .describe("Video duration in seconds: '4', '6', or '8' (default: '6')"),
+      .describe(
+        "Video duration in seconds. MUST be one of: '4', '6', or '8' (default: '6'). Other values will be rejected by Vertex AI."
+      ),
     resolution: z
       .string()
       .optional()
@@ -164,6 +165,32 @@ export const imageToVideo = {
     const location = args.location_id || "us-central1";
     const modelId = args.model_id || "veo-3.1-fast-generate-preview";
 
+    // Validate and parse duration_seconds - snap to nearest 4, 6, or 8
+    let durationSeconds = parseInt(args.duration_seconds || "6");
+    if (isNaN(durationSeconds)) durationSeconds = 6;
+
+    const validDurations = [4, 6, 8];
+    // Find nearest valid duration
+    durationSeconds = validDurations.reduce((prev, curr) => {
+      return Math.abs(curr - durationSeconds) < Math.abs(prev - durationSeconds)
+        ? curr
+        : prev;
+    });
+
+    // Tie-breaking: if equidistant (e.g. 5), the reduce above keeps the first one (4) because < is strict.
+    // However, user requested "nearest duration with the ceil", effectively meaning round up if equidistant.
+    // Let's explicitly handle the equidistant cases or just use a custom finder.
+    // 5 -> equidistant to 4 and 6. "With ceil" implies 6.
+    // 7 -> equidistant to 6 and 8. "With ceil" implies 8.
+
+    // Simpler logic for these specific values:
+    if (durationSeconds === 4 && parseInt(args.duration_seconds || "6") === 5) {
+      durationSeconds = 6;
+    }
+    if (durationSeconds === 6 && parseInt(args.duration_seconds || "6") === 7) {
+      durationSeconds = 8;
+    }
+
     const token = await fetchAccessToken();
 
     const url = `https://${location}-aiplatform.googleapis.com/v1/projects/${projectId}/locations/${location}/publishers/google/models/${modelId}:predictLongRunning`;
@@ -242,7 +269,7 @@ export const imageToVideo = {
 
     const parameters: any = {
       aspectRatio: args.aspect_ratio || "9:16",
-      durationSeconds: parseInt(args.duration_seconds || "6") || 6,
+      durationSeconds: durationSeconds,
       resolution: args.resolution || "720p",
       negativePrompt: args.negative_prompt,
       generateAudio: args.generate_audio || false,
@@ -302,14 +329,15 @@ export const imageToVideo = {
       [];
     const saveVideo = async (base64: string, index: number) => {
       if (!base64) return;
-      
+
       // Use provided output path or generate default with timestamp
       let filePath: string;
       if (args.output_path) {
         // User provided path - use as-is for first video, add index for subsequent
-        filePath = index === 0
-          ? args.output_path
-          : args.output_path.replace(/\.mp4$/i, `_${index}.mp4`);
+        filePath =
+          index === 0
+            ? args.output_path
+            : args.output_path.replace(/\.mp4$/i, `_${index}.mp4`);
       } else {
         // No path provided - generate timestamped default
         const defaultName = `video_output${index > 0 ? `_${index}` : ""}.mp4`;