Update rag_langchain.py

rag_langchain.py

@@ -16,7 +16,6 @@ from langchain.vectorstores import MongoDBAtlasVectorSearch
 from pymongo import MongoClient
 
 RAG_CHROMA  = "Chroma"
-RAG_MONGODB = "MongoDB"
 
 PDF_URL       = "https://arxiv.org/pdf/2303.08774.pdf"
 WEB_URL       = "https://openai.com/research/gpt-4"
@@ -91,7 +90,7 @@ def rag_ingestion(config):
     
     chunks = split_documents(config, docs)
     
-    store_chroma(chunks)
+    #store_chroma(chunks)
     store_mongodb(chunks)
 
 def retrieve_chroma():
@@ -126,7 +125,7 @@ def rag_chain(config, rag_option, prompt):
 
     if (rag_option == RAG_CHROMA):
         db = retrieve_chroma()
-    elif (rag_option == RAG_MONGODB):
+    else:
         db = retrieve_mongodb()
 
     rag_chain = RetrievalQA.from_chain_type(