cerevox 2.29.0 → 2.30.1

package/dist/mcp/servers/zerocut.js

@@ -490,6 +490,47 @@ server.registerTool('retrieve-rules-context', {
             await (0, promises_1.mkdir)((0, node_path_1.dirname)(projectRulesFile), { recursive: true });
             await (0, promises_1.writeFile)(projectRulesFile, promptContent);
         }
+        // 如果 purpose 为 freeform，复制 skills 和 knowledge 目录
+        if (purpose === 'freeform') {
+            // 递归复制目录的通用函数
+            const copyDirectory = async (src, dest) => {
+                const entries = await (0, promises_1.readdir)(src, { withFileTypes: true });
+                for (const entry of entries) {
+                    const srcPath = (0, node_path_1.resolve)(src, entry.name);
+                    const destPath = (0, node_path_1.resolve)(dest, entry.name);
+                    if (entry.isDirectory()) {
+                        await (0, promises_1.mkdir)(destPath, { recursive: true });
+                        await copyDirectory(srcPath, destPath);
+                    }
+                    else {
+                        const content = await (0, promises_1.readFile)(srcPath, 'utf-8');
+                        await (0, promises_1.writeFile)(destPath, content);
+                    }
+                }
+            };
+            // 复制 skills 目录
+            const sourceSkillsDir = (0, node_path_1.resolve)(__dirname, './prompts/skills');
+            const targetSkillsDir = (0, node_path_1.resolve)(projectLocalDir, '.trae', 'skills');
+            try {
+                await (0, promises_1.mkdir)(targetSkillsDir, { recursive: true });
+                await copyDirectory(sourceSkillsDir, targetSkillsDir);
+                console.log(`Skills directory copied to ${targetSkillsDir}`);
+            }
+            catch (skillsError) {
+                console.warn(`Failed to copy skills directory: ${skillsError}`);
+            }
+            // 复制 knowledge 目录
+            const sourceKnowledgeDir = (0, node_path_1.resolve)(__dirname, './prompts/knowledge');
+            const targetKnowledgeDir = (0, node_path_1.resolve)(projectLocalDir, '.trae', 'knowledge');
+            try {
+                await (0, promises_1.mkdir)(targetKnowledgeDir, { recursive: true });
+                await copyDirectory(sourceKnowledgeDir, targetKnowledgeDir);
+                console.log(`Knowledge directory copied to ${targetKnowledgeDir}`);
+            }
+            catch (knowledgeError) {
+                console.warn(`Failed to copy knowledge directory: ${knowledgeError}`);
+            }
+        }
         return {
             content: [
                 {
@@ -2303,135 +2344,45 @@ server.registerTool('generate-sound-effect', {
         return createErrorResponse(error, 'generate-sound-effect');
     }
 });
-server.registerTool('generate-song', {
-    title: 'Generate Song',
-    description: 'Generate a song with vocals and customizable parameters.',
+server.registerTool('generate-music', {
+    title: 'Generate Music',
+    description: 'Generate the music. Include background music or song.',
     inputSchema: {
+        prompt: zod_1.z.string().describe('The prompt to generate.'),
         type: zod_1.z
+            .enum(['bgm', 'song'])
+            .describe('The type of music. Defaults to background music.')
+            .default('bgm'),
+        model: zod_1.z
             .enum(['doubao', 'minimax'])
-            .default('doubao')
-            .describe('The model type of AI to use.'),
-        lyrics: zod_1.z.string().describe(`The lyrics to generate the song.
-
-- 完整歌词通常包括以下桥段：
-  - 前奏: intro，歌曲开始的音乐部分，主要用于引导歌曲的整体氛围。
-  - 主歌: verse，通常在前奏之后，歌曲中叙述歌曲故事或主题的部分。
-  - 副歌: chorus，一般在主歌之后，旋律有记忆点和感染力，是整首歌的高潮，进一步强化歌曲的主题和情感。
-  - 间奏: inst，歌曲中的纯音乐段落，用于连接不同的演唱部分。
-  - 尾奏: outro，歌曲结束后的音乐段落，用于营造歌曲结束的氛围。
-  - 桥段: bridge，通常出现在歌曲中段或接近结尾处，是一个过渡部分，用于连接不同的歌曲段落。 
-
-### 歌词示例 lyrics.txt
-
-\`\`\`txt
-[intro]
-[verse]
-记得那一天 那一天我们相恋
-说好彼此都不说再见
-遵守诺言 用心去相恋
-我为你撑伞 你为我取暖
-[inst]
-[chorus]
-当我把心交给你的那一天
-你却消失在我的眼前
-事到如今已经过了好多年
-是否你还像从前
-[outro]
-\`\`\`
-`),
+            .optional()
+            .describe('The model to use. Defaults to doubao.')
+            .default('doubao'),
         duration: zod_1.z
             .number()
             .min(30)
             .max(240)
-            .describe('The duration of the song in seconds (30-240).'),
-        genre: zod_1.z
-            .enum([
-            'Folk',
-            'Pop',
-            'Rock',
-            'Chinese Style',
-            'Hip Hop/Rap',
-            'R&B/Soul',
-            'Punk',
-            'Electronic',
-            'Jazz',
-            'Reggae',
-            'DJ',
-            'Pop Punk',
-            'Disco',
-            'Future Bass',
-            'Pop Rap',
-            'Trap Rap',
-            'R&B Rap',
-            'Chinoiserie Electronic',
-            'GuFeng Music',
-            'Pop Rock',
-            'Jazz Pop',
-            'Bossa Nova',
-            'Contemporary R&B',
-        ])
-            .optional()
-            .describe('The genre of the song.'),
-        mood: zod_1.z
-            .enum([
-            'Happy',
-            'Dynamic/Energetic',
-            'Sentimental/Melancholic/Lonely',
-            'Inspirational/Hopeful',
-            'Nostalgic/Memory',
-            'Excited',
-            'Sorrow/Sad',
-            'Chill',
-            'Relaxing',
-            'Romantic',
-            'Miss',
-            'Groovy/Funky',
-            'Dreamy/Ethereal',
-            'Calm/Relaxing',
-        ])
-            .optional()
-            .describe('The mood of the song.'),
-        gender: zod_1.z
-            .enum(['Female', 'Male'])
-            .optional()
-            .describe('The gender of the vocalist.'),
-        timbre: zod_1.z
-            .enum([
-            'Warm',
-            'Bright',
-            'Husky',
-            'Electrified voice',
-            'Sweet_AUDIO_TIMBRE',
-            'Cute_AUDIO_TIMBRE',
-            'Loud and sonorous',
-            'Powerful',
-            'Sexy/Lazy',
-        ])
-            .optional()
-            .describe('The timbre/voice quality of the vocalist.'),
+            .describe('The duration of the bgm or music.'),
         skipCopyCheck: zod_1.z
             .boolean()
-            .optional()
             .default(false)
             .describe('Whether to skip copyright check.'),
         saveToFileName: zod_1.z.string().describe('The filename to save.'),
     },
-}, async ({ type = 'doubao', lyrics, duration, genre, mood, gender, timbre, skipCopyCheck, saveToFileName, }, context) => {
+}, async ({ prompt, type, model, duration, skipCopyCheck, saveToFileName }, context) => {
     try {
         // 验证session状态
-        const currentSession = await validateSession('generate-song');
+        const currentSession = await validateSession('generate-music');
         const validatedFileName = validateFileName(saveToFileName);
-        console.log(`Generating Song with lyrics: ${lyrics.substring(0, 100)}... (${duration}s, genre: ${genre || 'auto'}, mood: ${mood || 'auto'})`);
+        console.log(`Generating Music with prompt: ${prompt.substring(0, 100)}... (${duration}s)`);
         const ai = currentSession.ai;
         let progress = 0;
-        const res = await ai.generateSong({
-            type,
-            lyrics: lyrics.trim(),
-            duration,
-            genre,
-            mood,
-            gender,
-            timbre,
+        if (type === 'bgm' && duration > 120) {
+            throw new Error('BGM duration must be at most 120 seconds.');
+        }
+        const finalPrompt = `${prompt.trim()} ${type === 'bgm' ? `纯音乐无歌词，时长${duration}秒` : `时长${duration}秒，使用${model}模型`}`;
+        const res = await ai.generateMusic({
+            prompt: finalPrompt,
             skipCopyCheck,
             onProgress: async (metaData) => {
                 try {
@@ -2443,18 +2394,12 @@ server.registerTool('generate-song', {
             },
         });
         if (!res) {
-            throw new Error('Failed to generate Song: no response from AI service');
+            throw new Error('Failed to generate BGM: no response from AI service');
         }
         if (res.url) {
-            console.log('Song generated successfully, saving to materials...');
+            console.log('BGM generated successfully, saving to materials...');
             const uri = await saveMaterial(currentSession, res.url, validatedFileName);
-            if (!res.captions) {
-                // 歌词获取失败，获取字幕
-                res.captions = await ai.voiceToCaptions({
-                    url: res.url,
-                });
-            }
-            const { url, duration: songDuration, captions, ...opts } = res;
+            const { url, duration: bgmDuration, captions, ...opts } = res;
             // 保存captions到本地
             if (captions) {
                 const captionsText = JSON.stringify(captions, null, 2);
@@ -2463,103 +2408,13 @@ server.registerTool('generate-song', {
                 // 保存到本地
                 await (0, promises_1.writeFile)(localPath, captionsText);
             }
-            const result = {
-                success: true,
-                // source: url,
-                uri,
-                durationMs: Math.floor((songDuration || duration) * 1000),
-                lyrics,
-                requestedDuration: duration,
-                genre,
-                mood,
-                gender,
-                timbre,
-                captions,
-                timestamp: new Date().toISOString(),
-                ...opts,
-            };
-            // Update media_logs.json
-            try {
-                await updateMediaLogs(currentSession, validatedFileName, result, 'audio');
-            }
-            catch (error) {
-                console.warn(`Failed to update media_logs.json for ${validatedFileName}:`, error);
-            }
-            return {
-                content: [
-                    {
-                        type: 'text',
-                        text: JSON.stringify(result),
-                    },
-                ],
-            };
-        }
-        else {
-            console.warn('Song generation completed but no URL returned');
-            return {
-                content: [
-                    {
-                        type: 'text',
-                        text: JSON.stringify({
-                            success: false,
-                            error: 'No Song URL returned from AI service',
-                            response: res,
-                            timestamp: new Date().toISOString(),
-                        }),
-                    },
-                ],
-            };
-        }
-    }
-    catch (error) {
-        return createErrorResponse(error, 'generate-song');
-    }
-});
-server.registerTool('generate-bgm', {
-    title: 'Generate BGM',
-    description: 'Generate the bgm.',
-    inputSchema: {
-        prompt: zod_1.z.string().describe('The prompt to generate.'),
-        duration: zod_1.z
-            .number()
-            .min(30)
-            .max(120)
-            .describe('The duration of the bgm.'),
-        saveToFileName: zod_1.z.string().describe('The filename to save.'),
-    },
-}, async ({ prompt, duration, saveToFileName }, context) => {
-    try {
-        // 验证session状态
-        const currentSession = await validateSession('generate-bgm');
-        const validatedFileName = validateFileName(saveToFileName);
-        console.log(`Generating BGM with prompt: ${prompt.substring(0, 100)}... (${duration}s)`);
-        const ai = currentSession.ai;
-        let progress = 0;
-        const res = await ai.generateBGM({
-            prompt: prompt.trim(),
-            duration,
-            onProgress: async (metaData) => {
-                try {
-                    await sendProgress(context, metaData.Result?.Progress ?? ++progress, metaData.Result?.Progress ? 100 : undefined, JSON.stringify(metaData));
-                }
-                catch (progressError) {
-                    console.warn('Failed to send progress update:', progressError);
-                }
-            },
-        });
-        if (!res) {
-            throw new Error('Failed to generate BGM: no response from AI service');
-        }
-        if (res.url) {
-            console.log('BGM generated successfully, saving to materials...');
-            const uri = await saveMaterial(currentSession, res.url, validatedFileName);
-            const { url, duration: bgmDuration, ...opts } = res;
             const result = {
                 success: true,
                 // source: url,
                 uri,
                 durationMs: Math.floor((bgmDuration || duration) * 1000),
                 prompt,
+                captions,
                 requestedDuration: duration,
                 timestamp: new Date().toISOString(),
                 ...opts,