Northern_Lights/cherry-studio
1
0
Fork 0
mirror of https://github.com/CherryHQ/cherry-studio.git synced 2025-12-31 00:10:22 +08:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

fix: aihubmix provider generate image logic (#8478)

Browse Source 
fix: aihubmix provider generate image

Co-authored-by: zhaochenxue <zhaochenxue@bixin.cn>
This commit is contained in:
chenxue 2025-07-25 14:14:32 +08:00 committed by GitHub
parent eef20e399c
commit 1a4d64595c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/renderer/src/aiCore/index.ts
View File

@ -170,6 +170,10 @@ export default class AiProvider {
}
public async generateImage(params: GenerateImageParams): Promise<string[]> {
if (this.apiClient instanceof AihubmixAPIClient) {
const client = this.apiClient.getClientForModel({ id: params.model } as Model)
return client.generateImage(params)
}
return this.apiClient.generateImage(params)
}

2
src/renderer/src/pages/paintings/AihubmixPage.tsx
View File

@ -187,7 +187,7 @@ const AihubmixPage: FC<{ Options: string[] }> = ({ Options }) => {
prompt,
model: painting.model,
imageSize: painting.aspectRatio?.replace('ASPECT_', '').replace('_', ':') || '1:1',
batchSize: painting.model.startsWith('imagen-4.0-ultra-generate-exp') ? 1 : painting.numberOfImages || 1,
batchSize: painting.model.startsWith('imagen-4.0-ultra-generate') ? 1 : painting.numberOfImages || 1,
personGeneration: painting.personGeneration
})
if (base64s?.length > 0) {

7
src/renderer/src/pages/paintings/config/aihubmixConfig.tsx
View File

@ -72,9 +72,8 @@ export const createModeConfigs = (): Record<AihubmixMode, ConfigItem[]> => {
label: 'Gemini',
title: 'Gemini',
options: [
{ label: 'imagen-4.0-preview', value: 'imagen-4.0-generate-preview-05-20' },
{ label: 'imagen-4.0-ultra-exp', value: 'imagen-4.0-ultra-generate-exp-05-20' },
{ label: 'imagen-3.0', value: 'imagen-3.0-generate-001' }
{ label: 'imagen-4.0-preview', value: 'imagen-4.0-generate-preview-06-06' },
{ label: 'imagen-4.0-ultra', value: 'imagen-4.0-ultra-generate-preview-06-06' }
]
},
{
@ -206,7 +205,7 @@ export const createModeConfigs = (): Record<AihubmixMode, ConfigItem[]> => {
max: 4,
initialValue: 4,
condition: (painting) =>
Boolean(painting.model?.startsWith('imagen-') && painting.model !== 'imagen-4.0-ultra-generate-exp-05-20')
Boolean(painting.model?.startsWith('imagen-') && painting.model !== 'imagen-4.0-ultra-generate-preview-06-06')
},
{
type: 'select',