Skip to main content

Agent Memory

CI Python 3.10+ License: MIT PyPI version

Persistent semantic memory for AI agents with intelligent decision-making.

Agent Memory CLI demo: exact query replays, shared-word trap correctly returns none

🚀 Created by: TheProdSDE


The problem

Most AI memory systems retrieve and inject past context into every prompt. This leads to wasted tokens, inconsistent responses, and agents that blindly replay stale or wrong answers.

Agent Memory adds a decision layer:

flowchart TD
    A[User Query] --> B[Resolve Memory]
    B --> C[Decision Engine]
    C -->|High confidence match| D[🔄 Replay — return stored answer]
    C -->|Moderate match| E[📋 Restore — inject as context]
    C -->|Needs validation| F[✅ Verify — validate before reuse]
    C -->|No match| G[❌ None — answer from scratch]

    style D fill:#0d47a1,color:#fff
    style E fill:#e65100,color:#fff
    style F fill:#1b5e20,color:#fff
    style G fill:#b71c1c,color:#fff

Every resolve() returns an explicit action with a scored, explainable rationale — not just a retrieved chunk. Adversarial eval: 25/25 (100%) on trap queries — see benchmarks.


When to use it — real use cases

Use case Without memory With Agent Memory Saving
Support bot handling 10k identical FAQ queries/day Every query costs 1 LLM call ~75% REPLAY on repeated questions, 0 LLM calls 75% cost reduction
Coding agent that re-derives project conventions each session Wastes 2–5 LLM calls per session to "remember" conventions Workflows are REPLAYED instantly on first query No re-derivation overhead
Research agent building knowledge over multiple sessions Each session starts cold; re-reads the same sources Facts and summaries are RESTORED as context Persistent cross-session knowledge
Customer onboarding bot answering the same steps repeatedly Always generates a response High-confidence workflows are REPLAYED verbatim Consistent identical answers
Tool-output caching for expensive API calls Calls the external API every time Results stored with TTL; REPLAY within TTL, re-call after Reduced external API cost
Policy-compliance agent that must verify facts before replaying Silent hallucination risk on stale data requires_verification=True ensures VERIFY fires; stale facts are never replayed silently Auditability + safety

Where Agent Memory saves real money

A GPT-4o call costs ~$0.005. A support agent handling 50,000 queries/day with 70% repeat rate:

  • Without memory: 50,000 × $0.005 = $250/day
  • With Agent Memory: 15,000 LLM calls + cache misses = $75/day
  • Saving: $175/day ($64k/year)

A REPLAY costs ~0.05ms of in-process computation. An LLM call takes 300–2,000ms and costs tokens.

Is it right for your use case?

Good fit:

  • Agent answers the same or similar questions across sessions
  • You have fact-sensitive answers that can go stale (prices, limits, policies)
  • Multiple agents or services share a knowledge base
  • You need audit trails — knowing which memory answered and why

Not the right tool:

  • Document RAG over a corpus of files → use a vector database for that
  • Replacing your application's source-of-truth database
  • Agents that never repeat similar queries

Features at a glance

Feature What it does
Decision engine Every resolve() returns REPLAY / RESTORE / VERIFY / NONE — never silent injection
Explainability decision.explain() shows per-component scores: semantic, recency, confidence, usage
Hybrid retrieval BM25 FTS5 + optional vector KNN + RRF fusion — fast and accurate
4 backends SQLite (default, zero-setup) · ChromaDB · Redis · PostgreSQL
Framework adapters Drop-in BaseMemory for LangChain and LlamaIndex
MCP server Works with Cursor, Claude Code, VS Code via Model Context Protocol
REST API FastAPI server with 9 endpoints + Swagger UI
Dashboard Streamlit UI — stats, memory browser, live resolve sandbox
Multi-agent SHARED / NAMESPACED / ISOLATED memory across multiple agents
Confidence learning Event-driven confidence updates + half-life temporal decay
Memory graph Relationship edges, path-finding, clusters, PageRank importance
Async API aremember, aresolve, alist, … — all operations have async counterparts
TTL & states Automatic expiry, archiving, near-duplicate consolidation

→ Full feature reference: docs/features.md


Performance

All numbers are measured — no projections. Charts generated from real benchmark runs.

Latency at scale (diverse unique content, no LRU cache)

resolve() latency vs store size

Store size p50 p95 p99 Notes
Any size (cache hit) 0.007ms 0.010ms — LRU cache, 60–80% of production queries
500 – 100K 9–19ms 35–84ms 70–136ms Diverse unique content, raw SQLite
1M (template-repeated) 130ms 310ms 385ms Worst case: 32K copies/template → 32K FTS5 matches

Key insight: latency scales with match count per query, not total store size. A 1M-entry store with diverse unique memories performs near the 10K numbers.

