@mixio-pro/kalaasetu-mcp 2.0.1-beta → 2.0.2-beta

package/package.json

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

package/src/index.ts

@@ -36,11 +36,16 @@ server.addTool(imageToVideo);
 // server.addTool(perplexityImages);
 // server.addTool(perplexityVideos);
 
+import { getGenerationStatus } from "./tools/get-status";
+
 // Fal AI Tools
 server.addTool(falListPresets);
 server.addTool(falGenerate);
 server.addTool(falUploadFile);
 
+// Unified Status Tool (works with both FAL and Vertex AI)
+server.addTool(getGenerationStatus);
+
 server.start({
   transportType: "stdio",
 });

package/src/tools/fal/generate.ts

@@ -119,7 +119,7 @@ export const falGenerate = {
           "Use the 'request_id' returned in an 'IN_PROGRESS' response or after a timeout error."
       ),
   }),
-  timeoutMs: 600000, // 10 minutes max
+  timeoutMs: 90000, // 90 seconds MCP timeout (internal timeout is 60s)
   execute: async (
     args: {
       preset_name?: string;
@@ -362,7 +362,7 @@ export const falGenerate = {
       }
 
       const startTime = Date.now();
-      const MAX_POLL_TIME = 600000; // 10 minutes
+      const MAX_POLL_TIME = 60000; // 60 seconds internal timeout - then return resume_id
       let pollCount = 0;
       const POLL_INTERVAL = 3000;
 

package/src/tools/get-status.ts

@@ -0,0 +1,174 @@
+/**
+ * Unified status checking tool for FAL and Vertex AI generation operations.
+ * This tool allows resuming/checking status of operations that timed out.
+ */
+
+import { z } from "zod";
+import { safeToolExecute } from "../utils/tool-wrapper";
+import { getGoogleAccessToken } from "../utils/google-auth";
+
+const FAL_KEY = process.env.FAL_KEY;
+
+/**
+ * Check FAL generation status using the status URL
+ */
+async function checkFalStatus(statusUrl: string): Promise<any> {
+  if (!FAL_KEY) {
+    throw new Error("FAL_KEY environment variable not set");
+  }
+
+  const response = await fetch(statusUrl, {
+    method: "GET",
+    headers: {
+      Authorization: `Key ${FAL_KEY}`,
+      "Content-Type": "application/json",
+    },
+  });
+
+  if (!response.ok) {
+    const errorText = await response.text();
+    throw new Error(`FAL API error [${response.status}]: ${errorText}`);
+  }
+
+  const statusResult = (await response.json()) as { status?: string };
+
+  if (statusResult.status === "COMPLETED") {
+    // Fetch the actual result
+    const responseUrl = statusUrl.replace(/\/status$/, "");
+    const resultResponse = await fetch(responseUrl, {
+      method: "GET",
+      headers: {
+        Authorization: `Key ${FAL_KEY}`,
+        "Content-Type": "application/json",
+      },
+    });
+
+    if (resultResponse.ok) {
+      return await resultResponse.json();
+    }
+  }
+
+  return statusResult;
+}
+
+/**
+ * Check Vertex AI operation status
+ */
+async function checkVertexStatus(
+  operationName: string,
+  projectId: string,
+  locationId: string
+): Promise<any> {
+  const accessToken = await getGoogleAccessToken();
+
+  const operationsUrl = `https://${locationId}-aiplatform.googleapis.com/v1/projects/${projectId}/locations/${locationId}/publishers/google/models/veo-3.1-generate-preview/operations/${operationName}`;
+
+  const response = await fetch(operationsUrl, {
+    method: "GET",
+    headers: {
+      Authorization: `Bearer ${accessToken}`,
+      "Content-Type": "application/json",
+    },
+  });
+
+  if (!response.ok) {
+    const errorText = await response.text();
+    throw new Error(`Vertex AI API error [${response.status}]: ${errorText}`);
+  }
+
+  return await response.json();
+}
+
+export const getGenerationStatus = {
+  name: "get_generation_status",
+  description:
+    "Check the status or retrieve the result of a generation operation that was started by 'fal_generate' or 'generateVideoi2v'. " +
+    "Use this when the original generation tool returned an 'IN_PROGRESS' status with a 'resume_id'. " +
+    "Pass the resume_id exactly as it was returned. " +
+    "For FAL operations, the resume_id is a full URL. " +
+    "For Vertex AI operations, the resume_id is an operation name.",
+  parameters: z.object({
+    resume_id: z
+      .string()
+      .describe(
+        "The resume_id returned by the original generation tool. " +
+          "For FAL: This is a full URL (starts with 'https://queue.fal.run/...'). " +
+          "For Vertex AI: This is an operation name."
+      ),
+    source: z
+      .enum(["fal", "vertex", "auto"])
+      .optional()
+      .default("auto")
+      .describe(
+        "Source of the operation: 'fal' for FAL AI, 'vertex' for Google Vertex AI, or 'auto' to auto-detect based on resume_id format."
+      ),
+    project_id: z
+      .string()
+      .optional()
+      .default("mixio-pro")
+      .describe("GCP Project ID (only needed for Vertex AI operations)."),
+    location_id: z
+      .string()
+      .optional()
+      .default("us-central1")
+      .describe("GCP region (only needed for Vertex AI operations)."),
+  }),
+  timeoutMs: 30000, // 30 seconds for status check
+  execute: async (args: {
+    resume_id: string;
+    source?: "fal" | "vertex" | "auto";
+    project_id?: string;
+    location_id?: string;
+  }) => {
+    return safeToolExecute(async () => {
+      const {
+        resume_id,
+        source = "auto",
+        project_id = "mixio-pro",
+        location_id = "us-central1",
+      } = args;
+
+      // Auto-detect source based on resume_id format
+      let detectedSource = source;
+      if (source === "auto") {
+        if (
+          resume_id.startsWith("https://queue.fal.run") ||
+          resume_id.startsWith("https://fal.run")
+        ) {
+          detectedSource = "fal";
+        } else {
+          detectedSource = "vertex";
+        }
+      }
+
+      let result: any;
+
+      if (detectedSource === "fal") {
+        result = await checkFalStatus(resume_id);
+      } else {
+        result = await checkVertexStatus(resume_id, project_id, location_id);
+      }
+
+      // Normalize the response
+      const status =
+        result.status || (result.done ? "COMPLETED" : "IN_PROGRESS");
+
+      return JSON.stringify(
+        {
+          source: detectedSource,
+          status,
+          resume_id,
+          result,
+          message:
+            status === "COMPLETED"
+              ? "Generation completed! The result is included in the 'result' field."
+              : status === "FAILED"
+              ? "Generation failed. Check the 'result' field for error details."
+              : "Generation is still in progress. Call this tool again with the same resume_id to check later.",
+        },
+        null,
+        2
+      );
+    }, "get_generation_status");
+  },
+};

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

@@ -152,7 +152,7 @@ export const imageToVideo = {
           "When using Veo, setting this to 'veo' (or setting auto_enhance=true) will trigger the LLM-based enhancer."
       ),
   }),
-  timeoutMs: 1200000, // 20 minutes
+  timeoutMs: 90000, // 90 seconds MCP timeout (internal timeout is 60s)
   async execute(
     args: {
       prompt?: string;
@@ -477,7 +477,7 @@ export const imageToVideo = {
       // Resume_id was already streamed, so if MCP client times out the LLM still has it
       let done = current ? !!current.done || !!current.response : false;
       const startTime = Date.now();
-      const MAX_POLL_TIME = 600000; // 10 minutes - full tool timeout is 20 mins
+      const MAX_POLL_TIME = 60000; // 60 seconds internal timeout - then return resume_id
 
       while (!done && Date.now() - startTime < MAX_POLL_TIME) {
         await wait(10000); // 10 second intervals