mcp-ai-music 1.0.6 → 1.0.7

package/dist/index.js

@@ -26,8 +26,8 @@ function getServiceBase() {
         process.env.SUNO_QUERY_BASE_URL ||
         process.env.SUNO_CALLBACK_URL);
 }
-// 构造回调地址
-function buildCallbackUrl(kind) {
+// 构造回调地址，可附带 chatSessionId
+function buildCallbackUrl(kind, chatSessionId) {
     const base = getServiceBase();
     if (!base) {
         throw new Error("未配置 SUNO_SERVICE_BASE（或兼容变量），用于设置服务端基地址");
@@ -37,7 +37,11 @@ function buildCallbackUrl(kind) {
         "upload-cover": "/suno/callback/upload-cover",
         "upload-extend": "/suno/callback/upload-extend",
     };
-    return `${base.replace(/\/$/, "")}${pathMap[kind]}`;
+    const url = `${base.replace(/\/$/, "")}${pathMap[kind]}`;
+    if (chatSessionId) {
+        return `${url}?chat_session_id=${encodeURIComponent(chatSessionId)}`;
+    }
+    return url;
 }
 // 查询同样使用统一基地址
 function getQueryBase() {
@@ -351,7 +355,7 @@ async function makeApiRequestRaw(url, method, body, retries = 3) {
     }
 }
 // 生成音乐处理函数
-async function handleGenerateMusic(input) {
+async function handleGenerateMusic(input, chatSessionId) {
     try {
         if (!input || typeof input !== 'object') {
             return {
@@ -456,7 +460,7 @@ async function handleGenerateMusic(input) {
                 errorMessage: "title长度超过限制，最大允许80字符。",
             };
         }
-        const callbackUrl = buildCallbackUrl("generate");
+        const callbackUrl = buildCallbackUrl("generate", chatSessionId);
         console.error(`使用回调地址: ${callbackUrl}`);
         // 构建请求体，根据API文档，当customMode=true且instrumental=false时，
         // 如果提供了lyrics，则使用lyrics作为prompt字段发送给API
@@ -505,7 +509,7 @@ async function handleGenerateMusic(input) {
     }
 }
 // 翻唱音乐处理函数
-async function handleCoverMusic(input) {
+async function handleCoverMusic(input, chatSessionId) {
     try {
         if (!input || typeof input !== 'object') {
             return {
@@ -559,7 +563,7 @@ async function handleCoverMusic(input) {
             instrumental,
             model,
             negativeTags,
-            callBackUrl: buildCallbackUrl("upload-cover"), // 必需的参数
+            callBackUrl: buildCallbackUrl("upload-cover", chatSessionId), // 必需的参数
         };
         // 根据Suno API文档，当instrumental=false且提供了lyrics时，使用lyrics作为prompt字段发送给API
         if (!instrumental && lyrics) {
@@ -596,7 +600,7 @@ async function handleCoverMusic(input) {
     }
 }
 // 扩展音乐处理函数
-async function handleExtendMusic(input) {
+async function handleExtendMusic(input, chatSessionId) {
     try {
         if (!input || typeof input !== 'object') {
             return {
@@ -658,7 +662,7 @@ async function handleExtendMusic(input) {
             continueAt,
             model,
             negativeTags,
-            callBackUrl: buildCallbackUrl("upload-extend"), // 必需的参数
+            callBackUrl: buildCallbackUrl("upload-extend", chatSessionId), // 必需的参数
         };
         // 根据Suno API文档，当instrumental=false且提供了lyrics时，使用lyrics作为prompt字段发送给API
         if (!instrumental && lyrics) {
@@ -755,6 +759,12 @@ async function handleQueryProgress(input) {
             status = 'complete';
         if (status === 'running')
             status = 'processing';
+        // 仅当存在最终音频链接（audioUrl 非空）才认为完成；
+        // 如果只有流式地址（streamAudioUrl）但 audioUrl 为空，则视为进行中
+        const hasFinalAudio = typeof result.audioUrl === 'string' && result.audioUrl.trim() !== '';
+        if (status === 'complete' && !hasFinalAudio) {
+            status = 'processing';
+        }
         let statusMessage = "";
         switch (status) {
             case 'pending':
@@ -783,7 +793,7 @@ async function handleQueryProgress(input) {
             status,
             progress,
             message: statusMessage,
-            result: status === 'complete' ? {
+            result: status === 'complete' && hasFinalAudio ? {
                 // 基本信息
                 taskId: result.taskId,
                 status: result.status,
@@ -953,14 +963,24 @@ server.setRequestHandler(types_js_1.CallToolRequestSchema, async (request) => {
     try {
         const toolName = request.params.name;
         const toolInput = request.params.arguments;
+        // 解析 meta.chatSessionId （如果存在）
+        const chatSessionId = request?.meta?.chatSessionId
+            ?? request?.params?.meta?.chatSessionId
+            ?? undefined;
+        if (chatSessionId) {
+            console.error(`接收到 chatSessionId: ${chatSessionId}`);
+        }
+        else {
+            console.error("未在请求中检测到 chatSessionId（meta.chatSessionId）");
+        }
         console.error(`收到工具调用请求: ${toolName}, 输入: ${JSON.stringify(toolInput)}`);
         switch (toolName) {
             case GENERATE_MUSIC_TOOL.name:
-                return await handleGenerateMusic(toolInput);
+                return await handleGenerateMusic(toolInput, chatSessionId);
             case COVER_MUSIC_TOOL.name:
-                return await handleCoverMusic(toolInput);
+                return await handleCoverMusic(toolInput, chatSessionId);
             case EXTEND_MUSIC_TOOL.name:
-                return await handleExtendMusic(toolInput);
+                return await handleExtendMusic(toolInput, chatSessionId);
             case QUERY_PROGRESS_TOOL.name:
                 return await handleQueryProgress(toolInput);
             default:

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "mcp-ai-music",
-  "version": "1.0.6",
+  "version": "1.0.7",
   "description": "AI作曲MCP服务 - 基于Suno4.5 API的音乐生成、翻唱、扩展和进度查询工具",
   "main": "dist/index.js",
   "scripts": {