Tuning levers (all measured — shipped by default)

Pareto frontier: latency vs implementation effort

Applied by default Impact
LRU cache (5s TTL, 256 entries) 10ms → 0.007ms for repeated queries
Bloom filter (NONE fast-path) 0.46ms → 0.010ms at keyword_search level
Stop-word FTS5 filter 12.4ms → 4.3ms — stops "how/do/i/my" from matching 80% of corpus
touch() no commit -9ms per REPLAY — WAL durable without fsync
PRAGMA cache_size=32MB + mmap -4ms vs default 2MB cache
_RRFBucket at module level -0.35ms/call — was recreated inside fuse() each call
Dynamic IDF stop words (≥5K docs) Filters corpus-saturated terms automatically

Points on the Pareto frontier above cannot improve latency without increasing implementation effort. LRU cache and Bloom filter are on the frontier — they ship by default.

Seeding throughput

Mode 10K 100K 1M
Standard (per-row commit) ~92s ~909s ~2.5h
Fast-seed (--fast-seed) 3s 17s 25s

→ Full methodology, charts, and tuning guide: docs/stress-testing.md


When to use it

Use Agent Memory when:

  • You want an agent to remember past interactions without injecting all of them into every prompt
  • You need explicit control over when memory is used (replay exact answers vs inject as context vs verify first)
  • You have different memory trust levels (user preferences vs potentially-stale facts vs tool outputs)
  • Multiple processes, services, or agents share the same memory store
  • You need audit trails — every replay is traceable to a specific stored entry with a score breakdown

Don't use it for:

  • Document RAG (search over a corpus of files) — use a vector database for that; Agent Memory stores query→answer experiences
  • A replacement for your database — it stores transient agent knowledge, not your application's source-of-truth data

Quick Start

pip install agent-memory-sdk
from agent_memory import Memory, MemoryAction

memory = Memory(persist_dir=".agent_memory")

# Store once after a good answer
memory.remember(
    "How do I reset my password?",
    "Go to Settings → Security → Reset Password.",
    type="conversation", tags=["auth"],
)

# Decide before every LLM call
decision = memory.resolve("How do I reset my password?")

if decision.action == MemoryAction.REPLAY:
    return decision.response          # exact match — no LLM call needed

if decision.action == MemoryAction.RESTORE:
    context = memory.format_restore_context(decision)
    return call_llm(query, system_extra=context)

# VERIFY or NONE — validate or answer fresh

→ Full integration pattern and API reference: docs/usage.md


Local Setup

pip install agent-memory-sdk

# Store something
agent-memory remember "How do I reset my password?" \
  "Go to Settings → Security → Reset Password." \
  --type conversation --tags auth,faq

# Ask it back
agent-memory resolve "I forgot my password"
# ✅ REPLAY  confidence: 0.87
# matched: "How do I reset my password?"  stored: 2026-01-01  reused 1×
# response: Go to Settings → Security → Reset Password.

# See what's stored
agent-memory stats

Option 2 — Redis or Postgres backend

# Spin up the services
docker compose -f docker-compose.dev.yml up -d

# Install the backend extra
pip install "agent-memory-sdk[redis]"      # or [postgres]

# Use it
agent-memory --backend redis remember "API limit" "1000 req/min" --type fact
agent-memory --backend redis resolve "What is the rate limit?"

Option 3 — Streamlit dashboard (visual exploration)

pip install "agent-memory-sdk[dashboard]"

# Seed demo data (optional)
python scripts/seed_demo.py --data-dir .agent_memory

# Open the dashboard
AGENT_MEMORY_DIR=.agent_memory agent-memory-dashboard
# → http://localhost:8501

Option 4 — Development / from source

git clone https://github.com/TheProdSDE/agent-memory-sdk.git
cd agent-memory-sdk
python -m venv .venv && source .venv/bin/activate
pip install -e ".[dev]"
make test          # run all tests
make check         # lint + type check

Dashboard

An interactive Streamlit dashboard for exploring memories, testing the resolve sandbox, and monitoring stats.

Agent Memory dashboard slideshow: stats, memory table, replay/verify/none resolve results

Stats — KPIs + charts Memories — searchable table
Stats tab: 31 total, donut chart by state, bar chart by type Memories tab: 29 rows with type, scope, confidence, access count
Resolve → REPLAY Resolve → VERIFY
REPLAY badge, confidence 0.88, full response shown VERIFY badge, context entry with fact response
pip install "agent-memory-sdk[dashboard]"
AGENT_MEMORY_DIR=.agent_memory agent-memory-dashboard   # → http://localhost:8501

# Seed demo data (optional — run only when you want it)
python scripts/seed_demo.py --data-dir .agent_memory

