Update rag.py

rag.py

@@ -79,11 +79,11 @@ def rag_batch(config):
     document_storage_chroma(chunks)
     document_storage_mongodb(chunks)
 
-def document_retrieval_chroma(llm):
+def document_retrieval_chroma():
     return Chroma(embedding_function = OpenAIEmbeddings(disallowed_special = ()),
                   persist_directory = CHROMA_DIR)
 
-def document_retrieval_mongodb(llm):
+def document_retrieval_mongodb():
     return MongoDBAtlasVectorSearch.from_connection_string(MONGODB_ATLAS_CLUSTER_URI,
                                                            MONGODB_DB_NAME + "." + MONGODB_COLLECTION_NAME,
                                                            OpenAIEmbeddings(disallowed_special = ()),