Short-term memory with MongoDB Atlas

export MONGODB_ATLAS_URI="your-atlas-connection-string"

export LANGSMITH_API_KEY="your-api-key"
export LANGSMITH_TRACING="true"

import { createAgent, tool } from "langchain";
import { MongoDBSaver } from "@langchain/langgraph-checkpoint-mongodb";
import { MongoClient } from "mongodb";
import * as z from "zod";

const getUserInfo = tool(() => "No user profile on file.", {
  name: "get_user_info",
  description: "Look up information about the current user.",
  schema: z.object({}),
});

const MONGODB_ATLAS_URI = process.env.MONGODB_ATLAS_URI;
if (!MONGODB_ATLAS_URI) {
  throw new Error("MONGODB_ATLAS_URI is required");
}

const client = new MongoClient(MONGODB_ATLAS_URI);
const checkpointer = new MongoDBSaver({ client });
await checkpointer.setup();

const agent = createAgent({
  model: "anthropic:claude-sonnet-4-6",
  tools: [getUserInfo],
  checkpointer,
});

const threadConfig = { configurable: { thread_id: "1" } };
let result = await agent.invoke(
  { messages: [{ role: "user", content: "Hi! My name is Bob." }] },
  threadConfig,
);
console.log(result.messages.at(-1)?.content);

result = await agent.invoke(
  { messages: [{ role: "user", content: "What's my name?" }] },
  threadConfig,
);
console.log(result.messages.at(-1)?.content);

await client.close();