scriptoriumadmin/Roo-Code
1
0
Fork 0
mirror of https://github.com/RooVetGit/Roo-Code.git synced 2026-08-28 05:27:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: add Google Gemini 3 Pro Image Preview to image generation models (#9440)

Browse source 
Co-authored-by: Roo Code <roomote@roocode.com>
Co-authored-by: Matt Rubens <mrubens@users.noreply.github.com>
This commit is contained in:
roomote[bot] 2025-11-20 14:19:37 -05:00 • committed by GitHub
parent f7d6daedff
commit 1589cc1849
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
4 changed files with 24 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
packages/types/src/image-generation.ts Normal file
View file

@ -0,0 +1,20 @@
/**
* Image generation model constants
*/
export interface ImageGenerationModel {
value: string
label: string
}
export const IMAGE_GENERATION_MODELS: ImageGenerationModel[] = [
{ value: "google/gemini-2.5-flash-image", label: "Gemini 2.5 Flash Image" },
{ value: "google/gemini-3-pro-image-preview", label: "Gemini 3 Pro Image Preview" },
{ value: "openai/gpt-5-image", label: "GPT-5 Image" },
{ value: "openai/gpt-5-image-mini", label: "GPT-5 Image Mini" },
]
/**
* Get array of model values only (for backend validation)
*/
export const IMAGE_GENERATION_MODEL_IDS = IMAGE_GENERATION_MODELS.map((m) => m.value)

1
packages/types/src/index.ts
View file

@ -7,6 +7,7 @@ export * from "./experiment.js"
export * from "./followup.js"
export * from "./global-settings.js"
export * from "./history.js"
export * from "./image-generation.js"
export * from "./ipc.js"
export * from "./marketplace.js"
export * from "./mcp.js"

6
src/core/tools/GenerateImageTool.ts
View file

@ -1,7 +1,7 @@
import path from "path"
import fs from "fs/promises"
import * as vscode from "vscode"
import type { GenerateImageParams } from "@roo-code/types"
import { GenerateImageParams, IMAGE_GENERATION_MODEL_IDS } from "@roo-code/types"
import { Task } from "../task/Task"
import { formatResponse } from "../prompts/responses"
import { fileExistsAtPath } from "../../utils/fs"
@ -12,8 +12,6 @@ import { OpenRouterHandler } from "../../api/providers/openrouter"
import { BaseTool, ToolCallbacks } from "./BaseTool"
import type { ToolUse } from "../../shared/tools"
const IMAGE_GENERATION_MODELS = ["google/gemini-2.5-flash-image", "openai/gpt-5-image", "openai/gpt-5-image-mini"]
export class GenerateImageTool extends BaseTool<"generate_image"> {
readonly name = "generate_image" as const
@ -137,7 +135,7 @@ export class GenerateImageTool extends BaseTool<"generate_image"> {
return
}
const selectedModel = state?.openRouterImageGenerationSelectedModel || IMAGE_GENERATION_MODELS[0]
const selectedModel = state?.openRouterImageGenerationSelectedModel || IMAGE_GENERATION_MODEL_IDS[0]
const fullPath = path.resolve(task.cwd, removeClosingTag("path", relPath))
const isOutsideWorkspace = isPathOutsideWorkspace(fullPath)

9
webview-ui/src/components/settings/ImageGenerationSettings.tsx
View file

@ -1,5 +1,6 @@
import React, { useState, useEffect } from "react"
import { VSCodeCheckbox, VSCodeTextField, VSCodeDropdown, VSCodeOption } from "@vscode/webview-ui-toolkit/react"
import { IMAGE_GENERATION_MODELS } from "@roo-code/types"
import { useAppTranslation } from "@/i18n/TranslationContext"
interface ImageGenerationSettingsProps {
@ -11,14 +12,6 @@ interface ImageGenerationSettingsProps {
setImageGenerationSelectedModel: (model: string) => void
}
// Hardcoded list of image generation models
const IMAGE_GENERATION_MODELS = [
{ 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
]
export const ImageGenerationSettings = ({
enabled,
onChange,