diff --git a/internal/registry/model_definitions.go b/internal/registry/model_definitions.go index 72ac3324..6bcf9a3e 100644 --- a/internal/registry/model_definitions.go +++ b/internal/registry/model_definitions.go @@ -166,6 +166,20 @@ func GetGeminiCLIModels() []*ModelInfo { OutputTokenLimit: 8192, SupportedGenerationMethods: []string{"generateContent", "countTokens", "createCachedContent", "batchGenerateContent"}, }, + { + ID: "gemini-2.5-flash-image", + Object: "model", + Created: time.Now().Unix(), + OwnedBy: "google", + Type: "gemini", + Name: "models/gemini-2.5-flash-image", + Version: "2.5", + DisplayName: "Gemini 2.5 Flash Image", + Description: "State-of-the-art image generation and editing model.", + InputTokenLimit: 1048576, + OutputTokenLimit: 8192, + SupportedGenerationMethods: []string{"generateContent", "countTokens", "createCachedContent", "batchGenerateContent"}, + }, } } diff --git a/internal/runtime/executor/gemini_cli_executor.go b/internal/runtime/executor/gemini_cli_executor.go index 281b2970..eb3f227e 100644 --- a/internal/runtime/executor/gemini_cli_executor.go +++ b/internal/runtime/executor/gemini_cli_executor.go @@ -518,7 +518,7 @@ func geminiModelDisallowsThinking(model string) bool { return false } lower := strings.ToLower(model) - for _, marker := range []string{"gemini-2.5-flash-image-preview"} { + for _, marker := range []string{"gemini-2.5-flash-image-preview", "gemini-2.5-flash-image"} { if strings.Contains(lower, marker) { return true }