From f2842657150bcd02470407d7c867fa6b2e438602 Mon Sep 17 00:00:00 2001
From: "\"Rami Chowdhury\"" <"rami@skyfi.com">
Date: Wed, 18 Jun 2025 13:12:27 -0400
Subject: [PATCH] feat(providers): update Google Gemini models to latest
 available models

- Update default model from gemini-2.0-flash to gemini-2.5-flash
- Add comprehensive list of 24 current Google Gemini models including:
  - Gemini 2.5 models (pro, flash, flash-lite, native audio, TTS)
  - Gemini 2.0 models (flash variants, image generation)
  - Gemini 1.5 models (flash, flash-8b, pro variants)
- Focus on core text generation models, excluding:
  - Embedding models (gemini-embedding-exp, text-embedding-004, embedding-001)
  - Specialized models (imagen-3.0, veo-2.0, aqa)
  - Versioned models ending in "001" (prefer latest/stable versions)
- Update documentation URL to point to official Gemini API docs
- Organize models by generation with clear comments for better maintainability

Reference: https://ai.google.dev/gemini-api/docs/models
---
 crates/goose/src/providers/google.rs | 33 ++++++++++++++++++++++------
 1 file changed, 26 insertions(+), 7 deletions(-)

diff --git a/crates/goose/src/providers/google.rs b/crates/goose/src/providers/google.rs
index 8182d3b06e95..a8e8b5647902 100644
--- a/crates/goose/src/providers/google.rs
+++ b/crates/goose/src/providers/google.rs
@@ -15,17 +15,36 @@ use std::time::Duration;
 use url::Url;
 
 pub const GOOGLE_API_HOST: &str = "https://generativelanguage.googleapis.com";
-pub const GOOGLE_DEFAULT_MODEL: &str = "gemini-2.0-flash";
+pub const GOOGLE_DEFAULT_MODEL: &str = "gemini-2.5-flash";
 pub const GOOGLE_KNOWN_MODELS: &[&str] = &[
+    // Gemini 2.5 models (latest generation)
+    "gemini-2.5-pro",
+    "gemini-2.5-pro-preview-06-05",
+    "gemini-2.5-pro-preview-05-06",
+    "gemini-2.5-flash",
+    "gemini-2.5-flash-preview-05-20",
+    "gemini-2.5-flash-lite-preview-06-17",
+    "gemini-2.5-flash-preview-native-audio-dialog",
+    "gemini-2.5-flash-exp-native-audio-thinking-dialog",
+    "gemini-2.5-flash-preview-tts",
+    "gemini-2.5-pro-preview-tts",
+    // Gemini 2.0 models
     "gemini-2.0-flash",
-    "gemini-2.0-flash-lite-preview-02-05",
-    "gemini-2.0-flash-thinking-exp-01-21",
-    "gemini-2.0-pro-exp-02-05",
-    "gemini-2.5-pro-exp-03-25",
-    "gemini-2.5-flash-preview-04-17",
+    "gemini-2.0-flash-exp",
+    "gemini-2.0-flash-preview-image-generation",
+    "gemini-2.0-flash-lite",
+    // Gemini 1.5 models
+    "gemini-1.5-flash",
+    "gemini-1.5-flash-latest",
+    "gemini-1.5-flash-002",
+    "gemini-1.5-flash-8b",
+    "gemini-1.5-flash-8b-latest",
+    "gemini-1.5-pro",
+    "gemini-1.5-pro-latest",
+    "gemini-1.5-pro-002",
 ];
 
-pub const GOOGLE_DOC_URL: &str = "https://ai.google/get-started/our-models/";
+pub const GOOGLE_DOC_URL: &str = "https://ai.google.dev/gemini-api/docs/models";
 
 #[derive(Debug, serde::Serialize)]
 pub struct GoogleProvider {