Change SDK to GA instead of preview

vanna-ai · Oct 15, 2024 · baf3dbe · baf3dbe
1 parent 55563c8
commit baf3dbe
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
@@ -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: