diff --git a/crates/google_ai/src/google_ai.rs b/crates/google_ai/src/google_ai.rs index e26750936d..9f6045a288 100644 --- a/crates/google_ai/src/google_ai.rs +++ b/crates/google_ai/src/google_ai.rs @@ -402,6 +402,8 @@ pub enum Model { Gemini25ProExp0325, #[serde(rename = "gemini-2.5-pro-preview-03-25")] Gemini25ProPreview0325, + #[serde(rename = "gemini-2.5-flash-preview-04-17")] + Gemini25FlashPreview0417, #[serde(rename = "custom")] Custom { name: String, @@ -426,6 +428,7 @@ impl Model { Model::Gemini20FlashLite => "gemini-2.0-flash-lite-preview", Model::Gemini25ProExp0325 => "gemini-2.5-pro-exp-03-25", Model::Gemini25ProPreview0325 => "gemini-2.5-pro-preview-03-25", + Model::Gemini25FlashPreview0417 => "gemini-2.5-flash-preview-04-17", Model::Custom { name, .. } => name, } } @@ -440,6 +443,7 @@ impl Model { Model::Gemini20FlashLite => "Gemini 2.0 Flash Lite", Model::Gemini25ProExp0325 => "Gemini 2.5 Pro Exp", Model::Gemini25ProPreview0325 => "Gemini 2.5 Pro Preview", + Model::Gemini25FlashPreview0417 => "Gemini 2.5 Flash Preview", Self::Custom { name, display_name, .. } => display_name.as_ref().unwrap_or(name), @@ -456,6 +460,7 @@ impl Model { Model::Gemini20FlashLite => 1_000_000, Model::Gemini25ProExp0325 => 1_000_000, Model::Gemini25ProPreview0325 => 1_000_000, + Model::Gemini25FlashPreview0417 => 1_000_000, Model::Custom { max_tokens, .. } => *max_tokens, } } diff --git a/crates/language_model/src/model/cloud_model.rs b/crates/language_model/src/model/cloud_model.rs index a6e9389816..0cea0d6966 100644 --- a/crates/language_model/src/model/cloud_model.rs +++ b/crates/language_model/src/model/cloud_model.rs @@ -105,6 +105,7 @@ impl CloudModel { | google_ai::Model::Gemini20FlashLite | google_ai::Model::Gemini25ProExp0325 | google_ai::Model::Gemini25ProPreview0325 + | google_ai::Model::Gemini25FlashPreview0417 | google_ai::Model::Custom { .. } => { LanguageModelAvailability::RequiresPlan(Plan::ZedPro) }