jimeng-ai-mcp/dist/examples/check-video-task.js

火山引擎即梦AI多模态生成服务MCP工具

"use strict";
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
    if (k2 === undefined) k2 = k;
    var desc = Object.getOwnPropertyDescriptor(m, k);
    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
      desc = { enumerable: true, get: function() { return m[k]; } };
    }
    Object.defineProperty(o, k2, desc);
}) : (function(o, m, k, k2) {
    if (k2 === undefined) k2 = k;
    o[k2] = m[k];
}));
var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
    Object.defineProperty(o, "default", { enumerable: true, value: v });
}) : function(o, v) {
    o["default"] = v;
});
var __importStar = (this && this.__importStar) || (function () {
    var ownKeys = function(o) {
        ownKeys = Object.getOwnPropertyNames || function (o) {
            var ar = [];
            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
            return ar;
        };
        return ownKeys(o);
    };
    return function (mod) {
        if (mod && mod.__esModule) return mod;
        var result = {};
        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
        __setModuleDefault(result, mod);
        return result;
    };
})();
Object.defineProperty(exports, "__esModule", { value: true });
const jimeng_client_1 = require("../src/jimeng-client");
const fs = __importStar(require("fs"));
/**
 * 从.env文件加载环境变量
 */
const loadEnv = (filePath = '.env') => {
    try {
        if (fs.existsSync(filePath)) {
            const envData = fs.readFileSync(filePath, 'utf8');
            envData.split('\n').forEach(line => {
                const match = line.match(/^([^=]+)=(.*)$/);
                if (match) {
                    process.env[match[1].trim()] = match[2].trim();
                }
            });
        }
    }
    catch (err) {
        console.error('无法加载.env文件:', err);
    }
};
/**
 * 主函数
 */
const main = async () => {
    try {
        // 加载环境变量
        loadEnv();
        // 获取任务ID参数
        const taskId = process.argv[2];
        if (!taskId) {
            console.error('错误: 缺少任务ID参数');
            console.error('用法: npx ts-node examples/check-video-task.ts <任务ID>');
            return;
        }
        // 获取API密钥
        const accessKey = process.env.JIMENG_ACCESS_KEY;
        const secretKey = process.env.JIMENG_SECRET_KEY;
        if (!accessKey || !secretKey) {
            console.error('错误: 缺少API密钥。请设置JIMENG_ACCESS_KEY和JIMENG_SECRET_KEY环境变量。');
            return;
        }
        // 创建客户端
        const client = new jimeng_client_1.JimengClient({
            accessKey,
            secretKey,
            debug: true // 打开调试模式
        });
        console.log('=== 即梦AI视频任务查询 ===');
        console.log(`正在查询任务: ${taskId}`);
        // 获取任务结果
        const resultResponse = await client.getVideoResult({
            task_id: taskId
        });
        // 显示任务状态
        console.log('\n任务状态:', resultResponse.status || '未知');
        if (resultResponse.status === "SUCCEED" && resultResponse.video_urls && resultResponse.video_urls.length > 0) {
            // 任务成功完成
            console.log('\n✓ 视频生成成功!');
            console.log('\n生成的视频链接:');
            resultResponse.video_urls.forEach((url, index) => {
                console.log(`${index + 1}. ${url}`);
            });
        }
        else if (resultResponse.status === "RUNNING") {
            // 任务仍在进行中
            console.log('\n任务仍在处理中，请稍后再次查询');
            console.log('重新检查命令: npx ts-node examples/check-video-task.ts ' + taskId);
        }
        else {
            // 任务失败或其他状态
            console.error('\n视频生成未成功:', resultResponse.error);
        }
    }
    catch (error) {
        console.error('发生错误:', error);
    }
};
// 运行主函数
main();