From 1a4d64595c1c6c3abf760e6ac3c368ca4bf7948b Mon Sep 17 00:00:00 2001
From: chenxue <DDU1222@users.noreply.github.com>
Date: Fri, 25 Jul 2025 14:14:32 +0800
Subject: [PATCH] fix: aihubmix provider generate image logic (#8478)

fix: aihubmix provider generate image

Co-authored-by: zhaochenxue <zhaochenxue@bixin.cn>
---
 src/renderer/src/aiCore/index.ts                           | 4 ++++
 src/renderer/src/pages/paintings/AihubmixPage.tsx          | 2 +-
 src/renderer/src/pages/paintings/config/aihubmixConfig.tsx | 7 +++----
 3 files changed, 8 insertions(+), 5 deletions(-)

diff --git a/src/renderer/src/aiCore/index.ts b/src/renderer/src/aiCore/index.ts
index 0fadfb7aa4..83646d502a 100644
--- a/src/renderer/src/aiCore/index.ts
+++ b/src/renderer/src/aiCore/index.ts
@@ -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)
   }
 
diff --git a/src/renderer/src/pages/paintings/AihubmixPage.tsx b/src/renderer/src/pages/paintings/AihubmixPage.tsx
index d2ef782c28..2b6d8e27e1 100644
--- a/src/renderer/src/pages/paintings/AihubmixPage.tsx
+++ b/src/renderer/src/pages/paintings/AihubmixPage.tsx
@@ -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) {
diff --git a/src/renderer/src/pages/paintings/config/aihubmixConfig.tsx b/src/renderer/src/pages/paintings/config/aihubmixConfig.tsx
index 6aaded1c2b..e188303a32 100644
--- a/src/renderer/src/pages/paintings/config/aihubmixConfig.tsx
+++ b/src/renderer/src/pages/paintings/config/aihubmixConfig.tsx
@@ -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',