Adds cloud Redis agent memory option (#17)

* updates references to Redis Agent Memory

* updates references to Redis Agent Memory in READMEs

* adds redis agent memory as dependency

* adds support for cloud agent memory

* updates ci to run integration tests against cloud service

* fixes async tests by changing to function scope

* passes timeout_ms in memory config

* thread config.timeout_ms into AgentMemory client

* close client on exit, read owner_id, fix topics filter

* simplifies test running

Author: justin-cechmanek

.github/workflows/ci.yml

@@ -103,7 +103,11 @@ jobs:
       - name: Docker Info
         run: docker info
       - name: Integration Tests
-        run: uv run python -m pytest tests/integration --run-integration
+        env:
+          AGENT_MEMORY_ENDPOINT: ${{ secrets.AGENT_MEMORY_ENDPOINT }}
+          AGENT_MEMORY_API_KEY: ${{ secrets.AGENT_MEMORY_API_KEY }}
+          AGENT_MEMORY_STORE_ID: ${{ secrets.AGENT_MEMORY_STORE_ID }}
+        run: uv run python -m pytest tests/integration
 
   integration-api:
     runs-on: ubuntu-latest
@@ -138,7 +142,7 @@ jobs:
             exit 0
           fi
           docker info
-          uv run python -m pytest tests/integration --run-integration --run-api-tests
+          uv run python -m pytest tests/integration
 
   build:
     runs-on: ubuntu-latest

.gitignore

@@ -11,3 +11,5 @@ dist/
 htmlcov/
 
 NeMo-Agent-Toolkit-3P-repositories.pdf
+
+.claude/settings.local.json

Makefile

@@ -1,4 +1,4 @@
-.PHONY: setup lint test test-integration test-integration-api validate build check clean
+.PHONY: setup lint test test-integration validate build check clean
 
 setup:
 	uv sync --group dev --extra test
@@ -11,10 +11,7 @@ test:
 	uv run python -m pytest
 
 test-integration:
-	uv run python -m pytest tests/integration --run-integration
-
-test-integration-api:
-	uv run python -m pytest tests/integration --run-integration --run-api-tests
+	uv run python -m pytest tests/integration
 
 validate:
 	uv run nat validate --config_file examples/tool_based_memory/configs/config.yml

README.md

@@ -6,7 +6,7 @@
 
 **Redis-backed memory for [NVIDIA NeMo Agent Toolkit](https://github.com/NVIDIA/NeMo-Agent-Toolkit)** — production [Redis Agent Memory](https://redis.github.io/agent-memory-server/quick-start/) integrations plus direct Redis plugins (the historical `nat.plugins.redis` stack).
 
-In NAT, long-term recall is usually wired through memory tools such as `get_memory` and `add_memory`, which delegate to a **`MemoryEditor`** implementation. This package adds Redis-backed options in **two families**: **Redis Agent Memory** (AMS over HTTP — richer lifecycle, optional auto-memory workflow) and **direct Redis** (JSON + vector search or plain KV inside `nat.plugins.redis`, no AMS).
+In NAT, long-term recall is usually wired through memory tools such as `get_memory` and `add_memory`, which delegate to a **`MemoryEditor`** implementation. This package adds Redis-backed options in **two families**: **Redis Agent Memory** (over HTTP — richer lifecycle, optional auto-memory workflow) and **direct Redis** (JSON + vector search or plain KV inside `nat.plugins.redis`).
 
 This repo is the standalone home for integrations that used to ship under
 [`packages/nvidia_nat_redis`](https://github.com/NVIDIA/NeMo-Agent-Toolkit/tree/develop/packages/nvidia_nat_redis) in the NeMo Agent Toolkit monorepo.
@@ -15,28 +15,28 @@ This repo is the standalone home for integrations that used to ship under
 1. `_type: redis_agent_memory_backend` — Redis Agent Memory as a NAT `MemoryEditor` long-term memory backend.
 2. `_type: redis_agent_memory_auto_memory` — A native Redis Agent Memory wrapper that uses working memory plus
 `memory_prompt` hydration on every turn.
-AMS runs as a separate service; these surfaces talk to it via the agent-memory client.
+Agent Memory runs as a separate service; these surfaces talk to it via the agent-memory client.
 
 1. `_type: redis_agent_memory_backend` — Redis Agent Memory behind NAT’s standard **`MemoryEditor`** contract (fits tool-driven long-term memory).
-2. `_type: redis_agent_memory_auto_memory` — A **`workflow` wrapper** (not just another editor): working memory, `memory_prompt` hydration every turn, turn capture, and promotion — the fullest AMS-shaped integration.
+2. `_type: redis_agent_memory_auto_memory` — A **`workflow` wrapper** (not just another editor): working memory, `memory_prompt` hydration every turn, turn capture, and promotion — the fullest Agent Memory integration.
 
 ### Direct Redis (simple in-Redis memory)
 
 Loaded via the `nat.plugins` setuptools entry point `nat_redis` (same name as
 the historical monorepo package):
 
-- `_type: redis_memory` — Vector search over JSON documents in Redis (RediSearch); requires a workflow `embedder` and Redis Stack (or Redis with search + JSON support). Implements **`MemoryEditor`**-style semantic memory without AMS.
+- `_type: redis_memory` — Vector search over JSON documents in Redis (RediSearch); requires a workflow `embedder` and Redis Stack (or Redis with search + JSON support). Implements **`MemoryEditor`**-style semantic memory without Redis Agent Memory.
 - `_type: redis` — NAT **object store** on plain Redis key–value storage (not semantic long-term memory).
 
 | | **Redis Agent Memory** | **Direct Redis** |
 | --- | --- | --- |
-| **Runs** | AMS service + Redis | Redis only (your NAT process uses the client) |
-| **Best for** | Learning-style memory, working memory, AMS filters and APIs | Lightweight Redis-native memory or KV object store |
-| **Tradeoff** | Operate AMS; HTTP path | Simpler ops for `redis` / `redis_memory`; vector path needs embedder + search-capable Redis |
+| **Runs** | Agent Memory service + Redis | Redis only (your NAT process uses the client) |
+| **Best for** | Learning-style memory, working memory, Agent Memory filters and APIs | Lightweight Redis-native memory or KV object store |
+| **Tradeoff** | Operate Agent Memory; HTTP path | Simpler ops for `redis` / `redis_memory`; vector path needs embedder + search-capable Redis |
 
 Use **Redis Agent Memory** when you want the Redis Agent Memory feature set.
 Use **`redis_memory`** when you only need a lightweight Redis-native `MemoryEditor`
-without AMS.
+without automatic Agent Memory.
 
 ### Python imports (same as NeMo)
 
@@ -87,7 +87,7 @@ versions.
 - Use `_type: redis_agent_memory_backend` when your workflow already uses NAT memory tools and you want Redis Agent Memory behind the standard `MemoryEditor` contract.
 - Use `_type: redis_agent_memory_auto_memory` when you want Redis Agent Memory to own working-memory continuity, prompt hydration, and turn capture on every request. This exposes the richness of Redis Agent Memory in its fullest form.
 - Use `_type: redis_memory` when you want the simpler direct-Redis memory from NeMo Agent Toolkit (Redis JSON + vector index, no Redis Agent Memory). You must configure an `embedder` reference and run a Redis deployment that supports the search commands used by the plugin.
-- Use `_type: redis` when you need NAT’s Redis-backed **object store** (KV), not vector or AMS-backed semantic memory.
+- Use `_type: redis` when you need NAT’s Redis-backed **object store** (KV), not vector or Agent Memory-backed semantic memory.
 
 ## Integration Modes
 
@@ -164,15 +164,17 @@ make build
 make check
 ```
 
-Integration tests are opt-in because they start real Redis and Agent Memory
-Server containers:
+`make test` runs the whole suite. Integration tests skip themselves when their
+infrastructure is absent (no Docker, unreachable Agent Memory Server, or missing
+API/cloud credentials), so a bare run stays green anywhere. To run just the
+integration tests where that infrastructure exists:
 
 ```bash
 make test-integration
 ```
 
 The example Compose files are intended for local development. They bind Redis
-and AMS to `127.0.0.1` and run AMS with auth disabled.
+and Redis Agent Memory to `127.0.0.1` and run Redis Agent Memory with auth disabled.
 
 See [CONTRIBUTING.md](CONTRIBUTING.md).
 

docs/configuration.md

@@ -48,9 +48,9 @@ Supported config fields:
 
 Runtime behavior:
 
-- `add_items()` creates long-term memories with optional AMS fields such as `session_id`, `namespace`, `entities`, `event_date`, and `memory_type`
+- `add_items()` creates long-term memories with optional Agent Memory fields such as `session_id`, `namespace`, `entities`, `event_date`, and `memory_type`
 - `add_items()` falls back to NAT `conversation_id` for `session_id` when no explicit value is provided
-- `search()` requires `user_id` and forwards AMS filters such as `session_id`, `namespace`, `topics`, `entities`, `memory_type`, `distance_threshold`, and `recency`
+- `search()` requires `user_id` and forwards Agent Memory filters such as `session_id`, `namespace`, `topics`, `entities`, `memory_type`, `distance_threshold`, and `recency`
 - `search()` falls back to NAT `user_id` when the current runtime context provides one
 - `remove_items()` supports direct deletion by `memory_id` / `memory_ids` or filtered deletion by search
 

examples/agent_auto_memory/README.md

@@ -41,7 +41,7 @@ The example expects:
 ## Start Services
 
 Compose starts Redis Stack and Redis Agent Memory containers for local development. Both
-ports bind to `127.0.0.1`, and AMS auth is disabled.
+ports bind to `127.0.0.1`, and Agent Memory auth is disabled.
 
 ```bash
 docker compose \

examples/tool_based_memory/README.md

@@ -37,7 +37,7 @@ Edit `examples/tool_based_memory/.env`:
 ## Start Services
 
 Compose starts Redis Stack and Agent Memory Server for local development. Both
-ports bind to `127.0.0.1`, and AMS auth is disabled.
+ports bind to `127.0.0.1`, and Agent Memory auth is disabled.
 
 ```bash
 docker compose \

pyproject.toml

@@ -25,6 +25,7 @@ dependencies = [
   "numpy>=1.24",
   "nvidia-nat-core>=1.6.0,<2.0.0",
   "redis>=5.0.0,<6.0.0",
+  "redis-agent-memory>=0.0.4,<1.0.0",
   "redisvl>=0.16.0,<1.0.0",
 ]
 

src/nat/plugins/redis/register.py

@@ -22,3 +22,4 @@
 from . import memory
 from . import object_store
 from nvidia_nat_redis.redis_agent_memory import register as _ams_register  # noqa: F401
+from nvidia_nat_redis.cloud_redis_agent_memory import memory as _cloud_ams_register  # noqa: F401

src/nvidia_nat_redis/cloud_redis_agent_memory/__init__.py

@@ -0,0 +1,12 @@
+# SPDX-FileCopyrightText: Copyright (c) 2026, Redis
+# SPDX-License-Identifier: Apache-2.0
+
+"""Public Redis Agent Memory cloud integration classes exported by this NAT plugin."""
+
+from .editor import CloudRedisAgentMemoryEditor
+from .memory import CloudRedisAgentMemoryBackendConfig
+
+__all__ = [
+    "CloudRedisAgentMemoryBackendConfig",
+    "CloudRedisAgentMemoryEditor",
+]