diff --git a/internal/client/gemini-cli_client.go b/internal/client/gemini-cli_client.go index 636c6e9a..6ae50214 100644 --- a/internal/client/gemini-cli_client.go +++ b/internal/client/gemini-cli_client.go @@ -38,8 +38,9 @@ const ( var ( previewModels = map[string][]string{ - "gemini-2.5-pro": {"gemini-2.5-pro-preview-05-06", "gemini-2.5-pro-preview-06-05"}, - "gemini-2.5-flash": {"gemini-2.5-flash-preview-04-17", "gemini-2.5-flash-preview-05-20"}, + "gemini-2.5-pro": {"gemini-2.5-pro-preview-05-06", "gemini-2.5-pro-preview-06-05"}, + "gemini-2.5-flash": {"gemini-2.5-flash-preview-04-17", "gemini-2.5-flash-preview-05-20"}, + "gemini-2.5-flash-lite": {"gemini-2.5-flash-lite-preview-06-17"}, } ) @@ -99,6 +100,7 @@ func (c *GeminiCLIClient) CanProvideModel(modelName string) bool { models := []string{ "gemini-2.5-pro", "gemini-2.5-flash", + "gemini-2.5-flash-lite", } return util.InArray(models, modelName) } diff --git a/internal/registry/model_definitions.go b/internal/registry/model_definitions.go index 941be357..83456f30 100644 --- a/internal/registry/model_definitions.go +++ b/internal/registry/model_definitions.go @@ -130,6 +130,20 @@ func GetGeminiCLIModels() []*ModelInfo { OutputTokenLimit: 65536, SupportedGenerationMethods: []string{"generateContent", "countTokens", "createCachedContent", "batchGenerateContent"}, }, + { + ID: "gemini-2.5-flash-lite", + Object: "model", + Created: time.Now().Unix(), + OwnedBy: "google", + Type: "gemini", + Name: "models/gemini-2.5-flash-lite", + Version: "2.5", + DisplayName: "Gemini 2.5 Flash Lite", + Description: "Our smallest and most cost effective model, built for at scale usage.", + InputTokenLimit: 1048576, + OutputTokenLimit: 65536, + SupportedGenerationMethods: []string{"generateContent", "countTokens", "createCachedContent", "batchGenerateContent"}, + }, } }