From 3965ce0fb24a6e82cd14e118b31e4864cbe39c70 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Nielson=20Jann=C3=A9?= <nielsonj@gmail.com>
Date: Mon, 17 Mar 2025 14:46:16 +0100
Subject: [PATCH] Fixes small bug in RetGenLangGraph

---
 generic_rag/graphs/ret_gen.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/generic_rag/graphs/ret_gen.py b/generic_rag/graphs/ret_gen.py
index 63605d3..fd51259 100644
--- a/generic_rag/graphs/ret_gen.py
+++ b/generic_rag/graphs/ret_gen.py
@@ -27,7 +27,7 @@ class RetGenLangGraph:
         self.last_invoke = None
 
     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
     
     def _retrieve(self, state: State) -> dict: