Update chroma_utils.py

chroma_utils.py

@@ -5,9 +5,11 @@ from langchain_chroma import Chroma
 from typing import List
 from langchain_core.documents import Document
 import os
+from langchain_google_genai import GoogleGenerativeAIEmbeddings
+
 
 text_splitter = RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=200, length_function=len)
-embedding_function = OpenAIEmbeddings()
+embedding_function = GoogleGenerativeAIEmbeddings(model="models/embedding-001")
 vectorstore = Chroma(persist_directory="./chroma_db", embedding_function=embedding_function)
 
 def load_and_split_document(file_path: str) -> List[Document]: