Northern_Lights/cherry-studio
1
0
Fork 0
mirror of https://github.com/CherryHQ/cherry-studio.git synced 2026-01-06 21:35:52 +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[]> { 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) 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, prompt,
model: painting.model, model: painting.model,
imageSize: painting.aspectRatio?.replace('ASPECT_', '').replace('_', ':') || '1:1', 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 personGeneration: painting.personGeneration
}) })
if (base64s?.length > 0) { 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', label: 'Gemini',
title: 'Gemini', title: 'Gemini',
options: [ options: [
{ label: 'imagen-4.0-preview', value: 'imagen-4.0-generate-preview-05-20' }, { label: 'imagen-4.0-preview', value: 'imagen-4.0-generate-preview-06-06' },
{ label: 'imagen-4.0-ultra-exp', value: 'imagen-4.0-ultra-generate-exp-05-20' }, { label: 'imagen-4.0-ultra', value: 'imagen-4.0-ultra-generate-preview-06-06' }
{ label: 'imagen-3.0', value: 'imagen-3.0-generate-001' }
] ]
}, },
{ {
@ -206,7 +205,7 @@ export const createModeConfigs = (): Record<AihubmixMode, ConfigItem[]> => {
max: 4, max: 4,
initialValue: 4, initialValue: 4,
condition: (painting) => 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', type: 'select',