diff --git a/src/core/tools/generateImageTool.ts b/src/core/tools/generateImageTool.ts index 97c29e0a6216..f3ffeb55caf1 100644 --- a/src/core/tools/generateImageTool.ts +++ b/src/core/tools/generateImageTool.ts @@ -12,10 +12,7 @@ import { safeWriteJson } from "../../utils/safeWriteJson" 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", - // Add more models as they become available -] +const IMAGE_GENERATION_MODELS = ["google/gemini-2.5-flash-image-preview", "google/gemini-2.5-flash-image-preview:free"] export async function generateImageTool( cline: Task, diff --git a/webview-ui/src/components/settings/ImageGenerationSettings.tsx b/webview-ui/src/components/settings/ImageGenerationSettings.tsx index f08284f7b5cb..667a31d57933 100644 --- a/webview-ui/src/components/settings/ImageGenerationSettings.tsx +++ b/webview-ui/src/components/settings/ImageGenerationSettings.tsx @@ -17,6 +17,7 @@ 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-preview:free", label: "Gemini 2.5 Flash Image Preview (Free)" }, // Add more models as they become available ]