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

Fixes small bug in RetGenLangGraph

Browse Source
This commit is contained in:
Nielson Janné 2025-03-17 14:46:16 +01:00
parent 2ba3eadec4
commit 3965ce0fb2
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
generic_rag/graphs/ret_gen.py
View File

@ -27,7 +27,7 @@ class RetGenLangGraph:
self.last_invoke = None self.last_invoke = None
def invoke(self, message: str) -> Union[dict[str, Any], Any]: def invoke(self, message: str) -> Union[dict[str, Any], Any]:
self.last_invoke = self.graph.invoke(message) self.last_invoke = self.graph.invoke({"question": message})
return self.last_invoke return self.last_invoke
def _retrieve(self, state: State) -> dict: def _retrieve(self, state: State) -> dict: