@felores/kie-ai-mcp-server 1.2.0 → 1.2.2

package/README.md

@@ -196,7 +196,7 @@ Generate music with AI using Suno models.
 - `customMode` (boolean, required): Enable advanced parameter customization
 - `instrumental` (boolean, required): Generate instrumental music (no lyrics)
 - `model` (enum, required): AI model version - "V3_5", "V4", "V4_5", "V4_5PLUS", or "V5"
-- `callBackUrl` (string, required): URL to receive task completion updates
+- `callBackUrl` (string, optional): URL to receive task completion updates (uses KIE_AI_CALLBACK_URL environment variable if not provided)
 - `style` (string, optional): Music style/genre (required in custom mode, max 1000 chars for V4_5+, V5; 200 for V3_5, V4)
 - `title` (string, optional): Track title (required in custom mode, max 80 chars)
 - `negativeTags` (string, optional): Music styles to exclude (max 200 chars)
@@ -205,7 +205,9 @@ Generate music with AI using Suno models.
 - `weirdnessConstraint` (number, optional): Creative deviation control (0-1, up to 2 decimal places)
 - `audioWeight` (number, optional): Audio feature balance (0-1, up to 2 decimal places)
 
-**Example:**
+**Examples:**
+
+With explicit callback URL:
 ```json
 {
   "prompt": "A calm and relaxing piano track with soft melodies",
@@ -218,7 +220,17 @@ Generate music with AI using Suno models.
 }
 ```
 
-**Note**: In custom mode, `style` and `title` are required. If `instrumental` is false, `prompt` is used as exact lyrics.
+Using environment variable (KIE_AI_CALLBACK_URL):
+```json
+{
+  "prompt": "A relaxing electronic music track",
+  "customMode": false,
+  "instrumental": false,
+  "model": "V4_5PLUS"
+}
+```
+
+**Note**: In custom mode, `style` and `title` are required. If `instrumental` is false, `prompt` is used as exact lyrics. The `callBackUrl` is optional and will use the `KIE_AI_CALLBACK_URL` environment variable if not provided.
 
 ## API Endpoints
 

package/dist/index.js

@@ -12,7 +12,7 @@ class KieAiMcpServer {
     constructor() {
         this.server = new Server({
             name: 'kie-ai-mcp-server',
-            version: '1.2.0',
+            version: '1.2.2',
         });
         // Initialize client with config from environment
         const config = {
@@ -295,7 +295,7 @@ class KieAiMcpServer {
                                 },
                                 callBackUrl: {
                                     type: 'string',
-                                    description: 'URL to receive task completion updates (required for all requests)',
+                                    description: 'URL to receive task completion updates (optional, will use KIE_AI_CALLBACK_URL env var if not provided)',
                                     format: 'uri'
                                 },
                                 style: {
@@ -340,7 +340,7 @@ class KieAiMcpServer {
                                     multipleOf: 0.01
                                 }
                             },
-                            required: ['prompt', 'customMode', 'instrumental', 'model', 'callBackUrl']
+                            required: ['prompt', 'customMode', 'instrumental', 'model']
                         }
                     }
                 ]
