refactor: 重构服务器部分并整理目录和注释

scripts/genkey.js

@@ -0,0 +1,21 @@
+/**
+ * @fileoverview 生成 API Key（CLI）
+ * @description 输出一个新的 `server.auth` Key，供写入 `config.yaml` 使用。
+ *
+ * 用法：`npm run genkey`
+ */
+
+import crypto from 'crypto';
+
+/**
+ * 生成随机 API Key（用于 `config.yaml` 的 `server.auth`）
+ * 格式：sk-{48位十六进制字符}
+ * @returns {string} API Key
+ */
+function generateApiKey() {
+    return 'sk-' + crypto.randomBytes(24).toString('hex');
+}
+
+console.log('>>> [GenAPIKey] 生成新的 API Key:');
+console.log(generateApiKey());
+console.log('\n>>> 请将此 Key 复制到 config.yaml 文件的 server.auth 字段中。');

scripts/init.js

@@ -0,0 +1,537 @@
+/**
+ * @fileoverview 运行环境初始化脚本（CLI）
+ * @description 用于下载/准备运行所需依赖（如 Camoufox、better-sqlite3 等），并按需更新 `config.yaml`。
+ *
+ * 用法：`npm run init`
+ */
+
+import fs from 'fs';
+import { execSync } from 'child_process';
+import path from 'path';
+import os from 'os';
+import { fileURLToPath } from 'url';
+import { gotScraping } from 'got-scraping';
+import compressing from 'compressing';
+import yaml from 'yaml';
+import { logger } from '../src/utils/logger.js';
+import { getHttpProxy, getProxyConfig } from '../src/utils/proxy.js';
+
+const __dirname = path.dirname(fileURLToPath(import.meta.url));
+const PROJECT_ROOT = path.join(__dirname, '..');
+const TEMP_DIR = path.join(PROJECT_ROOT, 'data', 'temp');
+const CONFIG_PATH = path.join(PROJECT_ROOT, 'config.yaml');
+
+// 确保临时目录存在
+if (!fs.existsSync(TEMP_DIR)) {
+    fs.mkdirSync(TEMP_DIR, { recursive: true });
+}
+
+/**
+ * 获取 Node.js ABI 版本
+ */
+function getNodeABI() {
+    return process.versions.modules;
+}
+
+/**
+ * 获取平台信息
+ */
+function getPlatformInfo() {
+    const platform = os.platform();
+    const arch = os.arch();
+    const nodeVersion = process.version;
+    const abi = getNodeABI();
+
+    return { platform, arch, nodeVersion, abi };
+}
+
+/**
+ * 验证平台支持
+ */
+function validatePlatform(platform, arch) {
+    const supported = {
+        'win32': ['x64'],
+        'darwin': ['x64', 'arm64'],
+        'linux': ['x64', 'arm64']
+    };
+
+    if (!supported[platform] || !supported[platform].includes(arch)) {
+        return false;
+    }
+
+    return true;
+}
+
+/**
+ * 验证 Node.js ABI 版本支持
+ */
+function validateABI(abi) {
+    const supportedABIs = [115, 121, 123, 125, 127, 128, 130, 131, 132, 133, 135, 136, 137, 139, 140, 141];
+    return supportedABIs.includes(parseInt(abi, 10));
+}
+
+/**
+ * 下载文件（带进度，流式，支持重试）
+ * @param {string} url - 下载地址
+ * @param {string} destPath - 目标文件路径
+ * @param {string|null} proxyUrl - 代理 URL
+ * @param {number} maxRetries - 最大重试次数
+ */
+async function downloadFile(url, destPath, proxyUrl = null, maxRetries = 3) {
+    for (let attempt = 1; attempt <= maxRetries; attempt++) {
+        try {
+            if (attempt > 1) {
+                logger.info('初始化', `第 ${attempt}/${maxRetries} 次尝试下载...`);
+                // 删除之前失败的文件
+                try {
+                    if (fs.existsSync(destPath)) {
+                        fs.unlinkSync(destPath);
+                    }
+                } catch (e) { }
+            } else {
+                logger.info('初始化', `开始下载: ${url}`);
+            }
+
+            await downloadFileOnce(url, destPath, proxyUrl);
+            return destPath;
+        } catch (error) {
+            logger.error('初始化', `下载失败 (尝试 ${attempt}/${maxRetries}): ${error.message}`);
+
+            if (attempt === maxRetries) {
+                throw error;
+            }
+
+            // 等待后重试（递增延迟）
+            const delay = attempt * 2000;
+            logger.info('初始化', `${delay / 1000} 秒后重试...`);
+            await new Promise(resolve => setTimeout(resolve, delay));
+        }
+    }
+}
+
+/**
+ * 单次下载尝试（内部函数）
+ */
+async function downloadFileOnce(url, destPath, proxyUrl = null) {
+    return new Promise((resolve, reject) => {
+        const options = {
+            http2: false,
+            timeout: {
+                request: 900000,  // 总请求超时 15 分钟
+                read: 180000      // 两次数据接收间隔超时 3 分钟
+            },
+            retry: {
+                limit: 0
+            },
+            headerGeneratorOptions: {
+                browsers: [],
+                devices: [],
+                locales: [],
+                operatingSystems: []
+            },
+            headers: {
+                'user-agent': 'Wget/1.21.4 (linux-gnu)',
+                'accept': '*/*',
+                'accept-encoding': 'identity',
+                'connection': 'keep-alive'
+            }
+        };
+
+        if (proxyUrl) {
+            options.proxyUrl = proxyUrl;
+        }
+
+        const downloadStream = gotScraping.stream(url, options);
+        const fileStream = fs.createWriteStream(destPath);
+
+        let downloadedSize = 0;
+        let totalSize = 0;
+        let lastLogTime = Date.now();
+
+        downloadStream.on('response', (response) => {
+            totalSize = parseInt(response.headers['content-length'] || '0', 10);
+            if (totalSize > 0) {
+                logger.info('初始化', `文件大小: ${(totalSize / 1024 / 1024).toFixed(2)} MB`);
+            }
+        });
+
+        downloadStream.on('data', (chunk) => {
+            downloadedSize += chunk.length;
+
+            // 每秒更新一次进度
+            const now = Date.now();
+            if (totalSize > 0 && now - lastLogTime > 1000) {
+                const percent = ((downloadedSize / totalSize) * 100).toFixed(1);
+                const downloadedMB = (downloadedSize / 1024 / 1024).toFixed(2);
+                const totalMB = (totalSize / 1024 / 1024).toFixed(2);
+                logger.info('初始化', `下载进度: ${percent}% (${downloadedMB}MB / ${totalMB}MB)`);
+                lastLogTime = now;
+            }
+        });
+
+        downloadStream.on('error', (error) => {
+            fileStream.close();
+            try {
+                fs.unlinkSync(destPath);
+            } catch (e) { }
+            reject(error);
+        });
+
+        fileStream.on('error', (error) => {
+            reject(error);
+        });
+
+        fileStream.on('finish', () => {
+            const finalSize = (downloadedSize / 1024 / 1024).toFixed(2);
+
+            // 验证下载完整性
+            if (totalSize > 0 && downloadedSize !== totalSize) {
+                const errorMsg = `下载不完整: 预期 ${(totalSize / 1024 / 1024).toFixed(2)} MB, 实际 ${finalSize} MB`;
+                logger.error('初始化', errorMsg);
+
+                // 清理损坏的文件
+                try {
+                    fs.unlinkSync(destPath);
+                } catch (e) { }
+
+                reject(new Error(errorMsg));
+                return;
+            }
+
+            logger.info('初始化', `下载完成: ${finalSize} MB`);
+            resolve(destPath);
+        });
+
+        downloadStream.pipe(fileStream);
+    });
+}
+
+/**
+ * 构建 better-sqlite3 下载 URL
+ */
+function getBetterSqlite3Url(platform, arch, abi) {
+    const version = '12.5.0';
+    const platformMap = {
+        'win32': 'win32',
+        'darwin': 'darwin',
+        'linux': 'linux'
+    };
+
+    const platformName = platformMap[platform];
+    const archName = arch; // x64 或 arm64
+
+    return `https://github.com/WiseLibs/better-sqlite3/releases/download/v${version}/better-sqlite3-v${version}-node-v${abi}-${platformName}-${archName}.tar.gz`;
+}
+
+/**
+ * 下载并安装 better-sqlite3
+ */
+async function installBetterSqlite3(platform, arch, abi, proxyUrl) {
+    logger.info('初始化', '开始安装 better-sqlite3...');
+
+    const url = getBetterSqlite3Url(platform, arch, abi);
+    const downloadPath = path.join(TEMP_DIR, 'better-sqlite3.tar.gz');
+
+    // 下载
+    await downloadFile(url, downloadPath, proxyUrl);
+
+    // 解压 .tar.gz 文件
+    logger.info('初始化', '正在解压 better-sqlite3...');
+    await compressing.tgz.uncompress(downloadPath, TEMP_DIR);
+
+    // 查找 better_sqlite3.node
+    const files = fs.readdirSync(TEMP_DIR, { recursive: true });
+    const nodeFile = files.find(f => f.endsWith('better_sqlite3.node'));
+    if (!nodeFile) {
+        throw new Error('未找到 better_sqlite3.node 文件');
+    }
+
+    // 复制到 node_modules
+    const buildDir = path.join(PROJECT_ROOT, 'node_modules', 'better-sqlite3', 'build', 'Release');
+    if (!fs.existsSync(buildDir)) {
+        fs.mkdirSync(buildDir, { recursive: true });
+    }
+
+    const sourcePath = path.join(TEMP_DIR, nodeFile);
+    const destPath = path.join(buildDir, 'better_sqlite3.node');
+    fs.copyFileSync(sourcePath, destPath);
+
+    logger.info('初始化', `better-sqlite3 安装成功: ${destPath}`);
+
+    // 清理
+    fs.unlinkSync(downloadPath);
+    // 清理解压后的所有文件
+    files.forEach(f => {
+        const filePath = path.join(TEMP_DIR, f);
+        try {
+            if (fs.existsSync(filePath)) {
+                const stat = fs.statSync(filePath);
+                if (stat.isDirectory()) {
+                    fs.rmSync(filePath, { recursive: true, force: true });
+                } else {
+                    fs.unlinkSync(filePath);
+                }
+            }
+        } catch (e) { }
+    });
+}
+
+/**
+ * 构建 Camoufox 下载 URL
+ */
+function getCamoufoxUrl(platform, arch) {
+    const version = '135.0.1-beta.24';
+    const platformMap = {
+        'win32': 'win',
+        'darwin': 'mac',
+        'linux': 'lin'
+    };
+
+    const archMap = {
+        'x64': 'x86_64',
+        'arm64': 'arm64'
+    };
+
+    const platformName = platformMap[platform];
+    const archName = archMap[arch];
+
+    return `https://github.com/daijro/camoufox/releases/download/v${version}/camoufox-${version}-${platformName}.${archName}.zip`;
+}
+
+/**
+ * 下载并安装 Camoufox
+ */
+async function installCamoufox(platform, arch, proxyUrl) {
+    logger.info('初始化', '开始安装 Camoufox 浏览器...');
+
+    const url = getCamoufoxUrl(platform, arch);
+    const downloadPath = path.join(TEMP_DIR, 'camoufox.zip');
+
+    // 下载
+    await downloadFile(url, downloadPath, proxyUrl);
+
+    // 解压 .zip 文件到 camoufox 目录
+    logger.info('初始化', '正在解压 Camoufox...');
+    const camoufoxDir = path.join(PROJECT_ROOT, 'camoufox');
+    if (!fs.existsSync(camoufoxDir)) {
+        fs.mkdirSync(camoufoxDir, { recursive: true });
+    }
+
+    await compressing.zip.uncompress(downloadPath, camoufoxDir);
+
+    // macOS 专用：复制 properties.json 到 MacOS 目录
+    if (platform === 'darwin') {
+        const resourcesPath = path.join(camoufoxDir, 'Camoufox.app', 'Contents', 'Resources', 'properties.json');
+        const macOSDir = path.join(camoufoxDir, 'Camoufox.app', 'Contents', 'MacOS');
+        const macOSPath = path.join(macOSDir, 'properties.json');
+
+        if (fs.existsSync(resourcesPath)) {
+            // 确保目标目录存在
+            if (!fs.existsSync(macOSDir)) {
+                fs.mkdirSync(macOSDir, { recursive: true });
+            }
+            fs.copyFileSync(resourcesPath, macOSPath);
+            logger.info('初始化', `已复制 properties.json 到 MacOS 目录`);
+        } else {
+            logger.warn('初始化', `未找到 properties.json: ${resourcesPath}`);
+        }
+    }
+
+    logger.info('初始化', `Camoufox 安装成功: ${camoufoxDir}`);
+
+    // 更新 config.yaml
+    updateConfigPath(platform, camoufoxDir);
+
+    // 创建 version.json
+    const versionJsonPath = path.join(camoufoxDir, 'version.json');
+    const versionData = {
+        version: "135.0",
+        release: "beta.24"
+    };
+    fs.writeFileSync(versionJsonPath, JSON.stringify(versionData, null, 2), 'utf8');
+    logger.info('初始化', `已生成 version.json: ${versionJsonPath}`);
+
+    // 清理
+    fs.unlinkSync(downloadPath);
+}
+
+/**
+ * 更新 config.yaml 中的 browser.path
+ */
+function updateConfigPath(platform, camoufoxDir) {
+    try {
+        const configContent = fs.readFileSync(CONFIG_PATH, 'utf8');
+
+        // 解析为文档对象 (CST)
+        const doc = yaml.parseDocument(configContent);
+
+        // 构造绝对路径
+        let browserPath;
+        if (platform === 'win32') {
+            browserPath = path.join(camoufoxDir, 'camoufox.exe');
+        } else if (platform === 'darwin') {
+            browserPath = path.join(camoufoxDir, 'Camoufox.app', 'Contents', 'MacOS', 'camoufox');
+        } else {
+            browserPath = path.join(camoufoxDir, 'camoufox');
+        }
+
+        // 规范化路径分隔符
+        browserPath = browserPath.replace(/\\/g, '/');
+
+        // 安全地更新路径，如果节点不存在则创建
+        if (!doc.has('browser')) {
+            doc.set('browser', { path: browserPath });
+        } else {
+            const browserNode = doc.get('browser');
+            if (browserNode && typeof browserNode.set === 'function') {
+                browserNode.set('path', browserPath);
+            } else {
+                // 如果 browser 不是对象（理论上不应该发生），强制覆盖
+                doc.set('browser', { path: browserPath });
+            }
+        }
+
+        // 转回字符串，保留注释
+        const updatedYaml = doc.toString();
+        fs.writeFileSync(CONFIG_PATH, updatedYaml, 'utf8');
+
+        logger.info('初始化', `已更新配置文件 browser.path: ${browserPath}`);
+    } catch (e) {
+        logger.error('初始化', '更新配置文件失败', { error: e.message });
+    }
+}
+
+/**
+ * 主流程
+ */
+(async () => {
+    try {
+        logger.info('初始化', '========================================');
+        logger.info('初始化', '依赖初始化脚本启动');
+        logger.info('初始化', '========================================');
+
+        // 显示系统信息
+        const { platform, arch, nodeVersion, abi } = getPlatformInfo();
+        logger.info('初始化', `操作系统: ${platform}`);
+        logger.info('初始化', `芯片架构: ${arch}`);
+        logger.info('初始化', `Node.js 版本: ${nodeVersion}`);
+        logger.info('初始化', `Node.js ABI 版本: ${abi}`);
+
+        // 验证平台支持
+        if (!validatePlatform(platform, arch)) {
+            logger.error('初始化', '不支持的平台！');
+            logger.error('初始化', `因该项目使用了 Camoufox 浏览器，没有您设备可用的预编译版本`);
+            logger.error('初始化', `支持的平台: Windows x64, macOS x64/arm64, Linux x64/arm64`);
+            process.exit(1);
+        }
+
+        logger.info('初始化', '平台支持检查通过');
+
+        // 验证 ABI 版本支持
+        if (!validateABI(abi)) {
+            logger.error('初始化', '不支持的 Node.js ABI 版本！');
+            logger.error('初始化', `当前 ABI 版本: ${abi}`);
+            logger.error('初始化', `支持的 ABI 版本: 115, 121, 123, 125, 127, 128, 130, 131, 132, 133, 135, 136, 137, 139, 140, 141`);
+            logger.error('初始化', `建议使用 Node.js 20.10.0 或更高版本`);
+            process.exit(1);
+        }
+
+        logger.info('初始化', 'ABI 版本检查通过');
+
+        // 读取并转换代理配置
+        let proxyUrl = null;
+        try {
+            const configContent = fs.readFileSync(CONFIG_PATH, 'utf8');
+            const config = yaml.parse(configContent);
+            const proxyConfig = getProxyConfig(config);
+            if (proxyConfig) {
+                proxyUrl = await getHttpProxy(proxyConfig);
+            }
+        } catch (e) {
+            logger.warn('初始化', '无法读取配置文件或转换代理，不使用代理');
+        }
+
+        // 安装 better-sqlite3
+        await installBetterSqlite3(platform, arch, abi, proxyUrl);
+
+        // 安装 Camoufox
+        await installCamoufox(platform, arch, proxyUrl);
+
+        // 修复 Camoufox 环境 (Linux)
+        fixCamoufoxEnv();
+
+        logger.info('初始化', '========================================');
+        logger.info('初始化', '所有依赖安装完成！');
+        logger.info('初始化', '========================================');
+        process.exit(0);
+
+    } catch (err) {
+        logger.error('初始化', '初始化失败', { error: err.message });
+        process.exit(1);
+    }
+})();
+
+/**
+ * 自动修复 Linux 下 Camoufox 的路径依赖
+ * 目的：建立软链接，欺骗 camoufox-js 以为浏览器安装在默认目录，从而防止自动下载
+ */
+function fixCamoufoxEnv() {
+    // 1. 仅在 Linux 下执行
+    if (os.platform() !== 'linux') return;
+
+    logger.info('初始化', '正在检查 Camoufox 环境配置...');
+
+    // --- 路径配置 ---
+    // 假设浏览器存放在项目根目录下的 camoufox 文件夹中
+    // 依赖 init.js 中已定义的 PROJECT_ROOT 变量
+    const customBrowserDir = path.join(PROJECT_ROOT, 'camoufox');
+
+    // 官方默认缓存路径: ~/.cache/camoufox
+    const defaultCacheDir = path.join(os.homedir(), '.cache');
+    const defaultLinkPath = path.join(defaultCacheDir, 'camoufox');
+
+    // 2. 预检查：确保源文件存在
+    if (!fs.existsSync(customBrowserDir)) {
+        logger.warn('初始化', `未找到自定义浏览器目录: ${customBrowserDir}`);
+        logger.warn('初始化', `请确保已将浏览器解压至项目根目录的 camoufox 文件夹`);
+        return;
+    }
+
+    // 3. 检查并修复软链接
+    if (fs.existsSync(defaultLinkPath)) {
+        const stats = fs.lstatSync(defaultLinkPath);
+        if (stats.isSymbolicLink()) {
+            const currentTarget = fs.readlinkSync(defaultLinkPath);
+            if (currentTarget === customBrowserDir) {
+                logger.info('初始化', 'Camoufox 路径映射已就绪');
+                return;
+            }
+            logger.info('初始化', '路径映射不一致，正在更新...');
+            fs.unlinkSync(defaultLinkPath);
+        } else {
+            // 备份旧的实体文件夹
+            logger.warn('初始化', `默认路径被占用，正在备份...`);
+            fs.renameSync(defaultLinkPath, `${defaultLinkPath}_backup_${Date.now()}`);
+        }
+    } else {
+        if (!fs.existsSync(defaultCacheDir)) {
+            fs.mkdirSync(defaultCacheDir, { recursive: true });
+        }
+    }
+
+    // 4. 创建软链接
+    try {
+        // 使用 shell 命令创建软链接，比 fs.symlinkSync 在某些 Linux 环境下更可靠
+        execSync(`ln -sf "${customBrowserDir}" "${defaultLinkPath}"`);
+
+        // 验证链接是否有效
+        if (fs.existsSync(defaultLinkPath)) {
+            const linkTarget = fs.readlinkSync(defaultLinkPath);
+            logger.info('初始化', `成功创建路径映射: ${defaultLinkPath} -> ${linkTarget}`);
+        } else {
+            logger.warn('初始化', '软链接创建后无法访问，可能存在权限问题');
+        }
+    } catch (e) {
+        logger.error('初始化', `创建软链接失败: ${e.message}`);
+    }
+}

