fix: Set default model configuration for the Google provider. (#131)

This commit is contained in:
Drew Hintz
2024-10-09 19:04:39 -05:00
committed by GitHub
parent 3eb5a93f1b
commit 798f346c5a
2 changed files with 2 additions and 0 deletions

View File

@@ -8,6 +8,7 @@ Providers in Goose mean "LLM providers" that Goose can interact with. Providers
* Azure
* Bedrock
* Databricks
* Google
* Ollama
* OpenAI

View File

@@ -102,6 +102,7 @@ def default_model_configuration() -> Tuple[str, str, str]:
"databricks-meta-llama-3-1-70b-instruct",
"databricks-meta-llama-3-1-70b-instruct",
),
"google": ("gemini-1.5-flash", "gemini-1.5-flash"),
}
processor, accelerator = recommended.get(provider, ("gpt-4o", "gpt-4o-mini"))
return provider, processor, accelerator