refactor: improved type safety and consistency for file handling

This commit is contained in:
kangfenmao 2025-01-14 21:02:55 +08:00
parent 247d1a1846
commit 66bc9cb3f9

View File

@ -123,7 +123,7 @@ export default class OpenAIProvider extends BaseProvider {
//QwenLM上传图片 //QwenLM上传图片
if (this.provider.id === 'qwenlm') { if (this.provider.id === 'qwenlm') {
let qwenlm_image_url: { type: string; image: string }[] = [] const qwenlm_image_url: { type: string; image: string }[] = []
for (const file of message.files || []) { for (const file of message.files || []) {
if (file.type === FileTypes.IMAGE && isVision) { if (file.type === FileTypes.IMAGE && isVision) {
@ -147,7 +147,8 @@ export default class OpenAIProvider extends BaseProvider {
role: message.role, role: message.role,
content: [...parts, ...qwenlm_image_url] content: [...parts, ...qwenlm_image_url]
} as ChatCompletionMessageParam } as ChatCompletionMessageParam
} else { }
for (const file of message.files || []) { for (const file of message.files || []) {
if (file.type === FileTypes.IMAGE && isVision) { if (file.type === FileTypes.IMAGE && isVision) {
const image = await window.api.file.base64Image(file.id + file.ext) const image = await window.api.file.base64Image(file.id + file.ext)
@ -164,7 +165,6 @@ export default class OpenAIProvider extends BaseProvider {
}) })
} }
} }
}
return { return {
role: message.role, role: message.role,