cerevox 2.37.1 → 2.38.1

package/dist/mcp/servers/zerocut.js

@@ -53,7 +53,6 @@ const node_path_1 = __importStar(require("node:path"));
 const doubao_voices_full_1 = require("./helper/doubao_voices_full");
 const node_fs_1 = require("node:fs");
 const coze_1 = require("../../utils/coze");
-const node_http_1 = __importDefault(require("node:http"));
 const node_child_process_1 = require("node:child_process");
 const node_util_1 = require("node:util");
 const os = __importStar(require("node:os"));
@@ -451,7 +450,7 @@ server.registerTool('retrieve-rules-context', {
             'stage-play',
             'story-telling',
             'creative-ad',
-            'professional',
+            'expert',
             'material-creation',
             'freeform',
             'custom',
@@ -464,9 +463,9 @@ server.registerTool('retrieve-rules-context', {
 - stage-play 创建舞台播放视频
 - story-telling 创建故事讲述视频
 - creative-ad 创建创意广告视频
-- professional 创建专业视频
-- material-creation 素材创作模式
-- freeform 自由创作模式
+- expert 以专家模式创建视频，必须用户主动要求才触发
+- material-creation 素材创作模式，必须用户主动要求才触发
+- freeform 自由创作模式，必须用户主动要求才触发
 - custom 自定义模式`),
     },
 }, async ({ purpose }) => {
@@ -492,7 +491,7 @@ server.registerTool('retrieve-rules-context', {
             purpose !== 'stage-play' &&
             purpose !== 'story-telling' &&
             purpose !== 'creative-ad' &&
-            purpose !== 'professional' &&
+            purpose !== 'expert' &&
             purpose !== 'material-creation' &&
             purpose !== 'freeform') {
             return createErrorResponse(`Project rules file not found: ${projectRulesFile}`, 'retrieve-rules-context');
@@ -1732,6 +1731,7 @@ server.registerTool('generate-video', {
             'pixv',
             'kenburns',
             'zero',
+            'zero-fast',
         ])
             .default('lite')
             .describe('除非用户明确提出使用其他模型，否则一律用lite模型；zero 系列模型适合创作8-23秒带故事情节的短片'),
@@ -1773,17 +1773,21 @@ server.registerTool('generate-video', {
     try {
         // 验证session状态
         const currentSession = await validateSession('generate-video');
-        if (!start_frame && type !== 'zero') {
+        const isZeroModel = type.startsWith('zero');
+        if (!start_frame && !isZeroModel) {
             return createErrorResponse('start_frame 不能为空', 'generate-video');
         }
-        if (type !== 'zero' && duration > 16) {
+        if (!isZeroModel && duration > 16) {
             return createErrorResponse('非 zero 系列模型的视频仅支持 16 秒以下时长', 'generate-video');
         }
-        else if (type === 'zero' && duration < 8) {
-            return createErrorResponse('zero 系列模型的视频仅支持 8 秒 ~ 23 秒时长', 'generate-video');
+        else if (isZeroModel && duration < 8) {
+            return createErrorResponse('zero 系列模型的视频仅支持 8 秒以上时长', 'generate-video');
+        }
+        if (type === 'zero-fast' && duration > 18) {
+            return createErrorResponse('zero-fast 模型的视频仅支持 18 秒以下时长', 'generate-video');
         }
         if (type === 'zero' && resolution !== '1080p') {
-            console.warn(`zero 系列模型的视频仅支持 1080p 分辨率，用户指定的分辨率为 %s，已自动将 ${resolution} 转换为 1080p`, resolution);
+            console.warn(`zero 模型的视频仅支持 1080p 分辨率，用户指定的分辨率为 %s，已自动将 ${resolution} 转换为 1080p`, resolution);
             resolution = '1080p';
         }
         // 校验 prompt 与 story_board.json 中场景设定的一致性以及视频时长与 timeline_analysis.json 中 proposed_video_scenes 的匹配
@@ -2178,6 +2182,21 @@ server.registerTool('generate-video', {
                 ],
             };
         }
+        else if (res.execute_id) {
+            return {
+                content: [
+                    {
+                        type: 'text',
+                        text: JSON.stringify({
+                            success: true,
+                            message: '该视频生成任务已提交，它是异步任务，且执行时间较长，你应立即调用工具 wait-for-workflow-finish 来等待任务结束，如该工具调用超时，你应立即再次重新调用直到任务结束。',
+                            execute_id: res.execute_id,
+                            workflow_id: res.workflow_id,
+                        }),
+                    },
+                ],
+            };
+        }
         else {
             console.warn('Video generation completed but no URL returned');
             return {
@@ -2200,6 +2219,73 @@ server.registerTool('generate-video', {
         return createErrorResponse(error, 'generate-video');
     }
 });
+server.registerTool('wait-for-workflow-finish', {
+    title: 'Wait Workflow Done',
+    description: 'Wait for a workflow to complete.',
+    inputSchema: {
+        workflowID: zod_1.z
+            .string()
+            .describe('The workflow_id of the workflow to wait for.'),
+        executeID: zod_1.z
+            .string()
+            .describe('The execute_id of the workflow to wait for.'),
+        saveToFileName: zod_1.z
+            .string()
+            .describe('The file name to save the video to.'),
+    },
+}, async ({ workflowID, executeID, saveToFileName }, context) => {
+    try {
+        const currentSession = await validateSession('wait-for-workflow-finish');
+        const ai = currentSession.ai;
+        let progress = 0;
+        const res = await ai.waitForWorkflowFinish({
+            workflowId: workflowID,
+            executeId: executeID,
+            onProgress: async (metaData) => {
+                try {
+                    await sendProgress(context, ++progress, undefined, JSON.stringify(metaData));
+                }
+                catch (progressError) {
+                    console.warn('Failed to send progress update:', progressError);
+                }
+            },
+        });
+        if (res.url) {
+            const uri = await saveMaterial(currentSession, res.url, saveToFileName);
+            // Update media_logs.json
+            try {
+                await updateMediaLogs(currentSession, saveToFileName, res, 'audio');
+            }
+            catch (error) {
+                console.warn(`Failed to update media_logs.json for ${saveToFileName}:`, error);
+            }
+            return {
+                content: [
+                    {
+                        type: 'text',
+                        text: JSON.stringify({
+                            success: true,
+                            uri,
+                            source: res.url,
+                            duration: res.duration,
+                        }),
+                    },
+                ],
+            };
+        }
+        return {
+            content: [
+                {
+                    type: 'text',
+                    text: JSON.stringify(res),
+                },
+            ],
+        };
+    }
+    catch (error) {
+        return createErrorResponse(error, 'wait-for-workflow-finish');
+    }
+});
 server.registerTool('generate-sound-effect', {
     title: 'Generate Sound Effect',
     description: 'Generate sound effect with text prompt.',
@@ -4208,163 +4294,182 @@ server.registerTool('run-ffmpeg-command', {
         return createErrorResponse(error, 'run-ffmpeg-command');
     }
 });
-server.registerTool('custom-edit-draft', {
-    title: 'Custom Edit Draft',
-    description: 'Launch timeline editor server for custom draft editing. Starts a local server at specified port with draft file configuration.',
-    inputSchema: {
-        draftContentFileName: zod_1.z
-            .string()
-            .optional()
-            .default('draft_content.json')
-            .describe('Draft content filename (default: draft_content.json)'),
-        port: zod_1.z
-            .number()
-            .optional()
-            .default(6789)
-            .describe('Server port (default: 6789)'),
-    },
-}, async ({ draftContentFileName = 'draft_content.json', port = 6789 }) => {
-    try {
-        // 使用项目本地目录
-        const workDir = (0, node_path_1.resolve)(process.env.ZEROCUT_PROJECT_CWD || process.cwd(), projectLocalDir);
-        // 检查 draft_content.json 文件是否存在
-        const draftFilePath = (0, node_path_1.resolve)(workDir, draftContentFileName);
-        if (!(0, node_fs_1.existsSync)(draftFilePath)) {
-            return createErrorResponse(new Error(`Draft content file not found: ${draftContentFileName}. Please generate draft content first using the generate-draft-content tool.`), 'custom-edit-draft');
-        }
-        // 备份原始的 draft_content.json 文件
-        const timestamp = new Date().toISOString().replace(/[:.]/g, '-');
-        const backupFileName = `${draftContentFileName}.backup.${timestamp}`;
-        const backupFilePath = (0, node_path_1.resolve)(workDir, backupFileName);
-        try {
-            const { copyFileSync } = await Promise.resolve().then(() => __importStar(require('fs')));
-            copyFileSync(draftFilePath, backupFilePath);
-            console.log(`Draft file backed up to: ${backupFileName}`);
-        }
-        catch (backupError) {
-            console.warn(`Failed to backup draft file: ${backupError}`);
-            // 继续执行，不因备份失败而中断
-        }
-        // 构建启动命令参数
-        const timelineEditorPath = (0, node_path_1.resolve)(__dirname, '../../../dist/timeline-editor/index.js');
-        console.log(`Starting timeline editor server...`);
-        console.log(`Working directory: ${workDir}`);
-        console.log(`Draft file: ${draftContentFileName}`);
-        console.log(`Server port: ${port}`);
-        console.log(`Timeline editor path: ${timelineEditorPath}`);
-        // 检查端口是否被占用，如果被占用则停止占用的进程
-        if (await isPortInUse(port)) {
-            console.log(`Port ${port} is in use, killing existing processes...`);
-            await killPortProcess(port);
-        }
-        // 直接启动服务器进程
-        const platform = os.platform();
-        const nodeCommand = platform === 'win32' ? 'node.exe' : 'node';
-        const serverProcess = (0, node_child_process_1.spawn)(nodeCommand, [
-            timelineEditorPath,
-            '--draft-file',
-            draftContentFileName,
-            '--project-dir',
-            workDir,
-            '--port',
-            port.toString(),
-        ], {
-            cwd: workDir,
-            stdio: 'pipe',
-            shell: platform === 'win32', // 在 Windows 下使用 shell
-        });
-        const serverUrl = `http://localhost:${port}`;
-        // 等待服务器启动 - 使用健康检查 API 轮询
-        await new Promise((resolve, reject) => {
-            const timeout = setTimeout(() => {
-                reject(new Error('Server startup timeout'));
-            }, 30000); // 增加超时时间到30秒
-            const checkHealth = async () => {
-                try {
-                    const healthUrl = `${serverUrl}/health`;
-                    const req = node_http_1.default.get(healthUrl, (res) => {
-                        let data = '';
-                        res.on('data', (chunk) => {
-                            data += chunk;
-                        });
-                        res.on('end', () => {
-                            try {
-                                const healthData = JSON.parse(data);
-                                if (healthData.status === 'ok') {
-                                    clearTimeout(timeout);
-                                    console.log('Server health check passed');
-                                    resolve(void 0);
-                                }
-                                else {
-                                    // 继续轮询
-                                    setTimeout(checkHealth, 1000);
-                                }
-                            }
-                            catch (parseError) {
-                                // 继续轮询
-                                setTimeout(checkHealth, 1000);
-                            }
-                        });
-                    });
-                    req.on('error', () => {
-                        // 服务器还未启动，继续轮询
-                        setTimeout(checkHealth, 1000);
-                    });
-                    req.setTimeout(2000, () => {
-                        req.destroy();
-                        // 继续轮询
-                        setTimeout(checkHealth, 1000);
-                    });
-                }
-                catch (error) {
-                    // 继续轮询
-                    setTimeout(checkHealth, 1000);
-                }
-            };
-            serverProcess.stdout?.on('data', (data) => {
-                console.log('Server output:', data.toString());
-            });
-            serverProcess.stderr?.on('data', (data) => {
-                console.error('Server error:', data.toString());
-            });
-            serverProcess.on('error', (error) => {
-                clearTimeout(timeout);
-                reject(error);
-            });
-            // 开始健康检查轮询
-            setTimeout(checkHealth, 2000); // 2秒后开始第一次检查
-        });
-        return {
-            content: [
-                {
-                    type: 'text',
-                    text: JSON.stringify({
-                        success: true,
-                        message: 'Timeline editor server started successfully',
-                        serverUrl: serverUrl,
-                        draftFile: draftContentFileName,
-                        backupFile: backupFileName,
-                        projectDirectory: workDir,
-                        port: port,
-                        pid: serverProcess.pid,
-                        instructions: [
-                            `Timeline editor is now running at: ${serverUrl}`,
-                            'Open the URL in your browser to edit the draft',
-                            'The server will automatically save changes to the draft file',
-                            `Original draft file has been backed up as: ${backupFileName}`,
-                            `Server process ID: ${serverProcess.pid}`,
-                            'The server will continue running in the background',
-                        ],
-                    }, null, 2),
-                },
-            ],
-        };
-    }
-    catch (error) {
-        console.error('Error starting timeline editor:', error);
-        return createErrorResponse(error, 'custom-edit-draft');
-    }
-});
+// server.registerTool(
+//   'custom-edit-draft',
+//   {
+//     title: 'Custom Edit Draft',
+//     description:
+//       'Launch timeline editor server for custom draft editing. Starts a local server at specified port with draft file configuration.',
+//     inputSchema: {
+//       draftContentFileName: z
+//         .string()
+//         .optional()
+//         .default('draft_content.json')
+//         .describe('Draft content filename (default: draft_content.json)'),
+//       port: z
+//         .number()
+//         .optional()
+//         .default(6789)
+//         .describe('Server port (default: 6789)'),
+//     },
+//   },
+//   async ({ draftContentFileName = 'draft_content.json', port = 6789 }) => {
+//     try {
+//       // 使用项目本地目录
+//       const workDir = resolve(
+//         process.env.ZEROCUT_PROJECT_CWD || process.cwd(),
+//         projectLocalDir
+//       );
+//       // 检查 draft_content.json 文件是否存在
+//       const draftFilePath = resolve(workDir, draftContentFileName);
+//       if (!existsSync(draftFilePath)) {
+//         return createErrorResponse(
+//           new Error(
+//             `Draft content file not found: ${draftContentFileName}. Please generate draft content first using the generate-draft-content tool.`
+//           ),
+//           'custom-edit-draft'
+//         );
+//       }
+//       // 备份原始的 draft_content.json 文件
+//       const timestamp = new Date().toISOString().replace(/[:.]/g, '-');
+//       const backupFileName = `${draftContentFileName}.backup.${timestamp}`;
+//       const backupFilePath = resolve(workDir, backupFileName);
+//       try {
+//         const { copyFileSync } = await import('fs');
+//         copyFileSync(draftFilePath, backupFilePath);
+//         console.log(`Draft file backed up to: ${backupFileName}`);
+//       } catch (backupError) {
+//         console.warn(`Failed to backup draft file: ${backupError}`);
+//         // 继续执行，不因备份失败而中断
+//       }
+//       // 构建启动命令参数
+//       const timelineEditorPath = resolve(
+//         __dirname,
+//         '../../../dist/timeline-editor/index.js'
+//       );
+//       console.log(`Starting timeline editor server...`);
+//       console.log(`Working directory: ${workDir}`);
+//       console.log(`Draft file: ${draftContentFileName}`);
+//       console.log(`Server port: ${port}`);
+//       console.log(`Timeline editor path: ${timelineEditorPath}`);
+//       // 检查端口是否被占用，如果被占用则停止占用的进程
+//       if (await isPortInUse(port)) {
+//         console.log(`Port ${port} is in use, killing existing processes...`);
+//         await killPortProcess(port);
+//       }
+//       // 直接启动服务器进程
+//       const platform = os.platform();
+//       const nodeCommand = platform === 'win32' ? 'node.exe' : 'node';
+//       const serverProcess = spawn(
+//         nodeCommand,
+//         [
+//           timelineEditorPath,
+//           '--draft-file',
+//           draftContentFileName,
+//           '--project-dir',
+//           workDir,
+//           '--port',
+//           port.toString(),
+//         ],
+//         {
+//           cwd: workDir,
+//           stdio: 'pipe',
+//           shell: platform === 'win32', // 在 Windows 下使用 shell
+//         }
+//       );
+//       const serverUrl = `http://localhost:${port}`;
+//       // 等待服务器启动 - 使用健康检查 API 轮询
+//       await new Promise((resolve, reject) => {
+//         const timeout = setTimeout(() => {
+//           reject(new Error('Server startup timeout'));
+//         }, 30000); // 增加超时时间到30秒
+//         const checkHealth = async () => {
+//           try {
+//             const healthUrl = `${serverUrl}/health`;
+//             const req = http.get(healthUrl, (res: any) => {
+//               let data = '';
+//               res.on('data', (chunk: any) => {
+//                 data += chunk;
+//               });
+//               res.on('end', () => {
+//                 try {
+//                   const healthData = JSON.parse(data);
+//                   if (healthData.status === 'ok') {
+//                     clearTimeout(timeout);
+//                     console.log('Server health check passed');
+//                     resolve(void 0);
+//                   } else {
+//                     // 继续轮询
+//                     setTimeout(checkHealth, 1000);
+//                   }
+//                 } catch (parseError) {
+//                   // 继续轮询
+//                   setTimeout(checkHealth, 1000);
+//                 }
+//               });
+//             });
+//             req.on('error', () => {
+//               // 服务器还未启动，继续轮询
+//               setTimeout(checkHealth, 1000);
+//             });
+//             req.setTimeout(2000, () => {
+//               req.destroy();
+//               // 继续轮询
+//               setTimeout(checkHealth, 1000);
+//             });
+//           } catch (error) {
+//             // 继续轮询
+//             setTimeout(checkHealth, 1000);
+//           }
+//         };
+//         serverProcess.stdout?.on('data', (data: any) => {
+//           console.log('Server output:', data.toString());
+//         });
+//         serverProcess.stderr?.on('data', (data: any) => {
+//           console.error('Server error:', data.toString());
+//         });
+//         serverProcess.on('error', (error: any) => {
+//           clearTimeout(timeout);
+//           reject(error);
+//         });
+//         // 开始健康检查轮询
+//         setTimeout(checkHealth, 2000); // 2秒后开始第一次检查
+//       });
+//       return {
+//         content: [
+//           {
+//             type: 'text',
+//             text: JSON.stringify(
+//               {
+//                 success: true,
+//                 message: 'Timeline editor server started successfully',
+//                 serverUrl: serverUrl,
+//                 draftFile: draftContentFileName,
+//                 backupFile: backupFileName,
+//                 projectDirectory: workDir,
+//                 port: port,
+//                 pid: serverProcess.pid,
+//                 instructions: [
+//                   `Timeline editor is now running at: ${serverUrl}`,
+//                   'Open the URL in your browser to edit the draft',
+//                   'The server will automatically save changes to the draft file',
+//                   `Original draft file has been backed up as: ${backupFileName}`,
+//                   `Server process ID: ${serverProcess.pid}`,
+//                   'The server will continue running in the background',
+//                 ],
+//               },
+//               null,
+//               2
+//             ),
+//           },
+//         ],
+//       };
+//     } catch (error) {
+//       console.error('Error starting timeline editor:', error);
+//       return createErrorResponse(error, 'custom-edit-draft');
+//     }
+//   }
+// );
 async function run() {
     // Start receiving messages on stdin and sending messages on stdout
     const transport = new stdio_js_1.StdioServerTransport();