From baf3dbeec1b10c3bb2111f39f5a9d53ebaaa1f39 Mon Sep 17 00:00:00 2001
From: Gui Quental <guilherme@quent.al>
Date: Tue, 15 Oct 2024 10:21:24 -0300
Subject: [PATCH] Change SDK to GA instead of preview

---
 src/vanna/google/gemini_chat.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/vanna/google/gemini_chat.py b/src/vanna/google/gemini_chat.py
index 95bffb70..60732472 100644
--- a/src/vanna/google/gemini_chat.py
+++ b/src/vanna/google/gemini_chat.py
@@ -30,7 +30,7 @@ def __init__(self, config=None):
             self.chat_model = genai.GenerativeModel(model_name)
         else:
             # Authenticate using VertexAI
-            from vertexai.preview.generative_models import GenerativeModel
+            from vertexai.generative_models import GenerativeModel
             self.chat_model = GenerativeModel(model_name)
 
     def system_message(self, message: str) -> any: