whai/cherry-studio
1
1
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

refactor: improved type safety and consistency for file handling

Browse Source
This commit is contained in:
kangfenmao 2025-01-14 21:02:55 +08:00
parent 247d1a1846
commit 66bc9cb3f9
1 changed files with 17 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
src/renderer/src/providers/OpenAIProvider.ts
View File

@ -123,7 +123,7 @@ export default class OpenAIProvider extends BaseProvider {
//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 || []) {
if (file.type === FileTypes.IMAGE && isVision) {
@ -147,22 +147,22 @@ export default class OpenAIProvider extends BaseProvider {
role: message.role,
content: [...parts, ...qwenlm_image_url]
} as ChatCompletionMessageParam
} else {
for (const file of message.files || []) {
if (file.type === FileTypes.IMAGE && isVision) {
const image = await window.api.file.base64Image(file.id + file.ext)
parts.push({
type: 'image_url',
image_url: { url: image.data }
})
}
if ([FileTypes.TEXT, FileTypes.DOCUMENT].includes(file.type)) {
const fileContent = await (await window.api.file.read(file.id + file.ext)).trim()
parts.push({
type: 'text',
text: file.origin_name + '\n' + fileContent
})
}
}
for (const file of message.files || []) {
if (file.type === FileTypes.IMAGE && isVision) {
const image = await window.api.file.base64Image(file.id + file.ext)
parts.push({
type: 'image_url',
image_url: { url: image.data }
})
}
if ([FileTypes.TEXT, FileTypes.DOCUMENT].includes(file.type)) {
const fileContent = await (await window.api.file.read(file.id + file.ext)).trim()
parts.push({
type: 'text',
text: file.origin_name + '\n' + fileContent
})
}
}