KnowledgeBase/server/scripts/test-vision-pipeline.js

"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 });
exports.testVisionPipeline = testVisionPipeline;
const core_1 = require("@nestjs/core");
const app_module_1 = require("./src/app.module");
const vision_pipeline_service_1 = require("./src/vision-pipeline/vision-pipeline.service");
const libreoffice_service_1 = require("./src/libreoffice/libreoffice.service");
const pdf2image_service_1 = require("./src/pdf2image/pdf2image.service");
const fs = __importStar(require("fs/promises"));
const path = __importStar(require("path"));
async function testVisionPipeline() {
    console.log('🚀 开始 Vision Pipeline 端到端测试\n');
    const app = await core_1.NestFactory.createApplicationContext(app_module_1.AppModule, {
        logger: ['error', 'warn', 'log'],
    });
    try {
        console.log('=== 测试 1: LibreOffice 服务 ===');
        const libreOffice = app.get(libreoffice_service_1.LibreOfficeService);
        const isHealthy = await libreOffice.healthCheck();
        console.log(`LibreOffice 健康检查: ${isHealthy ? '✅ 通过' : '❌ 失败'}`);
        if (!isHealthy) {
            console.log('⚠️  LibreOffice 服务未运行，跳过后续测试');
            return;
        }
        console.log('\n=== 测试 2: PDF 转图片服务 ===');
        const pdf2Image = app.get(pdf2image_service_1.Pdf2ImageService);
        const testPdf = '/home/fzxs/workspaces/demo/simple-kb/uploads/file-1766236004300-577549403.pdf';
        if (await fs.access(testPdf).then(() => true).catch(() => false)) {
            console.log(`测试 PDF: ${path.basename(testPdf)}`);
            const result = await pdf2Image.convertToImages(testPdf, {
                density: 150,
                quality: 75,
                format: 'jpeg',
            });
            console.log(`✅ 转换成功: ${result.images.length}/${result.totalPages} 页`);
            console.log(`   成功: ${result.successCount}, 失败: ${result.failedCount}`);
            await pdf2Image.cleanupImages(result.images);
            console.log('✅ 临时文件已清理');
        }
        else {
            console.log('⚠️  测试 PDF 文件不存在，跳过此测试');
        }
        console.log('\n=== 测试 3: Vision Pipeline 完整流程 ===');
        const visionPipeline = app.get(vision_pipeline_service_1.VisionPipelineService);
        const testFiles = [
            '/home/fzxs/workspaces/demo/simple-kb/uploads/file-1766236004300-577549403.pdf',
            '/home/fzxs/workspaces/demo/simple-kb/uploads/file-1765705143480-947461268.pdf',
        ];
        let testFile = null;
        for (const file of testFiles) {
            if (await fs.access(file).then(() => true).catch(() => false)) {
                testFile = file;
                break;
            }
        }
        if (testFile) {
            console.log(`测试文件: ${path.basename(testFile)}`);
            const recommendation = await visionPipeline.recommendMode(testFile);
            console.log(`模式推荐: ${recommendation.recommendedMode}`);
            console.log(`原因: ${recommendation.reason}`);
            if (recommendation.estimatedCost) {
                console.log(`预估成本: $${recommendation.estimatedCost.toFixed(2)}`);
            }
            if (recommendation.estimatedTime) {
                console.log(`预估时间: ${recommendation.estimatedTime.toFixed(1)}s`);
            }
            if (recommendation.warnings && recommendation.warnings.length > 0) {
                console.log(`警告: ${recommendation.warnings.join(', ')}`);
            }
            console.log('\n✅ Vision Pipeline 模块已正确配置');
            console.log('   注意: 完整流程测试需要配置有效的 Vision 模型 API Key');
        }
        else {
            console.log('⚠️  未找到测试文件，跳过完整流程测试');
        }
        console.log('\n=== 测试 4: 环境配置检查 ===');
        const configService = app.get(require('@nestjs/config').ConfigService);
        const requiredEnvVars = [
            'LIBREOFFICE_URL',
            'TEMP_DIR',
            'ELASTICSEARCH_HOST',
            'TIKA_HOST',
        ];
        for (const envVar of requiredEnvVars) {
            const value = configService.get(envVar);
            if (value) {
                console.log(`✅ ${envVar}: ${value}`);
            }
            else {
                console.log(`❌ ${envVar}: 未配置`);
            }
        }
        console.log('\n🎉 所有基础测试完成！');
    }
    catch (error) {
        console.error('❌ 测试失败:', error.message);
        console.error(error.stack);
    }
    finally {
        await app.close();
    }
}
if (require.main === module) {
    testVisionPipeline().catch(console.error);
}
//# sourceMappingURL=test-vision-pipeline.js.map