Update rag.py

rag.py

@@ -2,9 +2,13 @@
 from langchain_community.document_loaders import PyPDFLoader
 from langchain.text_splitter import RecursiveCharacterTextSplitter
 from langchain.vectorstores import Chroma
-from langchain.embeddings import OpenAIEmbeddings
+from langchain.embeddings import HuggingFaceEmbeddings
 from langchain.vectorstores import Chroma
+import os
+from dotenv import load_dotenv
 
+load_dotenv()
+os.environ['OPENAI_API_KEY'] = os.getenv('OPENROUTE_API_KEY')
 def dataIngestion( document):
     loader = PyPDFLoader(document)
     ingested_docs = loader.load()
@@ -19,6 +23,6 @@ def transform( ingested_docs):
 
 
 def vectorStoreAndEmbeddings(docs, query):
-    embeddings = OpenAIEmbeddings()
-    db = Chroma.from_documents(documents=docs, embedding_function=embeddings)
+    embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")
+    db = Chroma.from_documents(documents=docs, embedding=embeddings)
     return db.similarity_search(query)[0].page_content