Skip to content
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 6 additions & 1 deletion src/oss/langchain/short-term-memory.mdx
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down Expand Up @@ -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"}}
Expand All @@ -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",
Expand All @@ -1020,10 +1023,12 @@ const trimMessageHistory = createMiddleware({
},
});

const checkpointer = new MemorySaver();
const agent = createAgent({
model: "gpt-5-nano",
tools: [],
middleware: [trimMessageHistory],
checkpointer,
});
```
:::
Expand Down