Integrations

agent-memory-sdk is the core — every integration delegates to Memory.

Integration Install extra Example
Core SDK (SQLite) (none) basic_usage.py
LangChain BaseMemory [langchain] langchain_integration.py
LlamaIndex BaseMemory [llamaindex] llamaindex_integration.py
Redis backend [redis] redis_backend.py
PostgreSQL backend [postgres] postgres_backend.py
Multi-agent isolation (none) multi_agent.py
FastAPI REST server [api] rest_api.py
Confidence + Graph (none) confidence_and_graph.py
Benchmark harness (none) benchmark_harness.py

→ Setup instructions and code snippets for each: examples/README.md


Tech Stack

Component Technology
Language Python 3.10+
Storage SQLite · ChromaDB · Redis · PostgreSQL
Retrieval BM25 FTS5 + Vector KNN + RRF fusion
Interfaces MCP · FastAPI · Streamlit · CLI
Adapters LangChain BaseMemory · LlamaIndex BaseMemory
Search DSA Bloom filter (NONE fast-path) · Dynamic IDF stop words · RRF fusion
Testing pytest (270 tests) · ruff · mypy
CI/CD GitHub Actions — test matrix 3.10–3.13 → release gate → PyPI

No API keys required — everything runs locally.


Documentation

Doc Contents
docs/usage.md Integration pattern, API reference, MemoryEntry / MemoryDecision fields
docs/features.md Decision actions, hybrid retrieval, types, scopes, TTL, graph, multi-agent
docs/mcp.md MCP server setup for Cursor, Claude Code, VS Code; Docker config
docs/cli.md CLI commands, REST API server, dashboard launch, eval dataset format
docs/roadmap.md All shipped features, what's next, GitHub Project board
docs/release.md CI-automated release process, versioning, rollback
docs/architecture.md Retrieval pipeline, scoring policy, system design
docs/comparison.md Feature matrix vs Redis, mem0, Zep, LangMem, LlamaIndex, MemGPT
docs/stress-testing.md 10K / 100K / 1M latency benchmarks with methodology
docs/benchmarks.md Eval results and reproduce commands
docs/why-decision-layer.md The failure mode this project exists to fix
examples/README.md Index of all runnable examples
CONTRIBUTING.md Dev setup, test commands, PR checklist

Status & Roadmap

All planned features through v0.5.0 are shipped. Track what's next on the GitHub Project →

→ docs/roadmap.md

Release

Tag-triggered, fully CI-gated: git tag v0.x.y && git push origin v0.x.y

→ docs/release.md


Contributing

See CONTRIBUTING.md for dev setup, test commands, and the PR checklist.


License

MIT — see LICENSE.


Support


Agent Memory helps agents decide: Replay → Restore → Verify → Ignore

Built with ❤️ by TheProdSDE

mcp-name: io.github.theprodsde/agent-memory

Metadata

Release files for agent-memory-sdk 0.1.5

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for agent-memory-sdk 0.1.5
File Size Uploaded Provenance
agent_memory_sdk-0.1.5.tar.gz 3.9 MB Without provenance Details

Built distribution (wheel)

Table of built distributions (wheels) for agent-memory-sdk 0.1.5
File Interpreter ABI Platform
agent_memory_sdk-0.1.5-py3-none-any.whl Python 3 none any Details

Total release size: 4.0 MB

Release files / agent_memory_sdk-0.1.5.tar.gz

Download URL agent_memory_sdk-0.1.5.tar.gz
Size 3.9 MB
Tags Source
SHA-256 checksum
How to use checksums
f78a921b3746093226e510c06a54a0aae4e6c0a444d3791d1361b556c87160f1
BLAKE2b-256 checksum
How to use checksums
16711851146e70398b01e5a09693f32094e6274d69597f60d627e7c1b4490452
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/7.0.0 CPython/3.11.16

Release files / agent_memory_sdk-0.1.5-py3-none-any.whl

Download URL agent_memory_sdk-0.1.5-py3-none-any.whl
Size 106.4 kB
Tags Python 3
SHA-256 checksum
How to use checksums
55bc0de98671b7297bc43f0c159855387487b183ac022b953925bf109cf326ca
BLAKE2b-256 checksum
How to use checksums
89055101c421f3c03ac2b0ce790142e3940a74c9858dcb37b4bbfd2a35a14514
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/7.0.0 CPython/3.11.16

Security for agent-memory-sdk 0.1.5

Provenance

Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.

No provenance found

None of the files in this release were uploaded with provenance metadata.

A note on package security

PyPI is a repository for distributing packages, not a service for auditing them. While third-party security researchers and our staff do their best to detect and remove malicious packages, PyPI cannot guarantee the quality, security or suitability of any given package.

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