diff --git a/generic_rag/graphs/ret_gen.py b/generic_rag/graphs/ret_gen.py index b12f7e1..9cb69bb 100644 --- a/generic_rag/graphs/ret_gen.py +++ b/generic_rag/graphs/ret_gen.py @@ -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