aljaz/goose
1
0
Fork 0
mirror of https://github.com/aljazceru/goose.git synced 2025-12-24 01:24:28 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix: fix gemini model not found (#2157)

Browse Source
This commit is contained in:
Yingjie He
2025-04-11 10:41:48 -07:00
committed by GitHub
parent 0b800e05b7
commit 9bd27cebcb
3 changed files with 24 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
crates/goose/src/providers/google.rs
View File

@@ -17,14 +17,15 @@ 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_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",
"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",
"gemini-1.5-pro-latest",
"gemini-1.5-pro",
"gemini-1.5-flash-latest",
"gemini-1.5-flash",
"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",
];
pub const GOOGLE_DOC_URL: &str = "https://ai.google/get-started/our-models/";

25
ui/desktop/src/components/settings/models/GooseModels.tsx
View File

@@ -16,16 +16,17 @@ export const gooseModels: Model[] = [
{ 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' },
{ id: 20, name: 'claude-3-7-sonnet@20250219', provider: 'GCP Vertex AI' },
{ id: 21, name: 'claude-3-5-sonnet-v2@20241022', provider: 'GCP Vertex AI' },
{ id: 22, name: 'claude-3-5-sonnet@20240620', provider: 'GCP Vertex AI' },
{ id: 23, name: 'claude-3-5-haiku@20241022', provider: 'GCP Vertex AI' },
{ id: 24, name: 'gemini-2.0-pro-exp-02-05', provider: 'GCP Vertex AI' },
{ id: 25, name: 'gemini-2.0-flash-001', provider: 'GCP Vertex AI' },
{ id: 26, name: 'gemini-1.5-pro-002', provider: 'GCP Vertex AI' },
{ id: 27, name: 'gemini-2.5-pro-exp-03-25', provider: 'GCP Vertex AI' },
{ id: 16, name: 'gemini-2.5-pro-exp-03-25', provider: 'Google' },
{ id: 17, name: 'llama-3.3-70b-versatile', provider: 'Groq' },
{ id: 18, name: 'qwen2.5', provider: 'Ollama' },
{ id: 19, name: 'anthropic/claude-3.5-sonnet', provider: 'OpenRouter' },
{ id: 20, name: 'gpt-4o', provider: 'Azure OpenAI' },
{ id: 21, name: 'claude-3-7-sonnet@20250219', provider: 'GCP Vertex AI' },
{ id: 22, name: 'claude-3-5-sonnet-v2@20241022', provider: 'GCP Vertex AI' },
{ id: 23, name: 'claude-3-5-sonnet@20240620', provider: 'GCP Vertex AI' },
{ id: 24, name: 'claude-3-5-haiku@20241022', provider: 'GCP Vertex AI' },
{ id: 25, name: 'gemini-2.0-pro-exp-02-05', provider: 'GCP Vertex AI' },
{ id: 26, name: 'gemini-2.0-flash-001', provider: 'GCP Vertex AI' },
{ id: 27, name: 'gemini-1.5-pro-002', provider: 'GCP Vertex AI' },
{ id: 28, name: 'gemini-2.5-pro-exp-03-25', provider: 'GCP Vertex AI' },
];

3
ui/desktop/src/components/settings/models/hardcoded_stuff.tsx
View File

@@ -14,6 +14,7 @@ export const google_models = [
'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',
];
export const groq_models = ['llama-3.3-70b-versatile'];
@@ -49,7 +50,7 @@ export const default_models = {
};
export function getDefaultModel(key: string): string | undefined {
return default_models[key] || undefined;
return default_models[key as keyof typeof default_models] || undefined;
}
export const short_list = ['gpt-4o', 'claude-3-5-sonnet-latest'];