diff --git a/generic_rag/graphs/ret_gen.py b/generic_rag/graphs/ret_gen.py index 9a05349..280f8cb 100644 --- a/generic_rag/graphs/ret_gen.py +++ b/generic_rag/graphs/ret_gen.py @@ -61,7 +61,6 @@ class RetGenLangGraph: return pdf_sources for doc in self.last_retrieved_docs: - source_candidate = doc.metadata["source"] if "source" in doc.metadata and Path(doc.metadata["source"]).suffix.lower() == ".pdf": source = doc.metadata["source"] else: