add pointer to memory

2026-01-26 16:42:53 +08:00
parent f55241abf1
commit 6784f97e06
1 changed files with 2 additions and 2 deletions
--- a/lang_agent/graphs/react.py
+++ b/lang_agent/graphs/react.py
@@ -47,9 +47,9 @@ class ReactGraph(GraphBase):
        

        self.tool_manager:ToolManager = self.config.tool_manager_config.setup()
-        memory = MemorySaver()
+        self.memory = MemorySaver()
        tools = self.tool_manager.get_langchain_tools()
-        self.agent = create_agent(self.llm, tools, checkpointer=memory)
+        self.agent = create_agent(self.llm, tools, checkpointer=self.memory)
    
    def invoke(self, *nargs, as_stream:bool=False, as_raw:bool=False, **kwargs):
        """