diff --git a/src/oss/langchain/short-term-memory.mdx b/src/oss/langchain/short-term-memory.mdx
index fb771648f9..f7c0eb93e0 100644
--- a/src/oss/langchain/short-term-memory.mdx
+++ b/src/oss/langchain/short-term-memory.mdx
@@ -952,6 +952,7 @@ graph TD
 :::python
 ```python
 from langchain.messages import RemoveMessage
+from langchain_core.runnables import RunnableConfig
 from langgraph.graph.message import REMOVE_ALL_MESSAGES
 from langgraph.checkpoint.memory import InMemorySaver
 from langchain.agents import create_agent, AgentState
@@ -982,7 +983,8 @@ def trim_messages(state: AgentState, runtime: Runtime) -> dict[str, Any] | None:
 agent = create_agent(
     model,
     tools=tools,
-    middleware=[trim_messages]
+    middleware=[trim_messages],
+    checkpointer=InMemorySaver()
 )
 
 config: RunnableConfig = {"configurable": {"thread_id": "1"}}
@@ -1005,6 +1007,7 @@ If you'd like me to call you a nickname or use a different name, just say the wo
 ```typescript
 import { RemoveMessage } from "@langchain/core/messages";
 import { createAgent, createMiddleware, trimMessages, type AgentState } from "langchain";
+import { MemorySaver } from "@langchain/langgraph";
 
 const trimMessageHistory = createMiddleware({
   name: "TrimMessages",
@@ -1020,10 +1023,12 @@ const trimMessageHistory = createMiddleware({
   },
 });
 
+const checkpointer = new MemorySaver();
 const agent = createAgent({
     model: "gpt-5-nano",
     tools: [],
     middleware: [trimMessageHistory],
+    checkpointer,
 });
 ```
 :::