scripts/test.js

@@ -0,0 +1,234 @@
+/**
+ * @fileoverview 本地 HTTP 调用测试（CLI）
+ * @description 用交互式方式构造请求并调用本地服务的 `/v1/chat/completions`，用于快速验证服务可用性与流式输出。
+ *
+ * 用法：`npm run test`
+ */
+
+import { select, input } from '@inquirer/prompts';
+import fs from 'fs';
+import path from 'path';
+import http from 'http';
+import yaml from 'yaml';
+
+// 简易日志：脚本内部使用，避免引入服务端 logger 造成格式混淆
+const logger = {
+    info: (tag, msg) => console.log(`[${new Date().toLocaleTimeString()}] [INFO] [${tag}] ${msg}`),
+    warn: (tag, msg) => console.log(`[${new Date().toLocaleTimeString()}] [WARN] [${tag}] ${msg}`),
+    error: (tag, msg, meta) => console.error(`[${new Date().toLocaleTimeString()}] [ERROR] [${tag}] ${msg}`, meta || '')
+};
+
+// 读取本地配置：用于获取端口与鉴权 Token（读取失败时使用默认值）
+let config = { server: { port: 3000, auth: '' } };
+try {
+    if (fs.existsSync('config.yaml')) {
+        const file = fs.readFileSync('config.yaml', 'utf8');
+        const parsed = yaml.parse(file);
+        if (parsed && parsed.server) {
+            config.server.port = parsed.server.port || 3000;
+            config.server.auth = parsed.server.auth || '';
+        }
+    }
+} catch (e) {
+    logger.warn('Test', '无法读取 config.yaml，将使用默认设置');
+}
+
+/**
+ * 输入提示词
+ */
+async function promptForInput() {
+    const prompt = await input({
+        message: '输入提示词 (必填)',
+        validate: (val) => val.trim().length > 0 || '提示词不能为空'
+    });
+    return prompt.trim();
+}
+
+/**
+ * 输入图片路径
+ */
+async function promptForImages() {
+    const imagePaths = [];
+    while (true) {
+        const imgPath = await input({
+            message: `输入参考图片路径 (留空跳过，已添加 ${imagePaths.length} 张)`,
+        });
+
+        if (!imgPath.trim()) break;
+
+        const cleanPath = imgPath.trim().replace(/^["']|["']$/g, '');
+        if (fs.existsSync(cleanPath)) {
+            imagePaths.push(cleanPath);
+        } else {
+            logger.warn('Test', `图片不存在: ${cleanPath}`);
+        }
+    }
+    return imagePaths;
+}
+
+/**
+ * HTTP 测试模式 - OpenAI 格式
+ */
+async function testViaHttpOpenAI(prompt, modelId, imagePaths, isStreaming) {
+    const PORT = config.server.port;
+    const AUTH_TOKEN = config.server.auth;
+
+    if (!AUTH_TOKEN) {
+        logger.warn('Test', '警告: 未配置 API Key (server.auth)');
+    }
+
+    logger.info('Test', `HTTP 测试 - ${isStreaming ? '流式' : '非流式'} - 端口: ${PORT}`);
+
+    return new Promise((resolve, reject) => {
+        // 构造请求体
+        const messages = [];
+        const lastMessage = { role: 'user', content: [] };
+
+        if (prompt) {
+            lastMessage.content.push({ type: 'text', text: prompt });
+        }
+
+        for (const imgPath of imagePaths) {
+            if (fs.existsSync(imgPath)) {
+                const buffer = fs.readFileSync(imgPath);
+                const base64 = buffer.toString('base64');
+                const ext = path.extname(imgPath).slice(1).toLowerCase();
+                const mimeType = ext === 'jpg' ? 'jpeg' : ext;
+                lastMessage.content.push({
+                    type: 'image_url',
+                    image_url: { url: `data:image/${mimeType};base64,${base64}` }
+                });
+            } else {
+                logger.warn('Test', `图片不存在，已跳过: ${imgPath}`);
+            }
+        }
+
+        messages.push(lastMessage);
+
+        const body = {
+            messages,
+            stream: isStreaming,
+            model: modelId || 'default'
+        };
+
+        const bodyStr = JSON.stringify(body);
+
+        const options = {
+            hostname: '127.0.0.1',
+            port: PORT,
+            path: '/v1/chat/completions',
+            method: 'POST',
+            headers: {
+                'Content-Type': 'application/json',
+                'Content-Length': Buffer.byteLength(bodyStr),
+                'Authorization': `Bearer ${AUTH_TOKEN}`
+            }
+        };
+
+        const req = http.request(options, (res) => {
+            if (isStreaming) {
+                // 流式响应
+                let buffer = '';
+                let contentReceived = '';
+
+                res.on('data', chunk => {
+                    buffer += chunk.toString();
+                    const lines = buffer.split('\n');
+                    buffer = lines.pop(); // 保留未完成的行
+
+                    for (const line of lines) {
+                        if (!line.trim()) continue;
+
+                        // 心跳注释
+                        if (line.startsWith(':')) {
+                            process.stdout.write('💓');  // 显示心跳
+                            continue;
+                        }
+
+                        if (line.startsWith('data:')) {
+                            const data = line.slice(5).trim();
+                            if (data === '[DONE]') {
+                                console.log('\n📦 [DONE]');
+                                continue;
+                            }
+
+                            try {
+                                const chunk = JSON.parse(data);
+                                if (chunk.choices && chunk.choices[0].delta && chunk.choices[0].delta.content) {
+                                    const content = chunk.choices[0].delta.content;
+                                    contentReceived += content;
+                                    process.stdout.write(content); // 实时输出内容
+                                }
+                                if (chunk.error) {
+                                    console.log(`\n❌ 错误: ${chunk.error}`);
+                                }
+                            } catch (e) {
+                                // 忽略解析错误
+                            }
+                        }
+                    }
+                });
+
+                res.on('end', () => {
+                    console.log(''); // 换行
+                    if (res.statusCode === 200) {
+                        resolve({ choices: [{ message: { content: contentReceived } }] });
+                    } else {
+                        reject(new Error(`HTTP ${res.statusCode}`));
+                    }
+                });
+            } else {
+                // 非流式响应
+                let data = '';
+                res.on('data', chunk => data += chunk);
+                res.on('end', () => {
+                    if (res.statusCode === 200) {
+                        const response = JSON.parse(data);
+                        resolve(response);
+                    } else {
+                        reject(new Error(`HTTP ${res.statusCode}: ${data}`));
+                    }
+                });
+            }
+        });
+
+        req.on('error', reject);
+        req.write(bodyStr);
+        req.end();
+    });
+}
+
+/**
+ * 主流程
+ */
+(async () => {
+    try {
+        logger.info('Test', '=== API 独立测试脚本 ===');
+
+        // 1. 输入提示词
+        const prompt = await promptForInput();
+
+        // 2. 输入图片路径
+        const imagePaths = await promptForImages();
+
+        // 3. 选择流式模式
+        const isStreaming = await select({
+            message: '选择请求模式',
+            choices: [
+                { name: '流式 (stream: true)', value: true },
+                { name: '非流式 (stream: false)', value: false }
+            ]
+        });
+
+        // 4. 执行测试
+        logger.info('Test', '正在发送请求...');
+        await testViaHttpOpenAI(prompt, null, imagePaths, isStreaming);
+
+        logger.info('Test', '测试完成');
+        process.exit(0);
+
+    } catch (err) {
+        logger.error('Test', '测试失败', { error: err.message });
+        process.exit(1);
+    }
+})();