wanx-mcp4chatbot/dist/wanx-t2v.js

通义万相 MCP 服务器 聊天助手优化

import axios from 'axios';
import config from './config.js';
/**
 * 调用通义万相2.1文生视频API生成视频
 * @param {string} prompt - 文本描述
 * @returns {Promise<string>} - 成功返回视频URL，失败返回错误信息
 */
export async function generateVideo(prompt) {
    const apiKey = config.api.apiKey;
    const url = 'https://dashscope.aliyuncs.com/api/v1/services/aigc/video-generation/video-synthesis';
    const headers = {
        'X-DashScope-Async': 'enable',
        'Authorization': `Bearer ${apiKey}`,
        'Content-Type': 'application/json'
    };
    const payload = {
        model: 'wanx2.1-t2v-turbo',
        input: { prompt },
        parameters: {
            size: '832*480',
            duration: 5,
            prompt_extend: true,
        },
    };
    try {
        const res = await axios.post(url, payload, { headers });
        const taskId = res.data?.output?.task_id;
        if (taskId)
            return taskId;
        // 如果响应里没有task_id，说明有错误  
        throw res.data;
    }
    catch (err) {
        // err.response?.data 可能含详细错误  
        throw err.response?.data || err.message;
    }
}
/**
 * 步骤二：查询视频生成任务状态（可用作轮询）
 * @param {string} taskId - 第一步拿到的 task_id
 * @param {string} apiKey - 通义 DASHSCOPE_API_KEY
 * @returns {Promise<object>} - 包含任务状态信息的对象
 */
export async function queryVideoTaskStatus(taskId) {
    const apiKey = config.api.apiKey;
    const url = `https://dashscope.aliyuncs.com/api/v1/tasks/${taskId}`;
    const headers = {
        'Authorization': `Bearer ${apiKey}`
    };
    let videoUrl = null;
    let retries = 0;
    const maxRetries = config.maxRetries; // 从配置文件中获取最大重试次数
    const pollingInterval = config.pollingInterval; // 从配置文件中获取轮询间隔
    while (retries < maxRetries) {
        try {
            const res = await axios.get(url, { headers });
            // res.data.output.task_status 可能为 INITIAL, RUNNING, SUCCEEDED, FAILED, CANCELLED  
            // res.data.output.video_url 只有在 SUCCEEDED 时有值  
            if (res.data.output.task_status === 'SUCCEEDED') {
                videoUrl = res.data.output.video_url;
                break; // 如果状态为 SUCCEEDED，则跳出循环
            }
            else if (res.data.output.task_status === 'FAILED' || res.data.output.task_status === 'CANCELLED') {
                throw res.data.output.task_status; // 如果状态为 FAILED 或 CANCELLED，则抛出错误
            }
            // 如果状态不是 SUCCEEDED，则等待一段时间后继续轮询
            await new Promise(resolve => setTimeout(resolve, pollingInterval));
            retries++;
        }
        catch (err) {
            throw err.response?.data || err.message;
        }
    }
    if (!videoUrl) {
        throw new Error('视频生成任务超时或失败');
    }
    return videoUrl;
}