Update rag.py

rag.py

@@ -107,9 +107,9 @@ def rag_chain(config, openai_api_key, rag_option, prompt):
     llm = get_llm(config, openai_api_key)
     
     if (rag_option == RAG_CHROMA):
-        db = document_retrieval_chroma(llm)
+        db = document_retrieval_chroma()
     elif (rag_option == RAG_MONGODB):
-        db = document_retrieval_mongodb(llm)
+        db = document_retrieval_mongodb()
 
     rag_chain = RetrievalQA.from_chain_type(llm, 
                                             chain_type_kwargs = {"prompt": RAG_CHAIN_PROMPT},