Redis agent memory with redis-rs

Build a Redis-backed agent memory layer in Rust with redis-rs, Candle, and standard Redis commands — working memory in a Hash, long-term semantic recall as JSON with a vector index, and an event log in a Stream.

This guide shows you how to build a small Redis-backed agent memory layer in Rust with redis-rs and the Candle inference framework, using only standard Redis commands — no agent-memory SDK, no managed service. It includes a local web server built with tiny_http so you can send turns at the agent, watch working memory update in place, see semantically similar long-term memories recalled in real time, watch the write-time deduplication skip near-duplicates, and inspect the per-thread event log.

The embedder is Candle running the canonical sentence-transformers/all-MiniLM-L6-v2 PyTorch checkpoint — the same weights the Python example loads. Candle reads the .bin weights directly, so the vectors produced here are bit-identical to the Python ones, and the distance bands the Python walkthrough quotes carry over to this demo without recalibration. A memory written by one demo can be recalled by the other against the same Redis instance.

Overview

The memory layer splits across three Redis primitives, each handling one tier:

• Working memory for the active session is a Hash at agent:session:<thread_id> holding the goal, scratchpad, a rolling window of recent turns (as a JSON list inside one field), and a few audit timestamps. One HGETALL returns the whole session in a single round trip; every write refreshes the key's EXPIRE so idle sessions decay on their own.
• Long-term memory is a set of JSON documents at agent:mem:<id>, each carrying the memory text, a 384-dimensional embedding vector, and tag fields for user, namespace, kind (episodic / semantic), and source thread. A single Redis Search index covers the HNSW vector field and every metadata field, so one FT.SEARCH call performs the KNN with the metadata pre-filter in the same round trip. Write-time deduplication runs the same KNN at insert time and skips a new memory whose nearest existing entry is within a tighter threshold.
• Event log for the agent's actions and observations is a Stream at agent:events:<thread_id>, appended with XADD MAXLEN ~ so retention stays bounded automatically, replayed with XREVRANGE.

That gives you:

• A single round trip per tier: one HGETALL for the session, one FT.SEARCH for recall, one XADD for the event log.
• Sub-millisecond reads on every step of the agent loop, so the memory layer doesn't dominate per-step latency.
• Per-tier decay: short TTLs on working memory, longer on episodic memories, no TTL on semantic memories. Combined with a database-level eviction policy (LFU is the common choice), memory stays bounded under pressure.
• Scoping enforced inside the query: a recall query for user=alice will never see user=bob's memories, because the TAG filter goes into the same FT.SEARCH call as the KNN.

How it works

Each turn through the agent loop touches all three tiers in one pass: append to working memory, recall similar long-term memories, write the turn back as a new memory (with deduplication), and append one event to the log.

Per-turn flow

1. The application calls embedder.encode_one(text) to turn the incoming turn into a 384-dimensional Vec<f32>.
2. session.append_turn(thread_id, role, content, Some(user), ..., None) reads the per-thread Hash with HGETALL, appends the new turn to the rolling window in application code, trims it back to the configured maximum, and writes the Hash back with an HSET + EXPIRE pipeline inside a MULTI / EXEC. The session TTL refreshes on every write so an active thread stays alive.
3. memory.recall(&vec, user, Some(namespace), None, 5, None) runs FT.SEARCH with a TAG pre-filter and a KNN 5 clause. Redis returns the closest matching memories together with their cosine distances; memories beyond the recall threshold are dropped before they reach the agent so an unrelated query doesn't surface confident-looking false positives.
4. memory.remember(text, &vec, user, namespace, kind, source_thread, None) runs the same KNN with a tighter dedup threshold. If an existing memory is within the threshold, the new write is skipped and the existing memory's hit_count is incremented with JSON.NUMINCRBY; otherwise a fresh JSON document is written with JSON.SET and a per-kind EXPIRE — episodic defaults to seven days, semantic has no TTL by default.
5. events.record(thread_id, action, detail) appends one entry to the per-thread Stream with XADD MAXLEN ~, bounding retention to roughly a thousand entries per thread without an explicit cleanup job.

The embedding is computed once and reused for steps 3 and 4 — there's no point encoding the same text twice. Recall runs before the write, so the agent doesn't see its own just-written turn echoed back as a recalled memory.

The session store

AgentSession wraps the working-memory Hash and the rolling turn window (source):

