From 9baa7b0ef660e1d1e1dee25ce828221d1316bace Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Nielson=20Jann=C3=A9?= <nielsonj@gmail.com>
Date: Wed, 9 Apr 2025 11:17:09 +0200
Subject: [PATCH] Fix a small bug in report web page source

---
 generic_rag/graphs/ret_gen.py | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

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