@mixio-pro/kalaasetu-mcp 2.1.0 → 2.1.2-beta

package/src/tools/get-status.ts

@@ -171,55 +171,59 @@ export const getGenerationStatus = {
     resume_endpoint: string;
     source?: "fal" | "vertex" | "auto";
   }) => {
-    return safeToolExecute(async () => {
-      const { resume_endpoint, source = "auto" } = args;
-      const project_id = "mixio-pro";
-      const location_id = "us-central1";
-
-      // Auto-detect source based on resume_endpoint format
-      let detectedSource = source;
-      if (source === "auto") {
-        if (
-          resume_endpoint.startsWith("https://queue.fal.run") ||
-          resume_endpoint.startsWith("https://fal.run")
-        ) {
-          detectedSource = "fal";
-        } else {
-          detectedSource = "vertex";
+    return safeToolExecute(
+      async () => {
+        const { resume_endpoint, source = "auto" } = args;
+        const project_id = "mixio-pro";
+        const location_id = "us-central1";
+
+        // Auto-detect source based on resume_endpoint format
+        let detectedSource = source;
+        if (source === "auto") {
+          if (
+            resume_endpoint.startsWith("https://queue.fal.run") ||
+            resume_endpoint.startsWith("https://fal.run")
+          ) {
+            detectedSource = "fal";
+          } else {
+            detectedSource = "vertex";
+          }
         }
-      }
 
-      let result: any;
+        let result: any;
 
-      if (detectedSource === "fal") {
-        result = await checkFalStatus(resume_endpoint);
-      } else {
-        result = await checkVertexStatus(resume_endpoint);
-      }
+        if (detectedSource === "fal") {
+          result = await checkFalStatus(resume_endpoint);
+        } else {
+          result = await checkVertexStatus(resume_endpoint);
+        }
 
-      // Normalize the response
-      const status =
-        (result as any).status ||
-        ((result as any).done ? "COMPLETED" : "IN_PROGRESS");
-
-      const safeResult = sanitizeResponse(result);
-
-      return JSON.stringify(
-        {
-          source: detectedSource,
-          status,
-          resume_endpoint,
-          result: safeResult,
-          message:
-            status === "COMPLETED"
-              ? "Generation completed! Check 'result.response.saved_videos' for the video URLs."
-              : status === "FAILED"
-                ? "Generation failed. Check the 'result' field for error details."
-                : "Generation is still in progress. Call this tool again with the same resume_endpoint to check later.",
-        },
-        null,
-        2,
-      );
-    }, "get_generation_status");
+        // Normalize the response
+        const status =
+          (result as any).status ||
+          ((result as any).done ? "COMPLETED" : "IN_PROGRESS");
+
+        const safeResult = sanitizeResponse(result);
+
+        return JSON.stringify(
+          {
+            source: detectedSource,
+            status,
+            resume_endpoint,
+            result: safeResult,
+            message:
+              status === "COMPLETED"
+                ? "Generation completed! Check 'result.response.saved_videos' for the video URLs."
+                : status === "FAILED"
+                  ? "Generation failed. Check the 'result' field for error details."
+                  : "Generation is still in progress. Call this tool again with the same resume_endpoint to check later.",
+          },
+          null,
+          2,
+        );
+      },
+      "get_generation_status",
+      { toolName: "get_generation_status" },
+    );
   },
 };