use agent_memory_demo::session_store::AgentSession;

let client = redis::Client::open("redis://127.0.0.1/")?;
let session = AgentSession::new(
    &client,
    "agent:session:",
    3600,  // one hour TTL on the working-memory hash
    20,    // rolling window per thread
)?;

let thread_id = session.new_thread_id();
session.start(&thread_id, "alice", "demo-agent",
    "Plan next week's meetings.", None)?;
session.append_turn(
    &thread_id, "user",
    "Schedule a budget review with finance.",
    Some("alice"), Some("demo-agent"), None,
)?;
let state = session.load(&thread_id)?;
println!("{} {} {}",
    state.as_ref().map(|s| s.turn_count).unwrap_or(0),
    state.as_ref().map(|s| s.recent_turns.len()).unwrap_or(0),
    state.as_ref().map(|s| s.ttl_seconds).unwrap_or(0));

The data model is one Hash per thread. The rolling turn window is stored as a JSON string in a single field so the whole session loads in one HGETALL — the hash never grows in size or field count as the conversation goes on.

agent:session:9f3d2a4b8c61
  thread_id=9f3d2a4b8c61
  user=alice
  agent=demo-agent
  goal=Plan next week's meetings.
  scratchpad=Need to confirm finance's availability.
  turn_count=4
  created_ts=1715990400.12
  last_active_ts=1715990650.83
  recent_turns=[{"role":"user","content":"...","ts":...}, ...]

Every write — start, append_turn, set_goal, set_scratchpad — runs the HSET and EXPIRE inside a redis::pipe().atomic() block so a connection drop between the two writes can't leave the session without a TTL.

The long-term memory store

LongTermMemory owns the JSON documents, the vector index, the recall query, and the write-time deduplication (source):

use agent_memory_demo::embeddings::LocalEmbedder;
use agent_memory_demo::long_term_memory::LongTermMemory;

let memory = LongTermMemory::new(
    &client,
    "agentmem:idx",
    "agent:mem:",
    384,    // vector dimension
    0.20,   // dedup threshold — tight at write time
    0.55,   // recall threshold — looser at read time
)?;
let embedder = LocalEmbedder::new(None)?;
memory.create_index()?;  // idempotent

// Write a memory. The same KNN that powers recall also runs here
// at a tighter threshold so paraphrases of the same fact collapse.
let vec = embedder.encode_one("The user prefers light mode in editors.")?;
let result = memory.remember(
    "The user prefers light mode in editors.",
    &vec,
    "alice",
    "default",
    "semantic",
    "9f3d2a4b8c61",
    None,
)?;
println!("deduped={} id={} dist={:?}",
    result.deduped, result.id, result.existing_distance);

// Recall against a later question.
let q = embedder.encode_one("Which theme does this user like?")?;
let hits = memory.recall(&q, "alice", Some("default"), None, 5, None)?;
for h in &hits {
    println!("{:.3} [{}] {}",
        h.distance.unwrap_or(0.0), h.kind, h.text);
}

Data model

Each memory is a JSON document at agent:mem:<id>. The embedding is stored as a JSON array of floats so the document is human-readable from redis-cli; FT.SEARCH still expects the query vector as raw float32 bytes (byteorder::LittleEndian via embeddings::floats_to_bytes), regardless of how the indexed document stores it.

agent:mem:7c3f8a1b9e02
{
  "id": "7c3f8a1b9e02",
  "user": "alice",
  "namespace": "default",
  "kind": "semantic",
  "source_thread": "9f3d2a4b8c61",
  "text": "The user prefers light mode in editors.",
  "embedding": [0.013, -0.041, ...],
  "created_ts": 1715990400.12,
  "hit_count": 0
}

The Redis Search index is declared on the JSON document type with AS aliases so the query syntax stays compact:

FT.CREATE agentmem:idx
  ON JSON PREFIX 1 agent:mem:
  SCHEMA
    $.text          AS text          TEXT
    $.user          AS user          TAG
    $.namespace     AS namespace     TAG
    $.kind          AS kind          TAG
    $.source_thread AS source_thread TAG
    $.created_ts    AS created_ts    NUMERIC SORTABLE
    $.hit_count     AS hit_count     NUMERIC SORTABLE
    $.embedding     AS embedding     VECTOR HNSW 6
                                       TYPE FLOAT32 DIM 384
                                       DISTANCE_METRIC COSINE

The query

Both recall and dedup share the same hybrid query: a TAG pre-filter in parentheses followed by =>[KNN k @embedding $vec]. With DIALECT 2, Redis applies the filter first and KNN-ranks only the matching documents.

FT.SEARCH agentmem:idx
  "(@user:{alice} @namespace:{default} @kind:{semantic})
     =>[KNN 5 @embedding $vec AS distance]"
  PARAMS 2 vec <384-float32-bytes>
  SORTBY distance
  RETURN 8 user namespace kind source_thread text created_ts hit_count distance
  DIALECT 2

distance is the cosine distance (0 means identical, 2 means opposite). Recall and dedup share the same query shape; only the threshold differs — strict at write time so the index doesn't fill with paraphrases of the same fact, looser at read time so the agent gets a wider net of relevant memories.

Per-kind TTLs

remember resolves the entry's TTL from the memory's kind:

You can override per write by passing Some(ttl) as the last argument to remember. The defaults live in long_term_memory::default_ttl_for_kind; swap it out if you want a different tier map.

The event log

AgentEventLog is a thin wrapper over a per-thread Redis Stream (source):

use agent_memory_demo::event_log::AgentEventLog;

let events = AgentEventLog::new(&client, "agent:events:", 1000)?;
events.record(&thread_id, "turn_appended:user",
    "Schedule a budget review with finance.")?;
events.record(&thread_id, "memory_written",
    "wrote 7c3f8a1b9e02 as semantic")?;

for e in events.recent(&thread_id, 20)? {
    println!("{} {}", e.action, e.detail);
}

record calls XADD with MAXLEN ~ 1000. The tilde lets Redis trim in whole-node units instead of exactly-N units, which is much cheaper at the cost of overshooting the bound by up to a node's worth — the right tradeoff for an audit log where exact length doesn't matter.

The Stream is independent of the session Hash and the long-term JSON documents: it answers "what just happened" without competing with either of those for indexing or memory budget. Consumer groups (not used in this demo) would let downstream workers — summarisers, consolidators, audit pipelines — replay the log without losing position.

Concurrency caveats

The three helpers above trade correctness under heavy concurrency for clarity. Each is fine on a single-process demo, but lifting the code into a real multi-worker agent surfaces three races worth knowing about:

