Skip to main content

Persistent, fast, accurate memory layer for conversational LLM agents

Project description

memory-verse-avneesh

A persistent memory layer for a conversational LLM agent — built to make responses feel personalized and consistent across sessions, without adding noticeable latency and without ever confidently telling the model something false or stale about the user.

This is a from-scratch rebuild. The previous implementation is gone; this document is the plan the rebuild follows.

This project is built and distributed as an installable Python library published on PyPI (pip install memory-verse-avneesh), not as a standalone service — the third package in the -verse-avneesh family, alongside storage-verse-avneesh and llm-verse-avneesh. A host application (FastAPI, Flask, a CLI, whatever) imports it and calls it directly. This constraint shapes several decisions below: storage and LLM backends must be pluggable rather than hardcoded, the formation worker must be something the host process runs rather than something the library owns, and the public API surface has to be small and stable since other people's code will depend on it.

Design synthesized from production/research systems: Mem0 (extraction + ADD/UPDATE/DELETE/NOOP pipeline), Zep/Graphiti (bi-temporal knowledge graph), Letta/MemGPT (tiered, OS-inspired memory), and Stanford's Generative Agents (reflection / memory synthesis).


1. Problem statement

Given (user_id, new_message), produce a response that reflects everything worth knowing about this user from past interactions — without the user waiting for that "remembering" to happen, and without the system ever holding two conflicting "truths" about the user at once.

Two things matter equally: speed (the user is waiting) and accuracy (a wrong or stale memory actively makes the agent worse, not neutral).

2. Core design principle

Reading memory and forming memory are different problems with different cost budgets, and must never share a code path.

  • Read path — runs between "user hits send" and "model starts responding." Hard latency budget. No LLM reasoning about what to retrieve — only cache reads, index lookups, and arithmetic scoring.
  • Formation path — runs after the response has already been sent. No latency budget. This is where all the expensive reasoning (contradiction resolution, confidence judgment, deduplication) is allowed to happen, because nobody is waiting on it.

Two independent services connected by a durable queue, not one pipeline with async bits bolted on.

A third boundary, specific to this being a library rather than a service: the read path itself stops at retrieving memory — it does not generate the user-facing response. The library hands back structured memory context; the host application makes its own generation call (its own model, tools, streaming, provider) and, once it has a response, builds the Turn and hands it to the formation path itself. The library never calls an LLM to produce a response a user sees — only to retrieve (embeddings) or to reason about what's true (formation extraction/classification).

3. Memory tiers

Tier Contents Storage Access pattern
Tier 0 — Session Rolling recent turns, active task state Redis O(1) read, per-conversation key
Tier 1 — Core profile Small, precomputed, always-injected user profile Redis (backed by Postgres) O(1) read, whole blob, no search
Tier 2 — Active store Extracted facts (vector) + entities/relationships (bi-temporal graph) + keyword index Postgres (pgvector + edges table) Parallel vector / graph / keyword query
Tier 3 — Reflections Higher-level patterns synthesized from clusters of Tier 2 facts Postgres Retrieved like any other memory
Archival Decayed-out, low-relevance, or old memory Postgres (cold) Never on the hot path; audit/debug only

Tier 2 is deliberately one logical store with two representations of the same facts, not two separate subsystems:

  • Vector (Mem0-style): flat facts + embeddings, for fuzzy semantic recall ("what did we discuss about pricing").
  • Graph (Zep-style): entities + relationships as bi-temporal edges(source, relation, target, valid_from, valid_to, observed_at, recorded_at). Contradictions never delete a row: a new fact closes the old edge's valid_to and inserts a new edge. "Current truth" is just valid_to IS NULL. Full history is preserved for free.
  • Keyword/BM25 over the same store, run in parallel with the other two — catches exact names/IDs that embeddings sometimes miss.

Episodic memory is also separate from the Tier 2 table above — a durable, embedded record of every turn (episodes table), not just the distilled facts extracted from it. Written unconditionally by write_memory() for every turn, with no LLM judgment about what's "worth remembering" (that's what fact extraction already does; episodic memory's value is completeness — an actual answer to "what happened, when," not just "what do I know about the user"). Searched the same way as Tier 2 facts (embedding + ANN + rerank), sharing the same query embedding and retrieval gate, but reranked by relevance + recency only (no confidence/type weighting — an episode doesn't have those). Immutable except for explicit user-requested deletion; no decay, since it's the audit trail Tier 2 facts get distilled from, not a duplicate of Tier 2 itself.

