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

small refacter in ret_gen.py

Browse Source 
- improve readability
- reduce chance for potential mistake in future
This commit is contained in:
Nielson Janné 2025-03-28 18:19:55 +01:00
parent cd14c8add2
commit f3429a15b1
1 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
generic_rag/graphs/ret_gen.py
View File

@ -48,10 +48,11 @@ class RetGenLangGraph:
"""
Method that retrieves the PDF sources used during the last invoke.
"""
if self.last_invoke is None:
return []
pdf_sources = {}
if self.last_invoke is None:
return pdf_sources
for context in self.last_invoke["context"]:
try:
Path(context.metadata["source"]).suffix == ".pdf"
@ -84,10 +85,11 @@ class RetGenLangGraph:
"""
Method that retrieves the web sources used during the last invoke.
"""
if self.last_invoke is None:
return set()
web_sources = set()
if self.last_invoke is None:
return web_sources
for context in self.last_invoke["context"]:
try:
context.metadata["filetype"] == "web"