support store/checkpoint feature

Author: ks6088ts

docs/references.md

@@ -17,6 +17,9 @@
 - [Custom UI for Deep Agents](https://github.com/langchain-ai/deep-agents-ui)
 - [How to deploy self-hosted standalone server](https://docs.langchain.com/langgraph-platform/deploy-standalone-server)
 - [「現場で活用するための AI エージェント実践入門」リポジトリ](https://github.com/masamasa59/genai-agent-advanced-book)
+- [Add and manage memory](https://docs.langchain.com/oss/python/langgraph/add-memory)
+- [Persistence](https://langchain-ai.github.io/langgraph/concepts/persistence/)
+- [Chatbot with message summarization & external DB memory](https://github.com/langchain-ai/langchain-academy/blob/main/module-2/chatbot-external-memory.ipynb)
 
 ### LangChain
 

template_langgraph/agents/chat_with_tools_agent/agent.py

@@ -50,9 +50,16 @@ def __call__(self, inputs: dict):
 
 
 class ChatWithToolsAgent:
-    def __init__(self, tools=get_default_tools()):
+    def __init__(
+        self,
+        tools=get_default_tools(),
+        checkpointer=None,
+        store=None,
+    ):
         self.llm = AzureOpenAiWrapper().chat_model
         self.tools = tools
+        self.checkpointer = checkpointer
+        self.store = store
 
     def create_graph(self):
         """Create the main graph for the agent."""
@@ -83,6 +90,8 @@ def create_graph(self):
         # Compile the graph
         return workflow.compile(
             name=ChatWithToolsAgent.__name__,
+            checkpointer=self.checkpointer,
+            store=self.store,
         )
 
     def chat_with_tools(self, state: AgentState) -> AgentState:

template_langgraph/services/streamlits/pages/chat_with_tools_agent.py

@@ -9,6 +9,8 @@
     StreamlitCallbackHandler,
 )
 from langfuse.langchain import CallbackHandler
+from langgraph.checkpoint.memory import InMemorySaver
+from langgraph.store.memory import InMemoryStore
 
 from template_langgraph.agents.chat_with_tools_agent.agent import (
     AgentState,
@@ -68,7 +70,11 @@ def ensure_agent_graph(selected_tools: list) -> None:
     signature = tuple(tool.name for tool in selected_tools)
     graph_signature = st.session_state.get("graph_tools_signature")
     if "graph" not in st.session_state or graph_signature != signature:
-        st.session_state["graph"] = ChatWithToolsAgent(tools=selected_tools).create_graph()
+        st.session_state["graph"] = ChatWithToolsAgent(
+            tools=selected_tools,
+            checkpointer=InMemorySaver(),  # You can replace this with a persistent checkpointer if needed
+            store=InMemoryStore(),  # You can replace this with a persistent store if needed
+        ).create_graph()
         st.session_state["graph_tools_signature"] = signature
 
 
@@ -296,12 +302,18 @@ def build_graph_messages() -> list:
 
 def invoke_agent(graph_messages: list) -> AgentState:
     return st.session_state["graph"].invoke(
-        {"messages": graph_messages},
+        {
+            "messages": graph_messages,
+        },
         {
             "callbacks": [
                 StreamlitCallbackHandler(st.container()),
                 CallbackHandler(),
-            ]
+            ],
+            "configurable": {
+                "thread_id": "1",
+                "user_id": "user_1",
+            },
         },
     )