Merge pull request #12 from bbednarski9/fix/issue-8-nat-plugins-entrypoint

Use nat.plugins entry point group

Author: justin-cechmanek

README.md

@@ -22,8 +22,8 @@ AMS runs as a separate service; these surfaces talk to it via the agent-memory c
 
 ### Direct Redis (simple in-Redis memory)
 
-Loaded via the `nat_redis` setuptools entry point (same name as the historical
-monorepo package):
+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` — NAT **object store** on plain Redis key–value storage (not semantic long-term memory).
@@ -42,7 +42,7 @@ without AMS.
 
 Direct Redis support is implemented only under **`nat.plugins.redis`** (for example
 `from nat.plugins.redis.redis_editor import RedisEditor`), matching NeMo Agent
-Toolkit. The setuptools entry point **`nat_redis`** loads
+Toolkit. The `nat.plugins` setuptools entry point **`nat_redis`** loads
 `nat.plugins.redis.register`. Redis Agent Memory code lives under
 **`nvidia_nat_redis.redis_agent_memory`**.
 

docs/configuration.md

@@ -13,7 +13,7 @@ Direct Redis memory provides a more standard semantic memory layer for LLM appli
 1. Long-term memory backend: `_type: redis_agent_memory_backend`
 2. Native automatic wrapper: `_type: redis_agent_memory_auto_memory`
 
-**Direct Redis (entry point `nat_redis` → `nat.plugins.redis.register`)**
+**Direct Redis (`nat.plugins` entry point `nat_redis` → `nat.plugins.redis.register`)**
 
 3. In-Redis vector memory: `_type: redis_memory`
 4. Object store: `_type: redis`

pyproject.toml

@@ -40,7 +40,7 @@ documentation = "https://github.com/redis-developer/nemo-agent-toolkit-redis/tre
 issues = "https://github.com/redis-developer/nemo-agent-toolkit-redis/issues"
 source = "https://github.com/redis-developer/nemo-agent-toolkit-redis"
 
-[project.entry-points.'nat.components']
+[project.entry-points.'nat.plugins']
 nat_redis = "nat.plugins.redis.register"
 
 [dependency-groups]

src/nat/plugins/redis/register.py

@@ -17,7 +17,7 @@
 # isort:skip_file
 
 # Import any providers which need to be automatically registered here.
-# The ``nat.components`` entry point ``nat_redis`` targets this module.
+# The ``nat.plugins`` entry point ``nat_redis`` targets this module.
 
 from . import memory
 from . import object_store

src/nvidia_nat_redis/redis_agent_memory/register.py

@@ -4,7 +4,7 @@
 """
 Load Redis Agent Memory NAT components for entry-point registration.
 
-Imported by ``nat.plugins.redis.register``, the ``nat.components`` entry point
+Imported by ``nat.plugins.redis.register``, the ``nat.plugins`` entry point
 target (``nat_redis``). Importing this module runs the NAT registration
 decorators for the Redis Agent Memory backend and auto-memory wrapper.
 """

tests/test_entry_points.py

@@ -0,0 +1,28 @@
+# SPDX-FileCopyrightText: Copyright (c) 2026, Redis
+# SPDX-License-Identifier: Apache-2.0
+
+from __future__ import annotations
+
+from importlib.metadata import entry_points
+
+
+def test_nat_redis_entry_point_loads_registered_components() -> None:
+    """The package must expose Redis components through NAT's third-party plugin group."""
+    from nat.cli.type_registry import GlobalTypeRegistry
+
+    plugins = entry_points(group="nat.plugins")
+    redis_entry_points = [
+        plugin for plugin in plugins if plugin.name == "nat_redis" and plugin.value == "nat.plugins.redis.register"
+    ]
+
+    assert redis_entry_points, "Expected nat_redis to target nat.plugins.redis.register in the nat.plugins group"
+
+    redis_entry_points[0].load()
+
+    registry = GlobalTypeRegistry.get()
+    memory_types = {registered.local_name for registered in registry.get_registered_memorys()}
+    object_store_types = {registered.local_name for registered in registry.get_registered_object_stores()}
+
+    assert "redis_memory" in memory_types
+    assert "redis_agent_memory_backend" in memory_types
+    assert "redis" in object_store_types