Update rag_chain/chain.py

rag_chain/chain.py

@@ -115,6 +115,7 @@ def get_rag_chain(
     practitioners_db_dense_retriever = dense_retriever_client.get_dense_retriever()
 
     # Multiquery retriever using the dense retriever
+    # This retriever can be passed or not to the EnsembleRetriever. It uses GPT-3.5-turbo.
     practitioners_db_dense_multiquery_retriever = multi_query_retriever_setup(
         practitioners_db_dense_retriever
     )
@@ -132,7 +133,7 @@ def get_rag_chain(
     # Ensemble retriever for hyprid search (dense retriever seems to work better but the dense retriever is good for acronyms like RMT)
     practitioners_ensemble_retriever = EnsembleRetriever(
         retrievers=[
-            practitioners_db_dense_multiquery_retriever,
+            practitioners_db_dense_retriever,
             practitioners_db_sparse_retriever,
         ],
         weights=RETRIEVER_PARAMETERS["weights_ensemble_practitioners_db"],