Update gemini version to 1.5

vanna-ai · Oct 15, 2024 · 55563c8 · 55563c8
1 parent c21d8bf
commit 55563c8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/vanna/google/gemini_chat.py b/src/vanna/google/gemini_chat.py
@@ -15,7 +15,7 @@ def __init__(self, config=None):
         if "model_name" in config:
             model_name = config["model_name"]
         else:
-            model_name = "gemini-1.0-pro"
+            model_name = "gemini-1.5-pro"
 
         self.google_api_key = None