forked from AI_team/Philosophy-RAG-demo
Fix a small bug in report web page source
This commit is contained in:
parent
9bb9f0ea22
commit
9baa7b0ef6
@ -102,10 +102,9 @@ class RetGenLangGraph:
|
||||
|
||||
for doc in self.last_retrieved_docs:
|
||||
try:
|
||||
doc.metadata["filetype"] == "web"
|
||||
if doc.metadata["filetype"] == "web":
|
||||
web_sources.add(doc.metadata["source"])
|
||||
except KeyError:
|
||||
continue
|
||||
else:
|
||||
web_sources.add(doc.metadata["source"])
|
||||
|
||||
return web_sources
|
||||
|
||||
Loading…
Reference in New Issue
Block a user