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:
|
for doc in self.last_retrieved_docs:
|
||||||
try:
|
try:
|
||||||
doc.metadata["filetype"] == "web"
|
if doc.metadata["filetype"] == "web":
|
||||||
|
web_sources.add(doc.metadata["source"])
|
||||||
except KeyError:
|
except KeyError:
|
||||||
continue
|
continue
|
||||||
else:
|
|
||||||
web_sources.add(doc.metadata["source"])
|
|
||||||
|
|
||||||
return web_sources
|
return web_sources
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user