jimeng-ai-mcp/dist/examples/dog-generator.js

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

"use strict";
/**
 * 火山引擎即梦AI小狗图像生成测试
 */
Object.defineProperty(exports, "__esModule", { value: true });
const src_1 = require("../src");
// 访问密钥设置
const ACCESS_KEY = 'AKLTYjZkZWFlNGJkMDY3NDVkOGFiYTczYmViNmI5NDZkMjY';
const SECRET_KEY = 'Tm1abE9HSTVPRFV3WldFeU5EZGlOemcyWlRJMk1HUmpOalF4WmpNd1pqSQ==';
async function main() {
    var _a, _b;
    try {
        console.log('即梦AI小狗图像生成测试');
        console.log('-------------------');
        const prompt = '一只可爱的小狗，毛发蓬松，坐在草地上，阳光照射，高清细节，写实风格';
        console.log('提示词:', prompt);
        // 创建客户端实例
        const client = new src_1.JimengV4.JimengClient({
            debug: true, // 设置为true可以查看详细日志
            accessKey: ACCESS_KEY,
            secretKey: SECRET_KEY
        });
        console.log('\n正在生成图像，请稍候...');
        // 生成图像
        const startTime = Date.now();
        const result = await client.generateImage({
            prompt,
            width: 512,
            height: 512
        });
        const endTime = Date.now();
        console.log(`生成耗时: ${(endTime - startTime) / 1000}秒`);
        if (result.success && result.image_urls && result.image_urls.length > 0) {
            console.log('\n图像生成成功!');
            console.log('图像URL:');
            result.image_urls.forEach((url, index) => {
                console.log(`[${index + 1}] ${url}`);
            });
            // 如果有LLM优化后的提示词，显示出来
            if ((_b = (_a = result.raw_response) === null || _a === void 0 ? void 0 : _a.data) === null || _b === void 0 ? void 0 : _b.rephraser_result) {
                console.log('\nLLM优化后的提示词:');
                console.log(result.raw_response.data.rephraser_result);
            }
        }
        else {
            console.error('\n图像生成失败:', result.error);
            // 打印详细错误信息
            if (result.raw_response) {
                console.error('原始响应:', JSON.stringify(result.raw_response, null, 2));
            }
        }
    }
    catch (error) {
        console.error('\n发生错误:', error instanceof Error ? error.message : String(error));
        if (error instanceof Error && error.stack) {
            console.error('错误堆栈:', error.stack);
        }
    }
}
// 运行主函数
main();