Northern_Lights/cherry-studio
1
0
Fork 0
mirror of https://github.com/CherryHQ/cherry-studio.git synced 2026-01-05 04:19:02 +08:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

fix(ImageGenerationMiddleware): correctly process image URLs (#7198)

Browse Source
This commit is contained in:
Doekin 2025-06-14 22:39:32 +08:00 committed by GitHub
parent 3c5c63f555
commit bb8e5321e8
2 changed files with 13 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
src/renderer/src/aiCore/middleware/feat/ImageGenerationMiddleware.ts
View File

@ -97,11 +97,21 @@ export const ImageGenerationMiddleware: CompletionsMiddleware =
) )
} }
const b64_json_array = response.data?.map((item) => `data:image/png;base64,${item.b64_json}`) || [] let imageType: 'url' | 'base64' = 'base64'
const imageList =
response.data?.reduce((acc: string[], image) => {
if (image.url) {
acc.push(image.url)
imageType = 'url'
} else if (image.b64_json) {
acc.push(`data:image/png;base64,${image.b64_json}`)
}
return acc
}, []) || []
enqueue({ enqueue({
type: ChunkType.IMAGE_COMPLETE, type: ChunkType.IMAGE_COMPLETE,
image: { type: 'base64', images: b64_json_array } image: { type: imageType, images: imageList }
}) })
const usage = (response as any).usage || { prompt_tokens: 0, completion_tokens: 0, total_tokens: 0 } const usage = (response as any).usage || { prompt_tokens: 0, completion_tokens: 0, total_tokens: 0 }

2
src/renderer/src/types/chunk.ts
View File

@ -137,7 +137,7 @@ export interface ImageCompleteChunk {
/** /**
* The image content of the chunk * The image content of the chunk
*/ */
image?: { type: 'base64'; images: string[] } image?: { type: 'url' | 'base64'; images: string[] }
} }
export interface ThinkingDeltaChunk { export interface ThinkingDeltaChunk {