feat: update gemini models (#1135)

block · Feb 7, 2025 · 76f8034 · 76f8034
1 parent 5f8b522
commit 76f8034
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 12 deletions.
diff --git a/crates/goose/src/providers/google.rs b/crates/goose/src/providers/google.rs
@@ -13,14 +13,16 @@ 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-exp";
+pub const GOOGLE_DEFAULT_MODEL: &str = "gemini-2.0-flash";
 pub const GOOGLE_KNOWN_MODELS: &[&str] = &[
     "models/gemini-1.5-pro-latest",
     "models/gemini-1.5-pro",
     "models/gemini-1.5-flash-latest",
     "models/gemini-1.5-flash",
-    "models/gemini-2.0-flash-exp",
+    "models/gemini-2.0-flash",
+    "models/gemini-2.0-flash-lite-preview-02-05",
     "models/gemini-2.0-flash-thinking-exp-01-21",
+    "models/gemini-2.0-pro-exp-02-05",
 ];
 
 pub const GOOGLE_DOC_URL: &str = "https://ai.google/get-started/our-models/";

diff --git a/ui/desktop/src/components/settings/models/hardcoded_stuff.tsx b/ui/desktop/src/components/settings/models/hardcoded_stuff.tsx
@@ -10,14 +10,15 @@ export const goose_models: Model[] = [
   { id: 8, name: 'claude-3-5-haiku-latest', provider: 'Anthropic' },
   { id: 9, name: 'claude-3-opus-latest', provider: 'Anthropic' },
   { id: 10, name: 'gemini-1.5-pro', provider: 'Google' },
-  { id: 11, name: 'gemini-2.0-flash-exp', provider: 'Google' },
-  { id: 12, name: 'gemini-1.5-flash', provider: 'Google' },
-  { id: 13, name: 'gemini-2.0-flash-thinking-exp-01-21', provider: 'Google' },
-  { id: 14, name: 'gemma-2-2b-it', provider: 'Google' },
-  { id: 15, name: 'llama-3.3-70b-versatile', provider: 'Groq' },
-  { id: 16, name: 'qwen2.5', provider: 'Ollama' },
-  { id: 17, name: 'anthropic/claude-3.5-sonnet', provider: 'OpenRouter' },
-  { id: 18, name: 'gpt-4o', provider: 'Azure OpenAI' },
+  { id: 11, name: 'gemini-1.5-flash', provider: 'Google' },
+  { id: 12, name: 'gemini-2.0-flash', provider: 'Google' },
+  { id: 13, name: 'gemini-2.0-flash-lite-preview-02-05', provider: 'Google' },
+  { id: 14, name: 'gemini-2.0-flash-thinking-exp-01-21', provider: 'Google' },
+  { id: 15, name: 'gemini-2.0-pro-exp-02-05', provider: 'Google' },
+  { id: 16, name: 'llama-3.3-70b-versatile', provider: 'Groq' },
+  { id: 17, name: 'qwen2.5', provider: 'Ollama' },
+  { id: 18, name: 'anthropic/claude-3.5-sonnet', provider: 'OpenRouter' },
+  { id: 19, name: 'gpt-4o', provider: 'Azure OpenAI' },
 ];
 
 export const openai_models = ['gpt-4o-mini', 'gpt-4o', 'gpt-4-turbo', 'o1'];
@@ -31,10 +32,11 @@ export const anthropic_models = [
 
 export const google_models = [
   'gemini-1.5-pro',
-  'gemini-2.0-flash-exp',
   'gemini-1.5-flash',
+  'gemini-2.0-flash',
+  'gemini-2.0-flash-lite-preview-02-05',
   'gemini-2.0-flash-thinking-exp-01-21',
-  'gemma-2-2b-it',
+  'gemini-2.0-pro-exp-02-05',
 ];
 
 export const groq_models = ['llama-3.3-70b-versatile'];