From 67ba306d3e02ac704c41797b2726ddcf9f6f661c Mon Sep 17 00:00:00 2001 From: Ruben Lucas Date: Wed, 9 Apr 2025 16:06:15 +0200 Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20If=20check=20for=20pdf=20source?= =?UTF-8?q?=20because=20try=20doesn't=20fail?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- generic_rag/graphs/ret_gen.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/generic_rag/graphs/ret_gen.py b/generic_rag/graphs/ret_gen.py index 9cb69bb..9a05349 100644 --- a/generic_rag/graphs/ret_gen.py +++ b/generic_rag/graphs/ret_gen.py @@ -61,12 +61,11 @@ class RetGenLangGraph: return pdf_sources for doc in self.last_retrieved_docs: - try: - Path(doc.metadata["source"]).suffix == ".pdf" - except KeyError: - continue - else: + source_candidate = doc.metadata["source"] + if "source" in doc.metadata and Path(doc.metadata["source"]).suffix.lower() == ".pdf": source = doc.metadata["source"] + else: + continue if source not in pdf_sources: pdf_sources[source] = set()