From 949add2b6cc719bccaa67431f398e0ec252184a9 Mon Sep 17 00:00:00 2001
From: Drew Hintz <adhintz@utexas.edu>
Date: Tue, 8 Oct 2024 21:53:34 -0500
Subject: [PATCH] Set default model configuration for the Google provider. Add
 Google to the list of providers.

---
 docs/plugins/providers.md | 1 +
 src/goose/cli/config.py   | 1 +
 2 files changed, 2 insertions(+)

diff --git a/docs/plugins/providers.md b/docs/plugins/providers.md
index 7527f798..63e5e80f 100644
--- a/docs/plugins/providers.md
+++ b/docs/plugins/providers.md
@@ -8,6 +8,7 @@ Providers in Goose mean "LLM providers" that Goose can interact with. Providers
 * Azure
 * Bedrock
 * Databricks
+* Google
 * Ollama
 * OpenAI
 
diff --git a/src/goose/cli/config.py b/src/goose/cli/config.py
index 7bede0be..109706b7 100644
--- a/src/goose/cli/config.py
+++ b/src/goose/cli/config.py
@@ -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