Update initialize.py

initialize.py

@@ -3,6 +3,9 @@ from langchain_openai import OpenAIEmbeddings
 from langchain_community.vectorstores import Chroma
 import os
 import gen_splits
+from langchain_ollama import OllamaEmbeddings
+
+
 
 GEMINI_API_KEY = os.environ.get('GEMINI_API_KEY')
 OPENAI_API_KEY = os.environ.get('OPENAI_API_KEY')
@@ -10,7 +13,9 @@ OPENAI_API_KEY = os.environ.get('OPENAI_API_KEY')
 # Using Google GenAI Text Embeddings  
 # embedding_model = GoogleGenerativeAIEmbeddings(model="models/embedding-001", task_type="retrieval_query", google_api_key=GEMINI_API_KEY)         ###
 
-embedding_model = OpenAIEmbeddings(api_key=OPENAI_API_KEY, model="text-embedding-3-small")
+#embedding_model = OpenAIEmbeddings(api_key=OPENAI_API_KEY, model="text-embedding-3-small")
+
+embedding_model = OllamaEmbeddings(model="llama3")
 
 # Create Embeddings for Searching the Splits
 persist_directory = './chroma/'