Identity is a separate concept from the tiers above, not a tier itself — two distinct Postgres tables, neither written by the formation pipeline:

  • Expert identity: host-authored personas ("expert_email_writer"), keyed by a string id the host chooses, full CRUD via memory_verse_avneesh.identity. Selected explicitly per read_memory() call via identity_id — never auto-selected.
  • Person identity: one durable record per user_id, distinct from the Tier 1 profile cache (that's an ephemeral, formation-derived blob; this is a deliberate record the host writes), always fetched automatically by read_memory() when an IdentityStore is configured.

Both surface on MemoryContext as expert_identity / person_identity — combined together when both are present, e.g. an "expert email writer" persona applied with a specific person's own tone preferences layered on top.

Prospective memory (reminders table) is future intentions, not facts about the past or present — "do X later." Created explicitly through memory_verse_avneesh.prospective, by the host's own code or its own LLM calling it as a tool during generation; write_memory() never creates one automatically, there is no "this sounds like something to remind them about" extraction in this version. read_memory() always includes PENDING reminders with due_at <= now on MemoryContext.due_reminders when a ReminderStore is configured — a plain deterministic time comparison, not a similarity search, so it's included regardless of the current message's content (even a trivial "thanks!" still surfaces a due reminder). A reminder stays PENDING — and keeps being returned — until explicitly marked done or dismissed; passing due_at doesn't silently remove it.

4. Request-time workflow (read path)

Steps 1–4 are the library's job (read_memory()) — cache, index, or arithmetic only, never an LLM call deciding what to fetch. Steps 5–6 are the host application's own code, built on what the library returns; the library does not do them.

  1. Retrieval gate (heuristic, not LLM): skip Tier 2 (the embedding call + vector search) for turns that obviously don't need durable memory ("ok", "thanks") — Tier 0/1 are always read regardless, deliberately: they're O(1) cache reads, and dropping them on a one-word reply would break conversational continuity for no real speed win. Tier 2 is the part actually worth skipping.
  2. Parallel fetch: Tier 0 + Tier 1 reads, plus (gate permitting) one query embedding computed once and reused across all three Tier 2 channels (vector, graph, keyword) — fired concurrently, never in a sequential loop.
  3. Two-stage funnel: fast approximate fetch (ANN top-20 via HNSW) → deterministic rerank: score = w1·relevance + w2·recency_decay + w3·importance + w4·type_weight.
  4. Return structured context (MemoryContext: profile + ranked facts + recent turns + ranked episodes + identity + due reminders), packed to a token budget (facts and episodes have independent budgets; due reminders aren't budget-packed, since they're a plain time filter, not a ranked/truncated list). An optional convenience can flatten this to text, but the structured form is the real contract — the library's responsibility ends here.

— host-owned, outside the library —

  1. Generate: the host builds its own prompt/messages from the returned context (its own system prompt, tools, streaming, model, provider) and makes its own generation call.
  2. Persist + hand off: once the host has its own response, it constructs the Turn (it has both messages now), calls SessionCache.append_turn(), and pushes the turn to formation — fire-and-forget, so it never blocks the response already returned to the user.

Floor cost through step 4 (the part the library is responsible for): 2 cache reads (parallel) + 1 embedding + 3 parallel index lookups + 1 rerank pass. This is the speed ceiling the library controls; generation latency (step 5) is the host's own model choice, not the library's to own or optimize.

5. Formation workflow (write path, async)

Consumes turn-completed events from the durable queue, one turn at a time, per user. Each Turn was constructed by the host application (README Section 4, step 6) after its own generation call — the library only ever sees a turn once both messages already exist.

  1. Extract: one structured-output LLM call → typed candidates (fact / relation / preference), each with a confidence score and an explicit-vs-inferred flag.
  2. Resolve: for each candidate, retrieve its top-k nearest existing memories (same retrieval mechanism as the read path, reused).
  3. Classify operation: one LLM tool-call decides ADD / UPDATE / DELETE / NOOP against those candidates (Mem0's mechanism).
  4. Safety gate (deterministic, not LLM): identity- and constraint-class fields must additionally pass an explicit-statement-or-N-repetitions check regardless of step 3's decision. Prevents one bad extraction from silently overwriting who the user is.
  5. Write:
    • Relational fact → bi-temporal edge (see Tier 2 above).
    • Flat fact → vector row, with confidence + observation_count. Merge into an existing row above 0.85 cosine similarity instead of inserting a duplicate.
  6. Reflection (batched — e.g. hourly per active user, never per-turn): cluster recent writes, synthesize a Tier 3 summary where a pattern has emerged across ≥N observations.
  7. Decay sweep (batched — e.g. daily): old, unreinforced, unretrieved Tier 2 rows move to Archival. This keeps the active HNSW index small, which is what keeps step 3 of the read path fast as the system ages — decay and speed are the same mechanism.

6. Storage

  • Postgres: episodes (raw, append-only, embedded, source of truth — episodic memory, see Section 3) · memory_facts (Tier 2 vector rows) · memory_edges (Tier 2 bi-temporal graph) · reflections (Tier 3) · archival_* (cold copies) · expert_identities / person_identities (identity, see Section 3) · reminders (prospective memory, see Section 3 — no embedding column, indexed on (user_id, status, due_at) for the due-reminders lookup) — all under one required, host-chosen schema (MemoryConfig.postgres_schema), never a default public. pgvector + HNSW index for vector search (used by both memory_facts and episodes). Plain indexed edges table with recursive CTEs for 1–2 hop graph queries — no separate graph database at this scale.
  • Redis: Tier 0 session cache, Tier 1 profile cache, durable job stream (Redis Streams) feeding the formation worker pool.
  • Formation workers: a separate deployable from the API, scaled independently, so a restart never silently drops queued learning work.

7. Non-negotiables

  • User-facing visibility/control: view, edit, delete stored memories. Both ChatGPT and Claude treat this as core product surface, not an afterthought — it also doubles as the primary debugging tool during development.
  • Observability: structured logs (not print) and a full memory-operation audit trail — every ADD/UPDATE/DELETE/NOOP and every edge invalidation logged with its reasoning. This is the only way to see the system's judgment after the fact, since none of it is visible in the final response.
  • Per-user isolation: all storage and queue partitioning keyed by user_id, so one user's write load never contends with another's reads.

8. Build order

Do not build all tiers at once. Per production precedent (Mem0/Zep's own staged rollouts):

Phase 1 (MVP)

  • Tier 0 (session cache) + Tier 1 (core profile)
  • Tier 2, vector half only (flat facts + embeddings, no graph yet)
  • Formation pipeline: extract → resolve → ADD/UPDATE/DELETE/NOOP → safety gate
  • Basic decay sweep
  • User-facing memory view/edit/delete

This alone should deliver the large majority of the latency and accuracy win.

Phase 2

  • Tier 2 graph half (bi-temporal edges) + keyword/BM25 channel
  • Tier 3 reflections
  • Full observability/audit trail

Graduate to Phase 2 only once real usage data from Phase 1 shows where flat-vector retrieval is actually falling short — not speculatively upfront.

9. Packaging: distributed as a PyPI library

Repo layout — src layout (standard for publishable packages, avoids accidentally testing against the working directory instead of the installed package):

memory-verse-avneesh/                        (repo root)
├── pyproject.toml                   (PEP 621 metadata, build backend, optional-dependencies)
├── README.md
├── LICENSE
├── src/
│   └── memory_verse_avneesh/                (importable package — the actual library)
│       ├── __init__.py              (package metadata; the callables live in read/, formation/, management.py)
│       ├── py.typed                 (marks the package as type-hinted for downstream users)
│       ├── config.py                (settings/config objects, no global state)
│       ├── read/                    (read-path: Section 4)
│       │   ├── gate.py
│       │   ├── session_cache.py
│       │   ├── profile_cache.py
│       │   ├── retrieval.py
│       │   └── rerank.py
│       ├── formation/               (write-path: Section 5)
│       │   ├── extract.py
│       │   ├── resolve.py
│       │   ├── operations.py        (ADD/UPDATE/DELETE/NOOP)
│       │   ├── safety_gate.py
│       │   ├── reflection.py
│       │   ├── decay.py
│       │   └── worker.py            (exposes run_formation_worker() — host process runs this)
│       ├── storage/
│       │   ├── interfaces.py        (abstract backend protocols)
│       │   ├── postgres/            (facts, edges, reflections, migrations)
│       │   └── redis/               (session cache, profile cache, job stream)
│       ├── llm/
│       │   └── interfaces.py        (provider-agnostic LLM + embedding client protocols)
│       └── models/                  (shared pydantic schemas)
├── tests/
│   ├── unit/
│   └── integration/
└── examples/
    └── fastapi_app/                 (reference integration: how a host app wires this in)

Packaging decisions this implies:

  • Storage backends are pluggable via interfaces (storage/interfaces.py), with Postgres + Redis shipped as the default implementations — a library consumer isn't forced onto our exact infra choices, though those remain the recommended default.
  • LLM/embedding providers are pluggable the same way (llm/interfaces.py) — AWS Bedrock, OpenAI, Anthropic, or a local embedding model can all satisfy the same protocol. No hardcoded provider inside the core package.
  • The formation worker is exposed, not owned. The library provides run_formation_worker(); the host application decides whether to run it as an in-process asyncio task (simple deployments) or as a separate process/service (Phase 1 build-order default per Section 8) — the library doesn't assume either.
  • Optional extras in pyproject.toml so installing the library doesn't force every dependency: e.g. pip install memory-verse-avneesh[postgres,redis,bedrock].
  • Semantic versioning from the first published release, since a public API surface means breaking changes have real downstream cost.
  • examples/fastapi_app is a reference/demo of integrating the library into a service — it is not part of the published package.

10. Open decisions (to confirm before/while building)

  • LLM provider and model for extraction and operation-classification calls. (Generation is host-owned, not a library decision — see Section 4.)
  • Embedding model: local (e.g. sentence-transformers, in-process) vs. hosted API — local avoids a network hop on the one embedding call that sits on the critical path.
  • Deployment target for the formation worker pool (separate process vs. separate service).
  • PyPI package name — decided: memory-verse-avneesh (repo renamed to match), import name memory_verse_avneesh. Third package in the -verse-avneesh family alongside storage-verse-avneesh and llm-verse-avneesh.
  • Minimum supported Python version and how far back to support (affects typing syntax, async features available).

Project details


Download files

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

Source Distribution

memory_verse_avneesh-0.5.0.tar.gz (51.6 kB view details)

Uploaded Source

Built Distribution

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

memory_verse_avneesh-0.5.0-py3-none-any.whl (56.0 kB view details)

Uploaded Python 3

File details

Details for the file memory_verse_avneesh-0.5.0.tar.gz.

File metadata

  • Download URL: memory_verse_avneesh-0.5.0.tar.gz
  • Upload date:
  • Size: 51.6 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/7.0.0 CPython/3.13.14

File hashes

Hashes for memory_verse_avneesh-0.5.0.tar.gz
Algorithm Hash digest
SHA256 a11ab2ceaa9ddea6b04fe9a948a359201f83695365447b4026e66695b1733c6a
MD5 7133925d631c5f3101e9ee57c24fd232
BLAKE2b-256 6021a95e40c5b1ec0deeffc9a6d199b48981107e548103b945e4caa2a6628e34

See more details on using hashes here.

Provenance

The following attestation bundles were made for memory_verse_avneesh-0.5.0.tar.gz:

Publisher: publish.yml on avneeshrai07/memory-verse-avneesh

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

File details

Details for the file memory_verse_avneesh-0.5.0-py3-none-any.whl.

File metadata

File hashes

Hashes for memory_verse_avneesh-0.5.0-py3-none-any.whl
Algorithm Hash digest
SHA256 91100e1ae935a9c9d9f7a43b4ac7c18817b609125d76331609489a9db1848a99
MD5 2b0f6b3097e7346e027d409fd541f994
BLAKE2b-256 e1b942c60e46412041a13692f87019317be44a0c94124377a1c3a06f23d9d6c8

See more details on using hashes here.

Provenance

The following attestation bundles were made for memory_verse_avneesh-0.5.0-py3-none-any.whl:

Publisher: publish.yml on avneeshrai07/memory-verse-avneesh

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page