Skip to content

Commit 3aa9762

Browse files
authored
Update image generation model selection (#8698)
1 parent 49349e5 commit 3aa9762

File tree

4 files changed

+6
-4
lines changed

4 files changed

+6
-4
lines changed

src/core/tools/__tests__/generateImageTool.test.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,7 @@ describe("generateImageTool", () => {
4747
[EXPERIMENT_IDS.IMAGE_GENERATION]: true,
4848
},
4949
openRouterImageApiKey: "test-api-key",
50-
openRouterImageGenerationSelectedModel: "google/gemini-2.5-flash-image-preview",
50+
openRouterImageGenerationSelectedModel: "google/gemini-2.5-flash-image",
5151
}),
5252
}),
5353
},

src/core/tools/generateImageTool.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@ import { EXPERIMENT_IDS, experiments } from "../../shared/experiments"
1111
import { OpenRouterHandler } from "../../api/providers/openrouter"
1212

1313
// Hardcoded list of image generation models for now
14-
const IMAGE_GENERATION_MODELS = ["google/gemini-2.5-flash-image-preview"]
14+
const IMAGE_GENERATION_MODELS = ["google/gemini-2.5-flash-image", "openai/gpt-5-image", "openai/gpt-5-image-mini"]
1515

1616
export async function generateImageTool(
1717
cline: Task,

webview-ui/src/components/settings/ImageGenerationSettings.tsx

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,9 @@ interface ImageGenerationSettingsProps {
1313

1414
// Hardcoded list of image generation models
1515
const IMAGE_GENERATION_MODELS = [
16-
{ value: "google/gemini-2.5-flash-image-preview", label: "Gemini 2.5 Flash Image Preview" },
16+
{ value: "google/gemini-2.5-flash-image", label: "Gemini 2.5 Flash Image" },
17+
{ value: "openai/gpt-5-image", label: "GPT-5 Image" },
18+
{ value: "openai/gpt-5-image-mini", label: "GPT-5 Image Mini" },
1719
// Add more models as they become available
1820
]
1921

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,7 @@ describe("ImageGenerationSettings", () => {
4141
<ImageGenerationSettings
4242
{...defaultProps}
4343
openRouterImageApiKey="existing-key"
44-
openRouterImageGenerationSelectedModel="google/gemini-2.5-flash-image-preview"
44+
openRouterImageGenerationSelectedModel="google/gemini-2.5-flash-image"
4545
/>,
4646
)
4747

0 commit comments

Comments
 (0)