Update src/rag_pipeline.py

src/rag_pipeline.py

@@ -365,8 +365,8 @@ def run_rag(query_asked, input_dataset):
     strategy = global_vars["repacking_strategy"]
     vector_db = global_vars["vector_db"]
     print("Starting retrived docs")
-    # retrieved_docs = retrieve_documents(query_asked, input_dataset, embedding_model, vector_db)
-    retrieved_docs = retrieve_documents("where is broad run va", "msmarco", "BAAI/llm-embedder", "qdrant")
+    retrieved_docs = retrieve_documents(query_asked, input_dataset, embedding_model, vector_db)
+    # retrieved_docs = retrieve_documents("where is broad run va", "msmarco", "BAAI/llm-embedder", "qdrant")
     print(retrieved_docs)
     print("Starting reranking")
     if len(retrieved_docs) > 0: