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 #17)

Browse Source
This commit is contained in:
foxhui
2026-02-16 14:20:05 +08:00
Unverified
parent 6d5c843e10
commit aa6ab799c4
3 changed files with 69 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files
CHANGELOG.md
+1
View File
@@ -10,6 +10,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
### 🐛 Fixed
- **适配器**
- 修复 DeepSeek 文本生成适配器因接口格式更新导致无法获取生成结果
- 修复 豆包 适配器图片上传逻辑更改导致上传进度验证失败的问题
## [3.5.2] - 2026-02-10
src/backend/adapter/doubao.js
+32 -10
View File
@@ -60,17 +60,39 @@ async function generate(context, prompt, imgPaths, modelId, meta = {}) {
if (imgPaths && imgPaths.length > 0) {
logger.info('适配器', `开始上传 ${imgPaths.length} 张图片...`, meta);
const uploadBtn = page.locator('button[data-testid="image-creation-chat-input-picture-reference-button"]');
await uploadBtn.waitFor({ state: 'visible', timeout: 10000 });
await uploadFilesViaChooser(page, uploadBtn, imgPaths, {
uploadValidator: (response) => {
// 预先拦截 ApplyImageUpload 响应，动态收集实际上传路径
const expectedUploadPaths = new Set();
const applyUploadHandler = async (response) => {
try {
const url = response.url();
return response.status() === 200 &&
url.includes('bytedanceapi.com') &&
url.includes('Action=CommitImageUpload');
}
}, meta);
if (!url.includes('Action=ApplyImageUpload') || response.status() !== 200) return;
const json = await response.json();
const storeUri = json.Result?.UploadAddress?.StoreInfos?.[0]?.StoreUri;
if (storeUri) {
expectedUploadPaths.add(storeUri);
logger.debug('适配器', `已获取上传路径: ${storeUri}`, meta);
}
} catch { /* 忽略解析错误 */ }
};
page.on('response', applyUploadHandler);
try {
const uploadBtn = page.locator('button[data-testid="image-creation-chat-input-picture-reference-button"]');
await uploadBtn.waitFor({ state: 'visible', timeout: 10000 });
await uploadFilesViaChooser(page, uploadBtn, imgPaths, {
uploadValidator: (response) => {
if (response.status() !== 200 || response.request().method() !== 'POST') return false;
const url = response.url();
for (const path of expectedUploadPaths) {
if (url.includes(path)) return true;
}
return false;
}
}, meta);
} finally {
page.off('response', applyUploadHandler);
}
logger.info('适配器', '图片上传完成', meta);
}
src/backend/adapter/doubao_text.js
+36 -14
View File
@@ -45,21 +45,43 @@ async function generate(context, prompt, imgPaths, modelId, meta = {}) {
if (imgPaths && imgPaths.length > 0) {
logger.info('适配器', `开始上传 ${imgPaths.length} 张图片...`, meta);
// 点击上传菜单按钮
const uploadMenuBtn = page.locator('main button[aria-haspopup="menu"]').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"]');
await uploadFilesViaChooser(page, uploadItem, imgPaths, {
uploadValidator: (response) => {
// 预先拦截 ApplyImageUpload 响应，动态收集实际上传路径
const expectedUploadPaths = new Set();
const applyUploadHandler = async (response) => {
try {
const url = response.url();
return response.status() === 200 &&
url.includes('bytedanceapi.com') &&
url.includes('Action=CommitImageUpload');
}
}, meta);
if (!url.includes('Action=ApplyImageUpload') || response.status() !== 200) return;
const json = await response.json();
const storeUri = json.Result?.UploadAddress?.StoreInfos?.[0]?.StoreUri;
if (storeUri) {
expectedUploadPaths.add(storeUri);
logger.debug('适配器', `已获取上传路径: ${storeUri}`, meta);
}
} catch { /* 忽略解析错误 */ }
};
page.on('response', applyUploadHandler);
try {
// 点击上传菜单按钮
const uploadMenuBtn = page.locator('main button[aria-haspopup="menu"]').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"]');
await uploadFilesViaChooser(page, uploadItem, imgPaths, {
uploadValidator: (response) => {
if (response.status() !== 200 || response.request().method() !== 'POST') return false;
const url = response.url();
for (const path of expectedUploadPaths) {
if (url.includes(path)) return true;
}
return false;
}
}, meta);
} finally {
page.off('response', applyUploadHandler);
}
logger.info('适配器', '图片上传完成', meta);
}