rosag/Philosophy-RAG-demo
1
0
Fork 0
forked from AI_team/Philosophy-RAG-demo
Code Pull Requests Activity

🐛 If check for pdf source because try doesn't fail

Browse Source
This commit is contained in:
Ruben Lucas 2025-04-09 16:06:15 +02:00
parent ab1235bd28
commit 67ba306d3e
1 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
generic_rag/graphs/ret_gen.py
View File

@ -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()