Update main.py

main.py

@@ -26,8 +26,6 @@ fs = firestore.client()
 # --------- Gemini Configuration ---------
 genai.configure(api_key=os.getenv("Gemini"))
 chat_model = genai.GenerativeModel("gemini-2.0-flash-thinking-exp")
-embed_model = genai.EmbeddingModel("models/embedding-001")
-
 # --------- Paths for Cached Index ---------
 INDEX_PATH = "vector.index"
 DOCS_PATH = "documents.pkl"
@@ -85,7 +83,9 @@ def build_or_load_index():
     return documents, index
 
 def get_embeddings(texts):
-    response = embed_model.get_embeddings(texts=texts)
+    response= genai.embed_content(
+            model="models/text-embedding-004", content=texts, output_dimensionality=10
+        )
     return [e.values for e in response.embeddings]
 
 documents, index = build_or_load_index()