Fix a small bug in report web page source

This commit is contained in:
Nielson Janné 2025-04-09 11:17:09 +02:00
parent 9bb9f0ea22
commit 9baa7b0ef6

View File

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