Update rag_demo/rag/retriever.py

rag_demo/rag/retriever.py

@@ -129,5 +129,5 @@ class RAGPipeline:
         reranked_docs = self.rerank(query, docs, keep_top_k=10)
         return (
             self.generate_answer(query, reranked_docs),
-            [doc.metadata["filename"].split(".pdf")[0] for doc in reranked_docs],
+            list(set([doc.metadata["filename"].split(".pdf")[0] for doc in reranked_docs])),
         )