Jelajahi Sumber

Update image generation model selection (#8698)

Christiaan Arnoldus 2 bulan lalu
induk
melakukan
3aa9762d04

+ 1 - 1
src/core/tools/__tests__/generateImageTool.test.ts

@@ -47,7 +47,7 @@ describe("generateImageTool", () => {
 							[EXPERIMENT_IDS.IMAGE_GENERATION]: true,
 						},
 						openRouterImageApiKey: "test-api-key",
-						openRouterImageGenerationSelectedModel: "google/gemini-2.5-flash-image-preview",
+						openRouterImageGenerationSelectedModel: "google/gemini-2.5-flash-image",
 					}),
 				}),
 			},

+ 1 - 1
src/core/tools/generateImageTool.ts

@@ -11,7 +11,7 @@ import { EXPERIMENT_IDS, experiments } from "../../shared/experiments"
 import { OpenRouterHandler } from "../../api/providers/openrouter"
 
 // Hardcoded list of image generation models for now
-const IMAGE_GENERATION_MODELS = ["google/gemini-2.5-flash-image-preview"]
+const IMAGE_GENERATION_MODELS = ["google/gemini-2.5-flash-image", "openai/gpt-5-image", "openai/gpt-5-image-mini"]
 
 export async function generateImageTool(
 	cline: Task,

+ 3 - 1
webview-ui/src/components/settings/ImageGenerationSettings.tsx

@@ -13,7 +13,9 @@ interface ImageGenerationSettingsProps {
 
 // Hardcoded list of image generation models
 const IMAGE_GENERATION_MODELS = [
-	{ value: "google/gemini-2.5-flash-image-preview", label: "Gemini 2.5 Flash Image Preview" },
+	{ value: "google/gemini-2.5-flash-image", label: "Gemini 2.5 Flash Image" },
+	{ value: "openai/gpt-5-image", label: "GPT-5 Image" },
+	{ value: "openai/gpt-5-image-mini", label: "GPT-5 Image Mini" },
 	// Add more models as they become available
 ]
 

+ 1 - 1
webview-ui/src/components/settings/__tests__/ImageGenerationSettings.spec.tsx

@@ -41,7 +41,7 @@ describe("ImageGenerationSettings", () => {
 				<ImageGenerationSettings
 					{...defaultProps}
 					openRouterImageApiKey="existing-key"
-					openRouterImageGenerationSelectedModel="google/gemini-2.5-flash-image-preview"
+					openRouterImageGenerationSelectedModel="google/gemini-2.5-flash-image"
 				/>,
 			)