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

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.1.0",
         });
         // Initialize client with config from environment
         this.config = {
@@ -613,66 +614,92 @@ class KieAiMcpServer {
                 },
                 {
                     name: "bytedance_seedance_video",
-                    description: "Generate videos using ByteDance Seedance models (unified tool for both text-to-video and image-to-video)",
+                    description: "Generate videos with ByteDance Seedance 2.0 — multimodal inputs (image/video/audio references), native audio generation, standard and fast modes",
                     inputSchema: {
                         type: "object",
                         properties: {
                             prompt: {
                                 type: "string",
-                                description: "Text prompt for video generation (max 10000 characters)",
-                                minLength: 1,
-                                maxLength: 10000,
+                                description: "Text prompt for video generation (3-20000 characters)",
+                                minLength: 3,
+                                maxLength: 20000,
                             },
-                            image_url: {
+                            mode: {
                                 type: "string",
-                                description: "URL of input image for image-to-video generation (optional - if not provided, uses text-to-video)",
+                                description: "Generation mode — standard (seedance-2, higher quality) or fast (seedance-2-fast, iterative workflows)",
+                                enum: ["standard", "fast"],
+                                default: "standard",
+                            },
+                            first_frame_url: {
+                                type: "string",
+                                description: "URL of image to use as the first frame (optional)",
                                 format: "uri",
                             },
-                            quality: {
+                            last_frame_url: {
                                 type: "string",
-                                description: "Model quality level - lite for faster generation, pro for higher quality",
-                                enum: ["lite", "pro"],
-                                default: "lite",
+                                description: "URL of image to use as the last frame (optional)",
+                                format: "uri",
+                            },
+                            reference_image_urls: {
+                                type: "array",
+                                description: "Reference images for style/subject guidance (up to 9)",
+                                items: { type: "string", format: "uri" },
+                                maxItems: 9,
+                            },
+                            reference_video_urls: {
+                                type: "array",
+                                description: "Reference videos for motion/style guidance (up to 3)",
+                                items: { type: "string", format: "uri" },
+                                maxItems: 3,
+                            },
+                            reference_audio_urls: {
+                                type: "array",
+                                description: "Reference audio for sound-guided generation (up to 3)",
+                                items: { type: "string", format: "uri" },
+                                maxItems: 3,
                             },
                             aspect_ratio: {
                                 type: "string",
                                 description: "Aspect ratio of the generated video",
-                                enum: ["1:1", "9:16", "16:9", "4:3", "3:4", "21:9", "9:21"],
+                                enum: [
+                                    "1:1",
+                                    "9:16",
+                                    "16:9",
+                                    "4:3",
+                                    "3:4",
+                                    "21:9",
+                                    "9:21",
+                                    "adaptive",
+                                ],
                                 default: "16:9",
                             },
                             resolution: {
                                 type: "string",
-                                description: "Video resolution - 480p for faster generation, 720p for balance, 1080p for higher quality",
-                                enum: ["480p", "720p", "1080p"],
+                                description: "Video resolution — 480p for faster, 720p for balance",
+                                enum: ["480p", "720p"],
                                 default: "720p",
                             },
                             duration: {
-                                type: "string",
-                                description: "Duration of video in seconds (2-12)",
-                                pattern: "^[2-9]|1[0-2]$",
-                                default: "5",
-                            },
-                            camera_fixed: {
-                                type: "boolean",
-                                description: "Whether to fix the camera position",
-                                default: false,
-                            },
-                            seed: {
                                 type: "integer",
-                                description: "Random seed to control video generation. Use -1 for random",
-                                minimum: -1,
-                                maximum: 2147483647,
-                                default: -1,
+                                description: "Duration of video in seconds (4-15)",
+                                minimum: 4,
+                                maximum: 15,
+                                default: 5,
                             },
-                            enable_safety_checker: {
+                            generate_audio: {
                                 type: "boolean",
-                                description: "Enable content safety checking",
+                                description: "Generate native audio for the video",
                                 default: true,
                             },
-                            end_image_url: {
-                                type: "string",
-                                description: "URL of image the video should end with (image-to-video only)",
-                                format: "uri",
+                            web_search: {
+                                type: "boolean",
+                                description: "Enable web search to enhance prompt understanding",
+                                default: false,
+                            },
+                            nsfw_checker: {
+                                type: "boolean",
+                                description: "Enable NSFW content filtering",
+                                default: false,
                             },
                             callBackUrl: {
                                 type: "string",
@@ -1405,6 +1432,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 +1890,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":
@@ -1931,8 +1986,8 @@ class KieAiMcpServer {
                     },
                     {
                         uri: "kie://models/bytedance-seedance",
-                        name: "ByteDance Seedance",
-                        description: "Professional video generation with lite/pro quality modes",
+                        name: "ByteDance Seedance 2.0",
+                        description: "Multimodal video generation with native audio, image/video/audio references, standard and fast modes",
                         mimeType: "text/markdown",
                         annotations: {
                             audience: ["assistant"],
@@ -2020,6 +2075,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 +2494,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 +2622,7 @@ class KieAiMcpServer {
                     "qwen-image",
                     "openai-4o-image",
                     "flux-kontext-image",
+                    "topaz-upscale",
                     "recraft-remove-background",
                     "ideogram-reframe",
                     "midjourney",
@@ -2962,10 +3055,8 @@ class KieAiMcpServer {
             request.callBackUrl = this.getCallbackUrl(request.callBackUrl);
             const response = await this.client.generateByteDanceSeedanceVideo(request);
             if (response.code === 200 && response.data?.taskId) {
-                // Determine mode for user feedback
-                const isImageToVideo = !!request.image_url;
-                const mode = isImageToVideo ? "Image-to-Video" : "Text-to-Video";
-                const quality = request.quality || "lite";
+                const mode = request.mode || "standard";
+                const hasFrameInput = !!request.first_frame_url;
                 // Store task in database
                 await this.db.createTask({
                     task_id: response.data.taskId,
@@ -2979,24 +3070,35 @@ class KieAiMcpServer {
                             text: JSON.stringify({
                                 success: true,
                                 task_id: response.data.taskId,
-                                message: `ByteDance Seedance ${mode} generation task created successfully`,
+                                message: `ByteDance Seedance 2.0 ${mode} generation task created successfully`,
                                 parameters: {
-                                    mode: mode,
-                                    quality: quality,
+                                    mode,
                                     prompt: request.prompt.substring(0, 100) +
                                         (request.prompt.length > 100 ? "..." : ""),
                                     aspect_ratio: request.aspect_ratio || "16:9",
                                     resolution: request.resolution || "720p",
-                                    duration: request.duration || "5",
-                                    ...(isImageToVideo && { image_url: request.image_url }),
-                                    ...(request.end_image_url && {
-                                        end_image_url: request.end_image_url,
+                                    duration: request.duration || 5,
+                                    generate_audio: request.generate_audio !== false,
+                                    ...(hasFrameInput && {
+                                        first_frame_url: request.first_frame_url,
+                                    }),
+                                    ...(request.last_frame_url && {
+                                        last_frame_url: request.last_frame_url,
+                                    }),
+                                    ...(request.reference_image_urls?.length && {
+                                        reference_images: request.reference_image_urls.length,
+                                    }),
+                                    ...(request.reference_video_urls?.length && {
+                                        reference_videos: request.reference_video_urls.length,
+                                    }),
+                                    ...(request.reference_audio_urls?.length && {
+                                        reference_audios: request.reference_audio_urls.length,
                                     }),
                                 },
                                 next_steps: [
                                     "Use get_task_status to check generation progress",
                                     "Task completion will be sent to the provided callback URL",
-                                    `${mode} generation typically takes 2-5 minutes depending on quality and complexity`,
+                                    `${mode} mode generation typically takes 2-5 minutes depending on duration and complexity`,
                                 ],
                             }, null, 2),
                         },
@@ -3011,26 +3113,28 @@ class KieAiMcpServer {
         catch (error) {
             if (error instanceof z.ZodError) {
                 return this.formatError("bytedance_seedance_video", error, {
-                    prompt: "Required: Text prompt for video generation (max 10000 characters)",
-                    image_url: "Optional: URL of input image for image-to-video mode",
-                    quality: "Optional: Model quality - lite (faster) or pro (higher quality, default: lite)",
+                    prompt: "Required: Text prompt for video generation (3-20000 characters)",
+                    mode: 'Optional: Generation mode — "standard" or "fast" (default: standard)',
+                    first_frame_url: "Optional: URL of image to use as first frame",
+                    last_frame_url: "Optional: URL of image to use as last frame",
+                    reference_image_urls: "Optional: Reference images for style guidance (up to 9)",
+                    reference_video_urls: "Optional: Reference videos for motion guidance (up to 3)",
+                    reference_audio_urls: "Optional: Reference audio for sound-guided generation (up to 3)",
                     aspect_ratio: "Optional: Video aspect ratio (default: 16:9)",
-                    resolution: "Optional: Video resolution - 480p/720p/1080p (default: 720p)",
-                    duration: "Optional: Video duration in seconds 2-12 (default: 5)",
-                    camera_fixed: "Optional: Fix camera position (default: false)",
-                    seed: "Optional: Random seed for reproducible results (default: -1 for random)",
-                    enable_safety_checker: "Optional: Enable content safety checking (default: true)",
-                    end_image_url: "Optional: URL of ending image (image-to-video only)",
-                    callBackUrl: "Optional: URL for task completion notifications (uses KIE_AI_CALLBACK_URL env var if not provided)",
+                    resolution: 'Optional: Video resolution — "480p" or "720p" (default: 720p)',
+                    duration: "Optional: Video duration in seconds 4-15 (default: 5)",
+                    generate_audio: "Optional: Generate native audio (default: true)",
+                    web_search: "Optional: Enable web search for prompt enhancement (default: false)",
+                    nsfw_checker: "Optional: Enable NSFW content filtering (default: false)",
+                    callBackUrl: "Optional: URL for task completion notifications",
                 });
             }
             return this.formatError("bytedance_seedance_video", error, {
-                prompt: "Required: Text prompt for video generation (max 10000 characters)",
-                image_url: "Optional: URL of input image for image-to-video mode",
-                quality: "Optional: Model quality - lite or pro",
+                prompt: "Required: Text prompt for video generation",
+                mode: 'Optional: "standard" or "fast"',
                 aspect_ratio: "Optional: Video aspect ratio",
-                resolution: "Optional: Video resolution",
-                duration: "Optional: Video duration in seconds 2-12",
+                resolution: 'Optional: "480p" or "720p"',
+                duration: "Optional: Duration in seconds 4-15",
                 callBackUrl: "Optional: URL for task completion notifications",
             });
         }
@@ -3924,6 +4028,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 +4696,12 @@ class KieAiMcpServer {
                 category: "image",
                 quality: "premium",
             },
+            {
+                name: "topaz_upscale_image",
+                status: "available",
+                category: "image",
+                quality: "professional",
+            },
             {
                 name: "recraft_remove_background",
                 status: "available",
@@ -4722,10 +4887,10 @@ The system automatically detects user intent:
 ## 🔧 Intelligent Parameter Selection
 
 ### **Video Parameters**
-- **ByteDance Seedance**: 
-  - Default: \`quality: "lite"\`, \`resolution: "720p"\`
-  - High Quality: \`quality: "pro"\`, \`resolution: "1080p"\`
-  - Professional 720p: \`quality: "pro"\`, \`resolution: "720p"\`
+- **ByteDance Seedance 2.0**:
+  - Default: \`mode: "standard"\`, \`resolution: "720p"\`, \`generate_audio: true\`
+  - Fast/Iterative: \`mode: "fast"\`, \`resolution: "480p"\`
+  - Higher Quality: \`mode: "standard"\`, \`resolution: "720p"\`
 
 - **Veo3**:
   - Default: \`model: "veo3_fast"\`
@@ -4873,7 +5038,7 @@ These guidelines ensure optimal balance between quality requirements and cost ma
 | Model | Max Resolution | Quality Modes | Duration | Speed | Key Strengths |
 |-------|---------------|---------------|----------|-------|---------------|
 | **Google Veo3** | 1080p | veo3/veo3_fast | Default | Medium | Premium cinematic quality, 1080p support |
-| **ByteDance Seedance** | 1080p | lite/pro | 2-12s | Medium | Professional standard, quality modes |
+| **ByteDance Seedance 2.0** | 720p | standard/fast | 4-15s | Medium | Multimodal refs, native audio, adaptive aspect |
 | **Wan Video 2.5** | 1080p | Single | 5-10s | Fast | Quick generation, social media |
 | **Runway Aleph** | 1080p | Single | Source | Medium | Video-to-video editing, style transfer |
 
@@ -4881,32 +5046,33 @@ These guidelines ensure optimal balance between quality requirements and cost ma
 
 ### Default Settings (Cost-Effective)
 - **Resolution**: 720p (unless user requests high quality)
-- **Quality Mode**: lite/fast (unless user requests high quality)
-- **Model**: ByteDance Seedance lite as default
+- **Quality Mode**: standard/fast (unless user requests "fast" explicitly)
+- **Model**: ByteDance Seedance 2.0 standard as default
 
 ### High Quality Upgrades
-- **User says "high quality"**: Pro models + 1080p
-- **User says "high quality in 720p"**: Pro models + 720p
+- **User says "high quality"**: Standard mode + 720p (already default)
 - **User says "cinematic"**: Veo3 model
-- **User says "fast/quick"**: Lite models + 720p (already default)
+- **User says "fast/quick"**: Seedance fast mode + 480p
 
 ## Use Case Recommendations
 
 - **Cinematic/Premium Content**: Veo3 (model: "veo3")
-- **Professional/Commercial**: ByteDance Seedance (quality: "pro")
-- **Social Media/Fast**: Wan Video 2.5 or ByteDance lite
+- **Professional/Commercial**: ByteDance Seedance 2.0 (mode: "standard")
+- **Social Media/Fast**: ByteDance Seedance 2.0 fast or Wan Video 2.5
+- **Multimodal (refs + audio)**: ByteDance Seedance 2.0 with reference URLs
 - **Video Editing**: Runway Aleph (existing video transformation)
 
 ## Parameter Mapping
 
 ### Input Methods
 - **Text-to-Video**: All models (prompt only)
-- **Image-to-Video**: Veo3 (imageUrls), ByteDance (image_url), Wan (image_url)
+- **Image-to-Video**: Veo3 (imageUrls), Seedance (first_frame_url), Wan (image_url)
 - **Video-to-Video**: Runway Aleph (videoUrl)
+- **Multimodal Refs**: Seedance 2.0 (reference_image/video/audio_urls)
 
 ### Quality Control
 - **Veo3**: model selection (veo3 vs veo3_fast)
-- **ByteDance**: quality parameter (lite vs pro) + resolution
+- **Seedance 2.0**: mode (standard vs fast) + resolution
 - **Wan**: resolution parameter only
 - **Runway**: implicit (no quality settings)
 
@@ -4925,7 +5091,7 @@ These guidelines ensure optimal balance between quality requirements and cost ma
 ### **CRITICAL COST CONTROL RULES**
 - **Resolution**: ALWAYS use \`"720p"\` unless user explicitly requests high quality
 - **Quality Level**: ALWAYS use **lite/fast** versions unless user requests "high quality"
-- **Model Selection**: bytedance_seedance_video with \`quality: "lite"\` as default
+- **Model Selection**: bytedance_seedance_video with \`mode: "standard"\` as default
 
 ### **Quality Upgrade Logic**
 
@@ -5025,11 +5191,12 @@ 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
             veo3: "google_veo3-text-to-image.md",
-            "bytedance-seedance": "bytedance_seedance-v1-lite-text-to-video.md",
+            "bytedance-seedance": "bytedance_seedance-2.md",
             "wan-video": "wan_2-5-text-to-video.md",
             "runway-aleph": "runway_aleph_video.md",
             "kling-v2-1": "kling_v2-1-pro.md",

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") {
@@ -186,37 +187,38 @@ export class KieAiClient {
         return this.makeRequest("/jobs/createTask", "POST", jobRequest);
     }
     async generateByteDanceSeedanceVideo(request) {
-        // Determine model based on quality and mode (text-to-video vs image-to-video)
-        const isImageToVideo = !!request.image_url;
-        const quality = request.quality || "lite";
-        let model;
-        if (isImageToVideo) {
-            model =
-                quality === "pro"
-                    ? "bytedance/v1-pro-image-to-video"
-                    : "bytedance/v1-lite-image-to-video";
-        }
-        else {
-            model =
-                quality === "pro"
-                    ? "bytedance/v1-pro-text-to-video"
-                    : "bytedance/v1-lite-text-to-video";
-        }
+        // Seedance 2.0: two model variants
+        const model = request.mode === "fast"
+            ? "bytedance/seedance-2-fast"
+            : "bytedance/seedance-2";
         const input = {
             prompt: request.prompt,
             aspect_ratio: request.aspect_ratio || "16:9",
             resolution: request.resolution || "720p",
-            duration: request.duration || "5",
-            camera_fixed: request.camera_fixed || false,
-            seed: request.seed !== undefined ? request.seed : -1,
-            enable_safety_checker: request.enable_safety_checker === true,
+            duration: request.duration || 5,
+            generate_audio: request.generate_audio !== false,
+            nsfw_checker: request.nsfw_checker === true,
         };
-        // Add image-specific parameters
-        if (isImageToVideo) {
-            input.image_url = request.image_url;
-            if (request.end_image_url) {
-                input.end_image_url = request.end_image_url;
-            }
+        // Frame control
+        if (request.first_frame_url) {
+            input.first_frame_url = request.first_frame_url;
+        }
+        if (request.last_frame_url) {
+            input.last_frame_url = request.last_frame_url;
+        }
+        // Multimodal references
+        if (request.reference_image_urls?.length) {
+            input.reference_image_urls = request.reference_image_urls;
+        }
+        if (request.reference_video_urls?.length) {
+            input.reference_video_urls = request.reference_video_urls;
+        }
+        if (request.reference_audio_urls?.length) {
+            input.reference_audio_urls = request.reference_audio_urls;
+        }
+        // Web search
+        if (request.web_search) {
+            input.web_search = true;
         }
         const jobRequest = {
             model,
@@ -773,6 +775,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";