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

chore: 调整豆包操作逻辑 (ref #30)

Browse Source
This commit is contained in:
foxhui
2026-03-31 14:59:35 +08:00
Unverified
parent f629fe0c67
commit ae462ad520
1 changed files with 26 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/backend/adapter/doubao_text.js
+26 -25
View File
@@ -49,7 +49,32 @@ async function generate(context, prompt, imgPaths, modelId, meta = {}) {
const inputLocator = page.locator('textarea[data-testid="chat_input_input"]');
await waitForInput(page, inputLocator, { click: false });
// 2. 上传图片 (如果有)
// 2. 选择模型
const modelMenuName = MODEL_MENU_MAP[modelId] || MODEL_MENU_MAP['seed'];
logger.debug('适配器', `选择模型: ${modelId} -> ${String(modelMenuName)}`, meta);
await sleep(300, 500);
// 给予 1 秒的缓冲时间等待 React 渲染按钮
const modelSelectorBtn = page.locator('button[aria-haspopup="menu"]:visible:has([data-testid="deep-thinking-action-button"], [data-testid="mode-select-action-button"])').first();
let selectorExists = false;
try {
await modelSelectorBtn.waitFor({ state: 'attached', timeout: 1000 });
selectorExists = true;
} catch (e) {
selectorExists = false;
}
if (selectorExists) {
await safeClick(page, modelSelectorBtn, { bias: 'button' });
await sleep(300, 500);
const menuItem = page.getByRole('menuitem', { name: modelMenuName });
await menuItem.waitFor({ state: 'visible', timeout: 5000 });
await safeClick(page, menuItem, { bias: 'button' });
await sleep(200, 400);
}
// 3. 上传图片 (如果有)
if (imgPaths && imgPaths.length > 0) {
logger.info('适配器', `开始上传 ${imgPaths.length} 张图片...`, meta);
@@ -100,30 +125,6 @@ async function generate(context, prompt, imgPaths, modelId, meta = {}) {
logger.info('适配器', '图片上传完成', meta);
}
// 3. 选择模型
const modelMenuName = MODEL_MENU_MAP[modelId] || MODEL_MENU_MAP['seed'];
logger.debug('适配器', `选择模型: ${modelId} -> ${String(modelMenuName)}`, meta);
// 给予 3 秒的缓冲时间等待 React 渲染按钮
const modelSelectorBtn = page.locator('button[aria-haspopup="menu"]:visible:has([data-testid="deep-thinking-action-button"], [data-testid="mode-select-action-button"])').first();
let selectorExists = false;
try {
await modelSelectorBtn.waitFor({ state: 'attached', timeout: 3000 });
selectorExists = true;
} catch (e) {
selectorExists = false;
}
if (selectorExists) {
await safeClick(page, modelSelectorBtn, { bias: 'button' });
await sleep(300, 500);
const menuItem = page.getByRole('menuitem', { name: modelMenuName });
await menuItem.waitFor({ state: 'visible', timeout: 5000 });
await safeClick(page, menuItem, { bias: 'button' });
await sleep(200, 400);
}
// 4. 填写提示词
await safeClick(page, inputLocator, { bias: 'input' });
await humanType(page, inputLocator, prompt);