@@ -536,29 +536,57 @@ class KieAiMcpServer {
                 apiResponse = await this.client.getTaskStatus(task_id, localTask?.api_type);
                 // Update local database with API response
                 if (apiResponse?.data) {
-                    const { state, resultJson, failCode, failMsg } = apiResponse.data;
-                    // Map API state to our status
+                    const apiData = apiResponse.data;
+                    // Handle different response formats for different API types
                     let status = 'pending';
-                    if (state === 'success')
-                        status = 'completed';
-                    else if (state === 'fail')
-                        status = 'failed';
-                    else if (state === 'waiting')
-                        status = 'processing';
-                    // Parse resultJson if available
-                    if (resultJson) {
-                        try {
-                            parsedResult = JSON.parse(resultJson);
+                    let resultUrl = undefined;
+                    let errorMessage = undefined;
+                    if (localTask?.api_type === 'suno') {
+                        // Suno-specific status mapping
+                        const sunoStatus = apiData.status;
+                        if (sunoStatus === 'SUCCESS')
+                            status = 'completed';
+                        else if (sunoStatus === 'CREATE_TASK_FAILED' || sunoStatus === 'GENERATE_AUDIO_FAILED' ||
+                            sunoStatus === 'CALLBACK_EXCEPTION' || sunoStatus === 'SENSITIVE_WORD_ERROR')
+                            status = 'failed';
+                        else if (sunoStatus === 'PENDING' || sunoStatus === 'TEXT_SUCCESS' || sunoStatus === 'FIRST_SUCCESS')
+                            status = 'processing';
+                        // Extract audio URLs from Suno response
+                        if (apiData.response?.sunoData && apiData.response.sunoData.length > 0) {
+                            // Use the first audio URL as the primary result
+                            resultUrl = apiData.response.sunoData[0].audioUrl;
                         }
-                        catch (e) {
-                            // Invalid JSON in resultJson
+                        // Extract error message for Suno
+                        if (apiData.errorMessage) {
+                            errorMessage = apiData.errorMessage;
                         }
                     }
+                    else {
+                        // Original logic for other APIs (Nano Banana, Veo3)
+                        const { state, resultJson, failCode, failMsg } = apiData;
+                        if (state === 'success')
+                            status = 'completed';
+                        else if (state === 'fail')
+                            status = 'failed';
+                        else if (state === 'waiting')
+                            status = 'processing';
+                        // Parse resultJson if available
+                        if (resultJson) {
+                            try {
+                                parsedResult = JSON.parse(resultJson);
+                            }
+                            catch (e) {
+                                // Invalid JSON in resultJson
+                            }
+                        }
+                        resultUrl = parsedResult?.resultUrls?.[0] || undefined;
+                        errorMessage = failMsg || undefined;
+                    }
                     // Update database
                     await this.db.updateTask(task_id, {
                         status,
-                        result_url: parsedResult?.resultUrls?.[0] || undefined,
-                        error_message: failMsg || undefined
+                        result_url: resultUrl,
+                        error_message: errorMessage
                     });
                 }
             }
@@ -567,19 +595,57 @@ class KieAiMcpServer {
             }
             // Fetch updated local task
             const updatedTask = await this.db.getTask(task_id);
+            // Prepare response based on API type
+            let responseData = {
+                success: true,
+                task_id: task_id,
+                status: updatedTask?.status,
+                result_urls: updatedTask?.result_url ? [updatedTask.result_url] : [],
+                error: updatedTask?.error_message,
+                api_response: apiResponse,
+                message: updatedTask ? 'Task found' : 'Task not found in local database'
+            };
+            // Add Suno-specific information if applicable
+            if (localTask?.api_type === 'suno' && apiResponse?.data) {
+                const sunoData = apiResponse.data;
+                responseData.status = sunoData.status; // Use Suno's status directly
+                // Add detailed Suno information
+                if (sunoData.response?.sunoData) {
+                    responseData.audio_files = sunoData.response.sunoData.map((audio) => ({
+                        id: audio.id,
+                        audio_url: audio.audioUrl,
+                        stream_url: audio.streamAudioUrl,
+                        image_url: audio.imageUrl,
+                        title: audio.title,
+                        duration: audio.duration,
+                        model_name: audio.modelName,
+                        tags: audio.tags,
+                        create_time: audio.createTime
+                    }));
+                    // Update result_urls with all audio URLs
+                    responseData.result_urls = sunoData.response.sunoData.map((audio) => audio.audioUrl);
+                }
+                // Add Suno-specific metadata
+                responseData.suno_metadata = {
+                    task_type: sunoData.type,
+                    operation_type: sunoData.operationType,
+                    parent_music_id: sunoData.parentMusicId,
+                    parameters: sunoData.param ? JSON.parse(sunoData.param) : null,
+                    error_code: sunoData.errorCode,
+                    error_message: sunoData.errorMessage
+                };
+            }
+            else {
+                // Use original logic for other APIs
+                responseData.status = apiResponse?.data?.state || updatedTask?.status;
+                responseData.result_urls = parsedResult?.resultUrls || (updatedTask?.result_url ? [updatedTask.result_url] : []);
+                responseData.error = apiResponse?.data?.failMsg || updatedTask?.error_message;
+            }
             return {
                 content: [
                     {
                         type: 'text',
-                        text: JSON.stringify({
-                            success: true,
-                            task_id: task_id,
-                            status: apiResponse?.data?.state || updatedTask?.status,
-                            result_urls: parsedResult?.resultUrls || (updatedTask?.result_url ? [updatedTask.result_url] : []),
-                            error: apiResponse?.data?.failMsg || updatedTask?.error_message,
-                            api_response: apiResponse,
-                            message: updatedTask ? 'Task found' : 'Task not found in local database'
-                        }, null, 2)
+                        text: JSON.stringify(responseData, null, 2)
                     }
                 ]
             };
@@ -653,6 +719,10 @@ class KieAiMcpServer {
     async handleSunoGenerateMusic(args) {
         try {
             const request = SunoGenerateSchema.parse(args);
+            // Use environment variable as fallback if callBackUrl not provided
+            if (!request.callBackUrl && process.env.KIE_AI_CALLBACK_URL) {
+                request.callBackUrl = process.env.KIE_AI_CALLBACK_URL;
+            }
             const response = await this.client.generateSunoMusic(request);
             if (response.code === 200 && response.data?.taskId) {
                 // Store task in database
@@ -695,7 +765,7 @@ class KieAiMcpServer {
                 customMode: 'Required: Enable advanced customization (true/false)',
                 instrumental: 'Required: Generate instrumental music (true/false)',
                 model: 'Required: AI model version (V3_5, V4, V4_5, V4_5PLUS, V5)',
-                callBackUrl: 'Required: URL for task completion notifications',
+                callBackUrl: 'Optional: URL for task completion notifications (uses KIE_AI_CALLBACK_URL env var if not provided)',
                 style: 'Optional: Music style/genre (required in custom mode)',
                 title: 'Optional: Track title (required in custom mode, max 80 chars)',
                 negativeTags: 'Optional: Styles to exclude (max 200 chars)',

package/dist/kie-ai-client.js

@@ -78,7 +78,7 @@ export class KieAiClient {
             return this.makeRequest(`/jobs/recordInfo?taskId=${taskId}`, 'GET');
         }
         else if (apiType === 'suno') {
-            return this.makeRequest(`/generate?taskId=${taskId}`, 'GET');
+            return this.makeRequest(`/generate/record-info?taskId=${taskId}`, 'GET');
         }
         // Fallback: try jobs first, then veo, then generate (for tasks not in database)
         try {
@@ -90,7 +90,7 @@ export class KieAiClient {
             }
             catch (veoError) {
                 try {
-                    return await this.makeRequest(`/generate?taskId=${taskId}`, 'GET');
+                    return await this.makeRequest(`/generate/record-info?taskId=${taskId}`, 'GET');
                 }
                 catch (sunoError) {
                     throw error;

package/dist/types.d.ts

@@ -77,7 +77,7 @@ export declare const SunoGenerateSchema: z.ZodEffects<z.ZodObject<{
     customMode: z.ZodBoolean;
     instrumental: z.ZodBoolean;
     model: z.ZodEnum<["V3_5", "V4", "V4_5", "V4_5PLUS", "V5"]>;
-    callBackUrl: z.ZodString;
+    callBackUrl: z.ZodOptional<z.ZodString>;
     style: z.ZodOptional<z.ZodString>;
     title: z.ZodOptional<z.ZodString>;
     negativeTags: z.ZodOptional<z.ZodString>;
@@ -88,9 +88,9 @@ export declare const SunoGenerateSchema: z.ZodEffects<z.ZodObject<{
 }, "strip", z.ZodTypeAny, {
     prompt: string;
     model: "V3_5" | "V4" | "V4_5" | "V4_5PLUS" | "V5";
-    callBackUrl: string;
     customMode: boolean;
     instrumental: boolean;
+    callBackUrl?: string | undefined;
     style?: string | undefined;
     title?: string | undefined;
     negativeTags?: string | undefined;
@@ -101,9 +101,9 @@ export declare const SunoGenerateSchema: z.ZodEffects<z.ZodObject<{
 }, {
     prompt: string;
     model: "V3_5" | "V4" | "V4_5" | "V4_5PLUS" | "V5";
-    callBackUrl: string;
     customMode: boolean;
     instrumental: boolean;
+    callBackUrl?: string | undefined;
     style?: string | undefined;
     title?: string | undefined;
     negativeTags?: string | undefined;
@@ -114,9 +114,9 @@ export declare const SunoGenerateSchema: z.ZodEffects<z.ZodObject<{
 }>, {
     prompt: string;
     model: "V3_5" | "V4" | "V4_5" | "V4_5PLUS" | "V5";
-    callBackUrl: string;
     customMode: boolean;
     instrumental: boolean;
+    callBackUrl?: string | undefined;
     style?: string | undefined;
     title?: string | undefined;
     negativeTags?: string | undefined;
@@ -127,9 +127,9 @@ export declare const SunoGenerateSchema: z.ZodEffects<z.ZodObject<{
 }, {
     prompt: string;
     model: "V3_5" | "V4" | "V4_5" | "V4_5PLUS" | "V5";
-    callBackUrl: string;
     customMode: boolean;
     instrumental: boolean;
+    callBackUrl?: string | undefined;
     style?: string | undefined;
     title?: string | undefined;
     negativeTags?: string | undefined;

package/dist/types.js

@@ -32,7 +32,7 @@ export const SunoGenerateSchema = z.object({
     customMode: z.boolean(),
     instrumental: z.boolean(),
     model: z.enum(['V3_5', 'V4', 'V4_5', 'V4_5PLUS', 'V5']),
-    callBackUrl: z.string().url(),
+    callBackUrl: z.string().url().optional(),
     style: z.string().max(1000).optional(),
     title: z.string().max(80).optional(),
     negativeTags: z.string().max(200).optional(),
@@ -41,6 +41,11 @@ export const SunoGenerateSchema = z.object({
     weirdnessConstraint: z.number().min(0).max(1).multipleOf(0.01).optional(),
     audioWeight: z.number().min(0).max(1).multipleOf(0.01).optional()
 }).refine((data) => {
+    // Check if callBackUrl is provided directly or via environment variable
+    const hasCallBackUrl = data.callBackUrl || process.env.KIE_AI_CALLBACK_URL;
+    if (!hasCallBackUrl) {
+        return false;
+    }
     if (data.customMode) {
         if (data.instrumental) {
             return data.style && data.title;
@@ -51,6 +56,6 @@ export const SunoGenerateSchema = z.object({
     }
     return true;
 }, {
-    message: "In customMode: style and title are always required, prompt is required when instrumental is false",
-    path: ["customMode"]
+    message: "callBackUrl is required (either directly or via KIE_AI_CALLBACK_URL environment variable). In customMode: style and title are always required, prompt is required when instrumental is false",
+    path: ["callBackUrl"]
 });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@felores/kie-ai-mcp-server",
-  "version": "1.2.0",
+  "version": "1.2.2",
   "description": "MCP server for Kie.ai APIs (Nano Banana image generation/editing and Veo3 video generation)",
   "main": "dist/index.js",
   "bin": {