Use stable embeddings

NeonGeckoCom · Jan 17, 2024 · a75c702 · a75c702
1 parent ed7f9df
commit a75c702
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/neon_llm_gemini/gemini.py b/neon_llm_gemini/gemini.py
@@ -26,7 +26,7 @@
 
 import os
 from vertexai.preview.generative_models import GenerativeModel, Content, Part
-from vertexai.preview.language_models import TextEmbeddingModel
+from vertexai.language_models import TextEmbeddingModel
 from openai.embeddings_utils import distances_from_embeddings
 
 from typing import List, Dict