diff --git a/scripts/verify_parsing.js b/scripts/verify_parsing.js deleted file mode 100644 index 6a56527..0000000 --- a/scripts/verify_parsing.js +++ /dev/null @@ -1,90 +0,0 @@ - -import { parseRequest } from '../src/server/parseChat.js'; - -// Mock options -const mockOptions = { - tempDir: './data/temp', - imageLimit: 5, - backendName: 'test-backend', - resolveModelId: (id) => id === 'gpt-4-text' ? 'resolved-gpt-4' : null, - getImagePolicy: () => 'optional', - getModelType: (id) => id === 'gpt-4-text' ? 'text' : 'image', - requestId: 'test-req-id', - logger: { info: console.log, error: console.error, warn: console.warn, debug: console.log } -}; - -async function testTextParsing() { - console.log('--- Testing Text Model Parsing ---'); - const requestData = { - model: 'gpt-4-text', - stream: true, - messages: [ - { role: 'system', content: 'You are a cat.' }, - { role: 'user', content: 'Hello' }, - { role: 'assistant', content: 'Meow' }, - { - role: 'user', content: [ - { type: 'text', text: 'Look at this ' }, - { type: 'image_url', image_url: { url: 'https://example.com/image.png' } } - ] - } - ] - }; - - const result = await parseRequest(requestData, mockOptions); - - if (result.success) { - console.log('Success!'); - console.log('Prompt:\n' + result.data.prompt); - console.log('Image Paths:', result.data.imagePaths); - - if (result.data.prompt.includes('=== 系统指令') && result.data.prompt.includes('User: Look at this')) { - console.log('✅ Text Parsing Verification PASSED'); - } else { - console.error('❌ Text Parsing Verification FAILED'); - } - - } else { - console.error('Failed:', result.error); - } -} - -async function testImageParsing() { - console.log('\n--- Testing Image Model Parsing (Legacy) ---'); - const requestData = { - model: 'some-image-model', - stream: false, - messages: [ - { role: 'user', content: 'Draw a cat' } - ] - }; - - const options = { - ...mockOptions, - resolveModelId: (id) => 'resolved-image-model', - getModelType: () => 'image' - }; - - const result = await parseRequest(requestData, options); - - if (result.success) { - console.log('Success!'); - console.log('Prompt:', result.data.prompt); - if (result.data.prompt === 'Draw a cat') { - console.log('✅ Image Parsing Verification PASSED'); - } else { - console.error('❌ Image Parsing Verification FAILED'); - } - } else { - console.error('Failed:', result.error); - } -} - -(async () => { - try { - await testTextParsing(); - await testImageParsing(); - } catch (e) { - console.error(e); - } -})(); diff --git a/src/backend/adapter/lmarena_text.js b/src/backend/adapter/lmarena_text.js index 53dd1bc..0f269e7 100644 --- a/src/backend/adapter/lmarena_text.js +++ b/src/backend/adapter/lmarena_text.js @@ -22,9 +22,7 @@ import { logger } from '../../utils/logger.js'; // --- 配置常量 --- const TARGET_URL = 'https://lmarena.ai/c/new?mode=direct'; - - - +const TARGET_URL_SEARCH = 'https://lmarena.ai/zh/c/new?mode=direct&chat-modality=search'; /** * 执行生图任务 @@ -39,9 +37,13 @@ async function generateImage(context, prompt, imgPaths, modelId, meta = {}) { const { page, config } = context; const textareaSelector = 'textarea'; + // 根据 modelId (codeName) 反查模型配置,判断是否为搜索模式 + const modelConfig = manifest.models.find(m => m.codeName === modelId) || {}; + const targetUrl = modelConfig.search ? TARGET_URL_SEARCH : TARGET_URL; + try { - logger.info('适配器', '开启新会话...', meta); - const gotoResult = await gotoWithCheck(page, TARGET_URL); + logger.info('适配器', `开启新会话... (搜索模式: ${!!modelConfig.search})`, meta); + const gotoResult = await gotoWithCheck(page, targetUrl); if (gotoResult.error) return gotoResult; // 1. 等待输入框加载 @@ -276,9 +278,10 @@ export const manifest = { { id: 'amazon.nova-pro-v1:0', codeName: 'a14546b5-d78d-4cf6-bb61-ab5b8510a9d6', imagePolicy: 'optional', type: 'text' }, { id: 'glm-4.6v', codeName: '019b151a-7c3b-72a2-8811-0bf9317c2ef5', imagePolicy: 'optional', type: 'text' }, { id: 'gpt-5.2-high', codeName: '019b1448-dafa-7f92-90c3-50e159c2263c', imagePolicy: 'optional', type: 'text' }, - { id: 'gpt-5.2', codeName: '019b1448-d548-78f4-8b98-788d72cbd057', imagePolicy: 'foptional', type: 'text' }, + { id: 'gpt-5.2', codeName: '019b1448-d548-78f4-8b98-788d72cbd057', imagePolicy: 'optional', type: 'text' }, { id: 'glm-4.6v-flash', codeName: '019b1536-49c0-73b2-8d45-403b8571568d', imagePolicy: 'optional', type: 'text' }, - { id: 'qwen3-omni-flash', codeName: '0199c9dc-e157-7458-bd49-5942363be215', imagePolicy: 'optional', type: 'text' } + { id: 'qwen3-omni-flash', codeName: '0199c9dc-e157-7458-bd49-5942363be215', imagePolicy: 'optional', type: 'text' }, + { id: 'gemini-3-pro-grounding', codeName: '019abdb7-6957-71c1-96a2-bfa79e8a094f', imagePolicy: 'optional', type: 'text', search: true } ], // 模型 ID 解析