• Working memory is read-modify-write. AgentSession::append_turn calls HGETALL, mutates the recent_turns vector in application code, and writes the Hash back with HSET. Two concurrent turns on the same thread can both read the same recent_turns, append different entries, and write back — last writer wins, the other turn is silently lost. The robust fix is either a WATCH / MULTI / EXEC loop around the read-modify-write or a small Lua script that does the append atomically server-side. (Each helper holds a Mutex<Connection> that protects individual Redis commands against TCP-frame interleaving, but the lock is released between the HGETALL and the HSET, so it doesn't make the read-modify-write itself atomic.)

• Long-term dedup is not atomic. LongTermMemory::remember runs a FT.SEARCH KNN lookup, decides whether the candidate is a duplicate, and (if not) calls JSON.SET. Two workers seeing the same fact in flight can each fail to see the other's not-yet-committed write and both insert a new memory. The pragmatic fix is to accept that the index will occasionally hold near-duplicates and run a background consolidator that periodically scans for memory pairs within a tight distance and merges them, rather than trying to make the write itself atomic.

• The active thread is server state. The demo server keeps a single current_thread_id (a Mutex<String>) that /new_thread and /reset mutate. handle_turn reads it under the mutex but then drops the lock immediately, so a turn racing with a thread rotation can apply to the previous thread. This is cosmetic for a one-user browser demo. A multi-user agent would carry the thread id on the request itself rather than as shared server state.

Those caveats are deliberate. A more conservative implementation would obscure the Redis-shaped parts of the pattern; the demo prioritizes a small, readable code path that maps directly onto the commands in the prose above.

Pre-seeding long-term memory

In a real deployment the memory store fills up organically as the agent reasons over user turns: each turn produces zero or more memories that flow into the store, with deduplication catching repeats. For the demo, src/seed_memory.rs pre-loads a small set of mixed semantic and episodic memories so the very first recall query returns something useful (source):

use agent_memory_demo::seed_memory::seed;

let memory = LongTermMemory::new(&client, "agentmem:idx", "agent:mem:",
    384, 0.20, 0.55)?;
let embedder = LocalEmbedder::new(None)?;
memory.create_index()?;
seed(&memory, &embedder, "default", "default", "seed")?;

The seed list mixes long-lived facts and preferences (semantic) with snapshots of past sessions (episodic), so the Kind to write control in the demo has something to switch between when a new turn is being remembered.

The interactive demo

src/main.rs runs a tiny_http server on port 8094, spawning a worker thread per request. The HTML page exposes three live panels — working memory, recalled memories, event log — plus a memories table for admin actions. Endpoints:

The server holds one LocalEmbedder, one AgentSession, one LongTermMemory, and one AgentEventLog for the lifetime of the process, wrapped in an Arc so the worker threads can share them without copying. The "current thread" is a Mutex<String> that the New thread button rotates — every browser tab inherits the same thread until you explicitly start a new one.

Run the demo locally

1. Clone the redis/docs repository and change into the example directory:

   git clone https://github.com/redis/docs.git
   cd docs/content/develop/use-cases/agent-memory/rust
   

2. Build the project. You'll need a recent Rust toolchain (1.75 or later):

   cargo build --release
   

   Building in release mode matters here — Candle is CPU-bound on token throughput, so the debug build is several times slower at the embedding step.

3. Make sure a Redis instance with Redis Search and Redis JSON is running locally on port 6379. Redis Stack ships both, or Redis 8 with the Search and JSON modules enabled.

4. Start the demo. The first run downloads the sentence-transformers/all-MiniLM-L6-v2 weights, tokenizer, and config from the Hugging Face Hub into the local cache (~/.cache/huggingface/hub/ by default):

   cargo run --release
   

5. Open http://localhost:8094 and try some turns:

   • "Remind me which theme I prefer in editors." — paraphrase of a seeded semantic memory ("The user dislikes dark mode and prefers a high-contrast light theme..."). You should see that memory recalled with a cosine distance around 0.47, comfortably under the 0.55 default recall threshold.
   • "What did we discuss about the order-routing outage?" — paraphrase of a seeded episodic memory; the postmortem memory should recall around 0.44. Switch the Kind to write dropdown to skip so the question itself doesn't enter long-term memory.
   • "I prefer concise answers without filler phrases." — paraphrase of a seeded semantic memory. Switch the Kind to write dropdown to semantic so the dedup KNN runs in the same kind as the seed (dedup is scoped per kind, on purpose, so an episodic write can't collapse onto a semantic memory). You should then see the write deduped onto the existing memory at a cosine distance around 0.15, with hit_count ticking up in the memories table.
   • "My favorite color is teal." — unrelated to any seed; nothing recalls above the threshold (every seed lands above 0.8), and the new memory is written as episodic (or semantic, depending on the dropdown) under a fresh id.
   • Switch the User field to bob and re-ask any of the above — recall returns nothing because the seed memories live under default. That's the TAG pre-filter at work inside FT.SEARCH.
   • Slide the Recall threshold down to 0.30 to see borderline paraphrases drop out of the recall set, then back up to 0.70 to watch them return.

   Candle reads the same PyTorch weights as the Python sentence-transformers library, so distances here match the Python demo bit-for-bit. sentence-transformers/all-MiniLM-L6-v2 puts a faithful paraphrase in the 0.15 – 0.50 cosine-distance range, a loose paraphrase or related topic in the 0.50 – 0.80 range, and unrelated queries above 0.8 — which is what motivates the 0.55 default recall threshold and the 0.20 default dedup threshold. A stricter embedding model (or a domain-tuned one) would let you tighten both; a noisier one would push them up. The right thresholds are always a function of the model, the corpus, and how conservative the agent needs to be about accepting a memory as a match.

The server is read/write against your local Redis. The default memory index is agentmem:idx, JSON keys live under agent:mem:, session Hashes under agent:session:, and event Streams under agent:events:. Useful flags (pass them after --, for example cargo run --release -- --no-reset):

• --no-reset — keep the existing long-term memories across restarts instead of dropping and re-seeding.
• --session-ttl-seconds — change the working-memory TTL (default 3600).
• --dedup-threshold — change the cosine-distance cutoff for write-time deduplication.
• --recall-threshold — change the default cosine-distance cutoff for recall.