Skip to main content
Pre-release

This release is a pre-release and may not be stable for production use.

LangMem

LangMem helps agents learn and adapt from their interactions over time.

It provides tooling to extract important information from conversations, optimize agent behavior through prompt refinement, and maintain long-term memory.

It offers both functional primitives you can use with any storage system and native integration with LangGraph's storage layer.

This lets your agents continuously improve, personalize their responses, and maintain consistent behavior across sessions.

Key features

  • 🧩 Core memory API that works with any storage system
  • 🧠 Memory management tools that agents can use to record and search information during active conversations "in the hot path"
  • ⚙️ Background memory manager that automatically extracts, consolidates, and updates agent knowledge
  • Native integration with LangGraph's Long-term Memory Store, available by default in all LangGraph Platform deployments

Installation

pip install -U langmem

Configure your environment with an API key for your favorite LLM provider:

export ANTHROPIC_API_KEY="sk-..."  # Or another supported LLM provider

Here's how to create an agent that actively manages its own long-term memory in just a few lines:

# Import core components (1)
from langgraph.prebuilt import create_react_agent
from langgraph.store.memory import InMemoryStore
from langmem import create_manage_memory_tool, create_search_memory_tool

# Set up storage
store = InMemoryStore() # (2)

# Create an agent with memory capabilities (3)
agent = create_react_agent(
    "anthropic:claude-3-5-sonnet-latest",
    tools=[
        # Memory tools use LangGraph's BaseStore for persistence (4)
        create_manage_memory_tool(namespace=("memories",)),
        create_search_memory_tool(namespace=("memories",)),
    ],
    store=store,
)
  1. The memory tools work in any LangGraph app. Here we use create_react_agent to run an LLM with tools, but you can add these tools to your existing agents or build custom memory systems without agents.

  2. InMemoryStore keeps memories in process memory—they'll be lost on restart. For production, use the AsyncPostgresStore or a similar DB-backed store to persist memories across server restarts.

  3. The memory tools (create_manage_memory_tool and create_search_memory_tool) let you control what gets stored. The agent extracts key information from conversations, maintains memory consistency, and knows when to search past interactions. See Memory Tools for configuration options.

Then use the agent:

# Store a new memory (1)
agent.invoke(
    {"messages": [{"role": "user", "content": "Remember that I prefer dark mode."}]}
)

# Retrieve the stored memory (2)
response = agent.invoke(
    {"messages": [{"role": "user", "content": "What are my lighting preferences?"}]}
)
print(response["messages"][-1].content)
# Output: "You've told me that you prefer dark mode."
  1. The agent gets to decide what and when to store the memory. No special commands needed—just chat normally and the agent uses create_manage_memory_tool to store relevant details.

  2. The agent maintains context between chats. When you ask about previous interactions, the LLM can invoke create_search_memory_tool to search for memories with similar content. See Memory Tools to customize memory storage and retrieval, and see the hot path quickstart for a more complete example on how to include memories without the agent having to expliictly search.

The agent can now store important information from conversations, search its memory when relevant, and persist knowledge across conversations.

Next Steps

For more examples and detailed documentation:

Requirements

  • Python 3.11+
  • Access to a supported LLM provider (Anthropic, OpenAI, etc.)
  • Optional: a LangGraph BaseStore implementation for persistent storage (for the stateful primitives); if you're deploying on LangGraph Platform, this is included without any additional configuration.

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

langmem-0.0.5rc13.tar.gz (47.7 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

langmem-0.0.5rc13-py3-none-any.whl (52.3 kB view details)

Uploaded Python 3

File details

Details for the file langmem-0.0.5rc13.tar.gz.

File metadata

  • Download URL: langmem-0.0.5rc13.tar.gz
  • Upload date:
  • Size: 47.7 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: uv/0.4.29

File hashes

Hashes for langmem-0.0.5rc13.tar.gz
Algorithm Hash digest
SHA256 b1144f2ec275311311e0681373c0ce0c746110d6733c37fc5b4aa18cc8bb75a0
MD5 b76c8914d44453f3dca6a2ce2bf558e2
BLAKE2b-256 4742a490a5bf5095fffa4dc5f537d66bd3a29b331fd89d39999b84dbd67c1904

See more details on using hashes here.

File details

Details for the file langmem-0.0.5rc13-py3-none-any.whl.

File metadata

File hashes

Hashes for langmem-0.0.5rc13-py3-none-any.whl
Algorithm Hash digest
SHA256 a80029efa516489a6aae5ab25499b6c9beb7db0b8e76bcfbeef0c476107796a7
MD5 5d7dd93c48fbea43cba06024a5c1bb6e
BLAKE2b-256 86482d37f87084553f4f109c554430f95e2b18ab2cf0819d22043f6ee08aeb91

See more details on using hashes here.

Release history Release notifications | RSS feed

0.0.30

2 files

0.0.29

2 files

0.0.28

2 files

0.0.27

2 files

0.0.26

2 files

0.0.25

2 files

0.0.24

2 files

0.0.23

2 files

0.0.22

2 files

0.0.21

2 files

0.0.20

2 files

0.0.19

2 files

0.0.17

2 files

0.0.16

2 files

0.0.15

2 files

0.0.14

1 file

0.0.13

2 files

0.0.12

1 file

0.0.11

2 files

0.0.10

2 files

0.0.9

2 files

0.0.8

2 files

0.0.6

2 files

This release

0.0.5rc13 This release

2 files

0.0.4

2 files

0.0.3

2 files

0.0.2

2 files

0.0.1

2 files

Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page