michel/Philosophy-RAG-demo
1
0
Fork 0
forked from AI_team/Philosophy-RAG-demo
Code Pull Requests Activity

🎨 Reset retrieved sources

Browse Source
This commit is contained in:
Ruben Lucas 2025-04-09 16:30:39 +02:00
parent 67ba306d3e
commit df2afd73cb
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
generic_rag/graphs/cond_ret_gen.py
View File

@ -95,6 +95,10 @@ class CondRetGenLangGraph:
def _query_or_respond(self, state: MessagesState) -> dict[str, BaseMessage]: def _query_or_respond(self, state: MessagesState) -> dict[str, BaseMessage]:
"""Generate tool call for retrieval or respond.""" """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]) llm_with_tools = self.chat_model.bind_tools([self._retrieve])
response = llm_with_tools.invoke(state["messages"]) response = llm_with_tools.invoke(state["messages"])
return {"messages": [response]} return {"messages": [response]}