fix: reranking was done two times

rag_demo/rag/retriever.py

@@ -135,24 +135,17 @@ class RAGPipeline:
         logger.info(f"Query type: {query_type}")
         if query_type == "Sources_needed":
             docs = self.search(query, k=10)
-            # disabling reranking for now because HFApi is too slow
-            # reranked_docs = self.rerank(query, docs, keep_top_k=10)
-            reranked_docs = docs
         else:
-            reranked_docs = []
+            docs = []
 
-        answer = self.generate_answer(query, reranked_docs)
+        answer = self.generate_answer(query, docs)
 
-        if reranked_docs:
-            annotated_answer = self.add_context(answer, reranked_docs)
+        if docs:
+            annotated_answer = self.add_context(answer, docs)
         else:
             annotated_answer = answer
 
         return (
             annotated_answer,
-            list(
-                set(
-                    [doc.metadata["filename"].split(".pdf")[0] for doc in reranked_docs]
-                )
-            ),
+            list(set([doc.metadata["filename"].split(".pdf")[0] for doc in docs])),
         )