chuan/WebAI2API
1
0
Fork 0
mirror of https://github.com/foxhui/WebAI2API.git synced 2026-06-16 21:03:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: 修复豆包前端更新导致的失效 (ref #58)

Browse Source
This commit is contained in:
foxhui
2026-04-13 02:08:34 +08:00
Unverified
parent 6353806da6
commit 42caf7e9a1
2 changed files with 14 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/backend/adapter/doubao.js
+5 -5
View File
@@ -41,13 +41,13 @@ async function generate(context, prompt, imgPaths, modelId, meta = {}) {
// 1. 点击进入图片生成模式
logger.debug('适配器', '进入图片生成模式...', meta);
const skillBtn = page.locator('button[data-testid="skill_bar_button_3"]');
const skillBtn = page.locator('button[data-skill-id="skill_bar_button_3"]');
await skillBtn.waitFor({ state: 'visible', timeout: 30000 });
await safeClick(page, skillBtn, { bias: 'button' });
// 2. 选择模型
logger.debug('适配器', `选择模型: ${codeName}...`, meta);
const modelBtn = page.locator('button[data-testid="image-creation-chat-input-picture-model-button"]');
const modelBtn = page.locator('button').filter({ hasText: /Seedream/i }).first();
await modelBtn.waitFor({ state: 'visible', timeout: 10000 });
await safeClick(page, modelBtn, { bias: 'button' });
await sleep(300, 500);
@@ -77,7 +77,7 @@ async function generate(context, prompt, imgPaths, modelId, meta = {}) {
page.on('response', applyUploadHandler);
try {
const uploadBtn = page.locator('button[data-testid="image-creation-chat-input-picture-reference-button"]');
const uploadBtn = page.locator('button').filter({ hasText: /Reference Image|参考图|參考圖/i }).first();
await uploadBtn.waitFor({ state: 'visible', timeout: 10000 });
await uploadFilesViaChooser(page, uploadBtn, imgPaths, {
@@ -98,7 +98,7 @@ async function generate(context, prompt, imgPaths, modelId, meta = {}) {
}
// 4. 填写提示词
const inputLocator = page.locator('div[data-testid="chat_input_input"][role="textbox"]');
const inputLocator = page.locator('#input-engine-container div[role="textbox"]');
await waitForInput(page, inputLocator, { click: true });
await humanType(page, inputLocator, prompt);
@@ -146,7 +146,7 @@ async function generate(context, prompt, imgPaths, modelId, meta = {}) {
});
// 6. 点击发送
const sendBtn = page.locator('button[data-testid="chat_input_send_button"]');
const sendBtn = page.locator('button#flow-end-msg-send');
await sendBtn.waitFor({ state: 'visible', timeout: 10000 });
logger.info('适配器', '点击发送...', meta);
await safeClick(page, sendBtn, { bias: 'button' });
src/backend/adapter/doubao_text.js
+9 -8
View File
@@ -46,7 +46,7 @@ async function generate(context, prompt, imgPaths, modelId, meta = {}) {
await gotoWithCheck(page, TARGET_URL);
// 1. 等待输入框加载
const inputLocator = page.locator('textarea[data-testid="chat_input_input"]');
const inputLocator = page.locator('textarea.semi-input-textarea');
await waitForInput(page, inputLocator, { click: false });
// 2. 选择模型
@@ -55,8 +55,9 @@ async function generate(context, prompt, imgPaths, modelId, meta = {}) {
await sleep(300, 500);
// 给予 1 秒的缓冲时间等待 React 渲染按钮
const modelSelectorBtn = page.locator('button[aria-haspopup="menu"]')
.filter({ has: page.locator('[data-testid="deep-thinking-action-button"], [data-testid="mode-select-action-button"]') })
await sleep(1000); // 确保有一定的渲染时间
const modelSelectorBtn = page.locator('#input-engine-container button[aria-haspopup="menu"]')
.filter({ hasText: /Fast|Think|Pro|快速|思考|专家|專家/ })
.first();
let selectorExists = false;
try {
@@ -81,7 +82,7 @@ async function generate(context, prompt, imgPaths, modelId, meta = {}) {
}
}
await safeClick(page, menuItem, { bias: 'button' });
await sleep(200, 400);
await sleep(600, 1000); // 留出充足时间等待模型选择浮窗自动关闭，防止遮挡上传图标
}
// 3. 上传图片 (如果有)
@@ -105,15 +106,15 @@ async function generate(context, prompt, imgPaths, modelId, meta = {}) {
page.on('response', applyUploadHandler);
try {
// 点击上传菜单按钮
// 点击上传菜单按钮（排除掉含有模型名称或带有“更多”文案的按钮）
const uploadMenuBtn = page.locator('#input-engine-container button[aria-haspopup="menu"]')
.filter({ hasNot: page.locator('[data-testid="deep-thinking-action-button"], [data-testid="mode-select-action-button"]') })
.filter({ hasNot: page.locator('text=/Fast|Think|Pro|快速|思考|专家|專家|更多/') })
.first();
await safeClick(page, uploadMenuBtn, { bias: 'button' });
await sleep(300, 500);
// 点击上传文件选项
const uploadItem = page.locator('div[data-testid="upload_file_panel_upload_item"][role="menuitem"]');
const uploadItem = page.locator('div[role="menuitem"]').filter({ hasText: /上传文件或图片|上傳檔案或圖片|Upload File or Image/ });
await uploadFilesViaChooser(page, uploadItem, imgPaths, {
uploadValidator: (response) => {
if (response.status() !== 200 || response.request().method() !== 'POST') return false;
@@ -187,7 +188,7 @@ async function generate(context, prompt, imgPaths, modelId, meta = {}) {
});
// 6. 点击发送
const sendBtn = page.locator('button[data-testid="chat_input_send_button"]');
const sendBtn = page.locator('button#flow-end-msg-send');
await sendBtn.waitFor({ state: 'visible', timeout: 10000 });
logger.info('适配器', '点击发送...', meta);
await safeClick(page, sendBtn, { bias: 'button' });