@felores/kie-ai-mcp-server 3.0.0 → 3.0.1

package/dist/index.js

@@ -5,7 +5,7 @@ import { CallToolRequestSchema, ErrorCode, ListToolsRequestSchema, ListResources
 import { KieAiClient } from "./kie-ai-client.js";
 import { TaskDatabase } from "./database.js";
 import { z } from "zod";
-import { NanoBananaImageSchema, Veo3GenerateSchema, SunoGenerateSchema, ElevenLabsTTSSchema, ElevenLabsSoundEffectsSchema, ByteDanceSeedanceVideoSchema, ByteDanceSeedreamImageSchema, QwenImageSchema, RunwayAlephVideoSchema, WanVideoSchema, MidjourneyGenerateSchema, OpenAI4oImageSchema, FluxKontextImageSchema, RecraftRemoveBackgroundSchema, IdeogramReframeSchema, KlingVideoSchema, HailuoVideoSchema, SoraVideoSchema, Flux2ImageSchema, WanAnimateSchema, ZImageSchema, GrokImagineSchema, InfiniTalkSchema, KlingAvatarSchema, } from "./types.js";
+import { NanoBananaImageSchema, Veo3GenerateSchema, SunoGenerateSchema, ElevenLabsTTSSchema, ElevenLabsSoundEffectsSchema, ByteDanceSeedanceVideoSchema, ByteDanceSeedreamImageSchema, QwenImageSchema, RunwayAlephVideoSchema, WanVideoSchema, MidjourneyGenerateSchema, OpenAI4oImageSchema, FluxKontextImageSchema, RecraftRemoveBackgroundSchema, IdeogramReframeSchema, KlingVideoSchema, HailuoVideoSchema, SoraVideoSchema, Flux2ImageSchema, WanAnimateSchema, ZImageSchema, GrokImagineSchema, InfiniTalkSchema, KlingAvatarSchema, TopazUpscaleImageSchema, } from "./types.js";
 class KieAiMcpServer {
     server;
     client;
@@ -21,6 +21,7 @@ class KieAiMcpServer {
             "flux_kontext_image",
             "flux2_image",
             "z_image",
+            "topaz_upscale_image",
             "recraft_remove_background",
             "ideogram_reframe",
             "midjourney_generate", // Also generates images (6 modes: txt2img, img2img, style ref, omni ref, video SD/HD)
@@ -52,7 +53,7 @@ class KieAiMcpServer {
     constructor() {
         this.server = new Server({
             name: "kie-ai-mcp-server",
-            version: "3.0.0",
+            version: "3.0.1",
         });
         // Initialize client with config from environment
         this.config = {
@@ -1405,6 +1406,32 @@ class KieAiMcpServer {
                         required: ["prompt"],
                     },
                 },
+                {
+                    name: "topaz_upscale_image",
+                    description: "Upscale and enhance images using Topaz Labs AI upscaler. Increases resolution with high-fidelity detail restoration, natural texture reconstruction, and improved clarity. Supports 1x-8x upscaling (max output 20,000px per side). Pricing: 10 credits (≤2K), 20 credits (4K), 40 credits (8K).",
+                    inputSchema: {
+                        type: "object",
+                        properties: {
+                            image_url: {
+                                type: "string",
+                                description: "URL of image to upscale (JPEG, PNG, WEBP, max 10MB)",
+                                format: "uri",
+                            },
+                            upscale_factor: {
+                                type: "string",
+                                description: "Upscale factor: 1x (enhance only), 2x (default), 4x, or 8x. Max output dimension is 20,000px.",
+                                enum: ["1", "2", "4", "8"],
+                                default: "2",
+                            },
+                            callBackUrl: {
+                                type: "string",
+                                description: "Optional: URL for task completion notifications (uses KIE_AI_CALLBACK_URL env var if not provided)",
+                                format: "uri",
+                            },
+                        },
+                        required: ["image_url"],
+                    },
+                },
                 {
                     name: "recraft_remove_background",
                     description: "Remove backgrounds from images using Recraft AI background removal model",
@@ -1837,6 +1864,8 @@ class KieAiMcpServer {
                         return await this.handleRunwayAlephVideo(args);
                     case "wan_video":
                         return await this.handleWanVideo(args);
+                    case "topaz_upscale_image":
+                        return await this.handleTopazUpscaleImage(args);
                     case "recraft_remove_background":
                         return await this.handleRecraftRemoveBackground(args);
                     case "ideogram_reframe":
@@ -2020,6 +2049,16 @@ class KieAiMcpServer {
                         },
                     },
                     // Specialized Tools
+                    {
+                        uri: "kie://models/topaz-upscale",
+                        name: "Topaz Image Upscale",
+                        description: "AI-powered image upscaling with detail restoration (1x-8x)",
+                        mimeType: "text/markdown",
+                        annotations: {
+                            audience: ["assistant"],
+                            priority: 0.5,
+                        },
+                    },
                     {
                         uri: "kie://models/recraft-bg-removal",
                         name: "Recraft Background Removal",
@@ -2429,6 +2468,33 @@ class KieAiMcpServer {
                             errorMessage = apiData.errorMessage;
                         }
                     }
+                    else if (localTask?.api_type === "topaz-upscale") {
+                        // Topaz Image Upscale-specific status mapping
+                        const state = apiData.state;
+                        if (state === "success")
+                            status = "completed";
+                        else if (state === "fail")
+                            status = "failed";
+                        else if (state === "waiting")
+                            status = "processing";
+                        // Parse resultJson for Topaz Image Upscale
+                        if (apiData.resultJson) {
+                            try {
+                                parsedResult = JSON.parse(apiData.resultJson);
+                                if (parsedResult.resultUrls &&
+                                    parsedResult.resultUrls.length > 0) {
+                                    resultUrl = parsedResult.resultUrls[0];
+                                }
+                            }
+                            catch (e) {
+                                // Invalid JSON in resultJson
+                            }
+                        }
+                        // Extract error message for Topaz Image Upscale
+                        if (apiData.failMsg) {
+                            errorMessage = apiData.failMsg;
+                        }
+                    }
                     else if (localTask?.api_type === "recraft-remove-background") {
                         // Recraft Remove Background-specific status mapping
                         const state = apiData.state;
@@ -2530,6 +2596,7 @@ class KieAiMcpServer {
                     "qwen-image",
                     "openai-4o-image",
                     "flux-kontext-image",
+                    "topaz-upscale",
                     "recraft-remove-background",
                     "ideogram-reframe",
                     "midjourney",
@@ -3924,6 +3991,61 @@ class KieAiMcpServer {
             });
         }
     }
+    async handleTopazUpscaleImage(args) {
+        try {
+            const request = TopazUpscaleImageSchema.parse(args);
+            // Use intelligent callback URL fallback
+            request.callBackUrl = this.getCallbackUrl(request.callBackUrl);
+            const response = await this.client.generateTopazUpscaleImage(request);
+            if (response.code === 200 && response.data?.taskId) {
+                // Store task in database
+                await this.db.createTask({
+                    task_id: response.data.taskId,
+                    api_type: "topaz-upscale",
+                    status: "pending",
+                });
+                return {
+                    content: [
+                        {
+                            type: "text",
+                            text: JSON.stringify({
+                                success: true,
+                                task_id: response.data.taskId,
+                                message: "Topaz Image Upscale task created successfully",
+                                parameters: {
+                                    image_url: request.image_url,
+                                    upscale_factor: request.upscale_factor,
+                                    callBackUrl: request.callBackUrl,
+                                },
+                                next_steps: [
+                                    "Use get_task_status to check generation progress",
+                                    "Task completion will be sent to the provided callback URL",
+                                    "Upscaling typically takes 30-90 seconds depending on image size and upscale factor",
+                                ],
+                            }, null, 2),
+                        },
+                    ],
+                };
+            }
+            else {
+                throw new Error(response.msg || "Failed to create Topaz Image Upscale task");
+            }
+        }
+        catch (error) {
+            if (error instanceof z.ZodError) {
+                return this.formatError("topaz_upscale_image", error, {
+                    image_url: "Required: URL of image to upscale (JPEG, PNG, WEBP, max 10MB)",
+                    upscale_factor: 'Optional: Upscale factor "1", "2" (default), "4", or "8". Max output dimension is 20,000px.',
+                    callBackUrl: "Optional: URL for task completion notifications (uses KIE_AI_CALLBACK_URL env var if not provided)",
+                });
+            }
+            return this.formatError("topaz_upscale_image", error, {
+                image_url: "Required: URL of image to upscale",
+                upscale_factor: 'Optional: Upscale factor "1", "2", "4", or "8"',
+                callBackUrl: "Optional: URL for task completion notifications",
+            });
+        }
+    }
     async handleRecraftRemoveBackground(args) {
         try {
             const request = RecraftRemoveBackgroundSchema.parse(args);
@@ -4537,6 +4659,12 @@ class KieAiMcpServer {
                 category: "image",
                 quality: "premium",
             },
+            {
+                name: "topaz_upscale_image",
+                status: "available",
+                category: "image",
+                quality: "professional",
+            },
             {
                 name: "recraft_remove_background",
                 status: "available",
@@ -5025,6 +5153,7 @@ These guidelines ensure optimal balance between quality requirements and cost ma
             "flux-kontext": "flux_kontext_image.md",
             "openai-4o-image": "openai_4o-image.md",
             "nano-banana": "google_nano-banana.md",
+            "topaz-upscale": "topaz_image-upscale.md",
             "recraft-bg-removal": "recraft_remove_background.md",
             "ideogram-reframe": "ideogram_reframe_image.md",
             // Video models

package/dist/kie-ai-client.d.ts

@@ -1,4 +1,4 @@
-import { KieAiConfig, KieAiResponse, NanoBananaImageRequest, Veo3GenerateRequest, SunoGenerateRequest, ElevenLabsTTSRequest, ElevenLabsSoundEffectsRequest, ByteDanceSeedanceVideoRequest, RunwayAlephVideoRequest, WanVideoRequest, ByteDanceSeedreamImageRequest, QwenImageRequest, MidjourneyGenerateRequest, OpenAI4oImageRequest, FluxKontextImageRequest, RecraftRemoveBackgroundRequest, IdeogramReframeRequest, KlingVideoRequest, HailuoVideoRequest, SoraVideoRequest, Flux2ImageRequest, WanAnimateRequest, ZImageRequest, GrokImagineRequest, InfiniTalkRequest, KlingAvatarRequest, ImageResponse, TaskResponse } from "./types.js";
+import { KieAiConfig, KieAiResponse, NanoBananaImageRequest, Veo3GenerateRequest, SunoGenerateRequest, ElevenLabsTTSRequest, ElevenLabsSoundEffectsRequest, ByteDanceSeedanceVideoRequest, RunwayAlephVideoRequest, WanVideoRequest, ByteDanceSeedreamImageRequest, QwenImageRequest, MidjourneyGenerateRequest, OpenAI4oImageRequest, FluxKontextImageRequest, RecraftRemoveBackgroundRequest, IdeogramReframeRequest, KlingVideoRequest, HailuoVideoRequest, SoraVideoRequest, Flux2ImageRequest, WanAnimateRequest, ZImageRequest, GrokImagineRequest, InfiniTalkRequest, KlingAvatarRequest, TopazUpscaleImageRequest, ImageResponse, TaskResponse } from "./types.js";
 export declare class KieAiClient {
     private config;
     constructor(config: KieAiConfig);
@@ -28,5 +28,6 @@ export declare class KieAiClient {
     generateZImage(request: ZImageRequest): Promise<KieAiResponse<TaskResponse>>;
     generateGrokImagine(request: GrokImagineRequest): Promise<KieAiResponse<TaskResponse>>;
     generateInfiniTalk(request: InfiniTalkRequest): Promise<KieAiResponse<TaskResponse>>;
+    generateTopazUpscaleImage(request: TopazUpscaleImageRequest): Promise<KieAiResponse<TaskResponse>>;
     generateKlingAvatar(request: KlingAvatarRequest): Promise<KieAiResponse<TaskResponse>>;
 }

package/dist/kie-ai-client.js

@@ -84,7 +84,8 @@ export class KieAiClient {
             apiType === "hailuo" ||
             apiType === "sora-video" ||
             apiType === "flux2-image" ||
-            apiType === "wan-animate") {
+            apiType === "wan-animate" ||
+            apiType === "topaz-upscale") {
             return this.makeRequest(`/jobs/recordInfo?taskId=${taskId}`, "GET");
         }
         else if (apiType === "runway-aleph-video") {
@@ -773,6 +774,17 @@ export class KieAiClient {
         };
         return this.makeRequest("/jobs/createTask", "POST", jobRequest);
     }
+    async generateTopazUpscaleImage(request) {
+        const jobRequest = {
+            model: "topaz/image-upscale",
+            input: {
+                image_url: request.image_url,
+                upscale_factor: request.upscale_factor || "2",
+            },
+            callBackUrl: request.callBackUrl || process.env.KIE_AI_CALLBACK_URL,
+        };
+        return this.makeRequest("/jobs/createTask", "POST", jobRequest);
+    }
     async generateKlingAvatar(request) {
         const quality = request.quality || "standard";
         const model = quality === "pro" ? "kling/ai-avatar-v1-pro" : "kling/v1-avatar-standard";

package/dist/types.d.ts

@@ -814,6 +814,20 @@ export declare const FluxKontextImageSchema: z.ZodEffects<z.ZodObject<{
     safetyTolerance?: number | undefined;
 }>;
 export type FluxKontextImageRequest = z.infer<typeof FluxKontextImageSchema>;
+export declare const TopazUpscaleImageSchema: z.ZodObject<{
+    image_url: z.ZodString;
+    upscale_factor: z.ZodDefault<z.ZodEnum<["1", "2", "4", "8"]>>;
+    callBackUrl: z.ZodOptional<z.ZodString>;
+}, "strip", z.ZodTypeAny, {
+    image_url: string;
+    upscale_factor: "1" | "2" | "4" | "8";
+    callBackUrl?: string | undefined;
+}, {
+    image_url: string;
+    callBackUrl?: string | undefined;
+    upscale_factor?: "1" | "2" | "4" | "8" | undefined;
+}>;
+export type TopazUpscaleImageRequest = z.infer<typeof TopazUpscaleImageSchema>;
 export declare const RecraftRemoveBackgroundSchema: z.ZodEffects<z.ZodObject<{
     image: z.ZodString;
     callBackUrl: z.ZodOptional<z.ZodString>;
@@ -1152,7 +1166,7 @@ export interface TaskResponse {
 export interface TaskRecord {
     id?: number;
     task_id: string;
-    api_type: "nano-banana" | "nano-banana-edit" | "nano-banana-image" | "veo3" | "suno" | "elevenlabs-tts" | "elevenlabs-sound-effects" | "bytedance-seedance-video" | "runway-aleph-video" | "wan-video" | "bytedance-seedream-image" | "qwen-image" | "midjourney" | "openai-4o-image" | "flux-kontext-image" | "recraft-remove-background" | "ideogram-reframe" | "kling-3.0-video" | "hailuo" | "sora-video" | "flux2-image" | "wan-animate" | "z-image" | "grok-imagine" | "infinitalk" | "kling-avatar";
+    api_type: "nano-banana" | "nano-banana-edit" | "nano-banana-image" | "veo3" | "suno" | "elevenlabs-tts" | "elevenlabs-sound-effects" | "bytedance-seedance-video" | "runway-aleph-video" | "wan-video" | "bytedance-seedream-image" | "qwen-image" | "midjourney" | "openai-4o-image" | "flux-kontext-image" | "recraft-remove-background" | "ideogram-reframe" | "kling-3.0-video" | "hailuo" | "sora-video" | "flux2-image" | "wan-animate" | "z-image" | "grok-imagine" | "infinitalk" | "kling-avatar" | "topaz-upscale";
     status: "pending" | "processing" | "completed" | "failed";
     created_at: string;
     updated_at: string;

package/dist/types.js

@@ -557,6 +557,12 @@ export const FluxKontextImageSchema = z
     message: "For image editing mode, safetyTolerance must be between 0 and 2",
     path: ["safetyTolerance"],
 });
+// Topaz Image Upscale - AI-powered image enhancement and upscaling
+export const TopazUpscaleImageSchema = z.object({
+    image_url: z.string().url(),
+    upscale_factor: z.enum(["1", "2", "4", "8"]).default("2"),
+    callBackUrl: z.string().url().optional(),
+});
 // Recraft Remove Background
 export const RecraftRemoveBackgroundSchema = z
     .object({

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@felores/kie-ai-mcp-server",
-  "version": "3.0.0",
-  "description": "MCP server for Kie.ai APIs (23 AI tools: Nano Banana image generation/editing/upscaling, Veo3 video generation, Suno music generation, ElevenLabs TTS/sound effects, ByteDance Seedance/Seedream video/image, Qwen image generation/editing, Runway Aleph video transformation, Midjourney AI generation, Wan video generation, Hailuo video generation, Kling video generation, OpenAI 4o image generation/editing/variants, Flux Kontext image generation/editing, Recraft background removal, Ideogram image reframing, Sora 2 video generation)",
+  "version": "3.0.1",
+  "description": "MCP server for Kie.ai APIs (24 AI tools: Nano Banana image generation/editing/upscaling, Veo3 video generation, Suno music generation, ElevenLabs TTS/sound effects, ByteDance Seedance/Seedream video/image, Qwen image generation/editing, Runway Aleph video transformation, Midjourney AI generation, Wan video generation, Hailuo video generation, Kling video generation, OpenAI 4o image generation/editing/variants, Flux Kontext image generation/editing, Recraft background removal, Ideogram image reframing, Sora 2 video generation, Topaz image upscaling)",
   "main": "dist/index.js",
   "bin": {
     "kie-ai-mcp-server": "dist/index.js"
@@ -37,6 +37,7 @@
     "flux-2",
     "recraft",
     "ideogram",
+    "topaz",
     "model-context-protocol",
     "ai-api",
     "unified-ai"