From 236b19c5b31a8f13cbcd2d14e5092e015ade711e Mon Sep 17 00:00:00 2001
From: tronikos <tronikos@users.noreply.github.com>
Date: Fri, 31 May 2024 21:57:14 -0700
Subject: [PATCH] Use gemini-1.5-flash-latest in
 google_generative_ai_conversation.generate_content (#118594)

---
 .../components/google_generative_ai_conversation/__init__.py  | 3 +--
 .../snapshots/test_init.ambr                                  | 4 ++--
 2 files changed, 3 insertions(+), 4 deletions(-)

diff --git a/homeassistant/components/google_generative_ai_conversation/__init__.py b/homeassistant/components/google_generative_ai_conversation/__init__.py
index b2723f82030..523198355d1 100644
--- a/homeassistant/components/google_generative_ai_conversation/__init__.py
+++ b/homeassistant/components/google_generative_ai_conversation/__init__.py
@@ -66,8 +66,7 @@ async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
                 }
             )
 
-        model_name = "gemini-pro-vision" if image_filenames else "gemini-pro"
-        model = genai.GenerativeModel(model_name=model_name)
+        model = genai.GenerativeModel(model_name=RECOMMENDED_CHAT_MODEL)
 
         try:
             response = await model.generate_content_async(prompt_parts)
diff --git a/tests/components/google_generative_ai_conversation/snapshots/test_init.ambr b/tests/components/google_generative_ai_conversation/snapshots/test_init.ambr
index aba3f35eb19..f68f4c6bf14 100644
--- a/tests/components/google_generative_ai_conversation/snapshots/test_init.ambr
+++ b/tests/components/google_generative_ai_conversation/snapshots/test_init.ambr
@@ -6,7 +6,7 @@
       tuple(
       ),
       dict({
-        'model_name': 'gemini-pro-vision',
+        'model_name': 'models/gemini-1.5-flash-latest',
       }),
     ),
     tuple(
@@ -32,7 +32,7 @@
       tuple(
       ),
       dict({
-        'model_name': 'gemini-pro',
+        'model_name': 'models/gemini-1.5-flash-latest',
       }),
     ),
     tuple(