From df2afd73cb67e5bead9614a317ca3042464f08a5 Mon Sep 17 00:00:00 2001
From: Ruben Lucas <ruben.lucas1997@gmail.com>
Date: Wed, 9 Apr 2025 16:30:39 +0200
Subject: [PATCH] =?UTF-8?q?=F0=9F=8E=A8=20Reset=20retrieved=20sources?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 generic_rag/graphs/cond_ret_gen.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/generic_rag/graphs/cond_ret_gen.py b/generic_rag/graphs/cond_ret_gen.py
index c884a8e..5d3a6ed 100644
--- a/generic_rag/graphs/cond_ret_gen.py
+++ b/generic_rag/graphs/cond_ret_gen.py
@@ -95,6 +95,10 @@ class CondRetGenLangGraph:
 
     def _query_or_respond(self, state: MessagesState) -> dict[str, BaseMessage]:
         """Generate tool call for retrieval or respond."""
+        # Reset last retrieved docs
+        self.last_retrieved_docs = {}
+        self.last_retrieved_sources = set()
+
         llm_with_tools = self.chat_model.bind_tools([self._retrieve])
         response = llm_with_tools.invoke(state["messages"])
         return {"messages": [response]}