Skip to content

Commit 8b4f0c5

Browse files
Brings repo into compliance with Nvidia NeMo third-party requirements
2 parents 95b9587 + 994ffc6 commit 8b4f0c5

8 files changed

Lines changed: 11 additions & 8 deletions

File tree

.gitignore

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,3 +9,5 @@ __pycache__/
99
build/
1010
dist/
1111
htmlcov/
12+
13+
NeMo-Agent-Toolkit-3P-repositories.pdf

README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,7 @@ Toolkit. The setuptools entry point **`nat_redis`** loads
5050
## Install
5151

5252
```bash
53-
pip install nvidia-nat-redis
53+
pip install nemo-agent-toolkit-redis
5454
```
5555

5656
For local development in this repo:

examples/agent_auto_memory/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ The inner agent in this example is NAT's `chat_completion` function, so the
1515
base plugin install is enough:
1616

1717
```bash
18-
pip install "nvidia-nat-redis"
18+
pip install "nemo-agent-toolkit-redis"
1919
```
2020

2121
For local development from this repository:

examples/tool_based_memory/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@ agent package in addition to this plugin.
1212
For a published install:
1313

1414
```bash
15-
pip install "nvidia-nat[langchain]" "nvidia-nat-redis"
15+
pip install "nvidia-nat[langchain]" "nemo-agent-toolkit-redis"
1616
```
1717

1818
For local development from this repository:

pyproject.toml

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@ build-backend = "setuptools.build_meta"
33
requires = ["setuptools>=69", "setuptools-scm>=8"]
44

55
[project]
6-
name = "nvidia-nat-redis"
6+
name = "nemo-agent-toolkit-redis"
77
dynamic = ["version"]
88
requires-python = ">=3.11,<3.14"
99
description = "Redis Agent Memory integration for NVIDIA NeMo Agent Toolkit"
@@ -39,7 +39,6 @@ issues = "https://github.com/redis-developer/nvidia-nat-redis/issues"
3939
source = "https://github.com/redis-developer/nvidia-nat-redis"
4040

4141
[project.entry-points.'nat.components']
42-
redis_agent_memory = "nvidia_nat_redis.redis_agent_memory.register"
4342
nat_redis = "nat.plugins.redis.register"
4443

4544
[dependency-groups]

src/nat/plugins/redis/register.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,3 +21,4 @@
2121

2222
from . import memory
2323
from . import object_store
24+
from nvidia_nat_redis.redis_agent_memory import register as _ams_register # noqa: F401

src/nvidia_nat_redis/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,6 @@
66
from importlib.metadata import PackageNotFoundError, version
77

88
try:
9-
__version__ = version("nvidia-nat-redis")
9+
__version__ = version("nemo-agent-toolkit-redis")
1010
except PackageNotFoundError: # pragma: no cover - source-tree usage before installation
1111
__version__ = "0.0.0"

src/nvidia_nat_redis/redis_agent_memory/register.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,8 +4,9 @@
44
"""
55
Load Redis Agent Memory NAT components for entry-point registration.
66
7-
The ``nat.components`` entry point targets this module. Importing the backend
8-
and wrapper modules runs their NAT registration decorators.
7+
Imported by ``nat.plugins.redis.register``, the ``nat.components`` entry point
8+
target (``nat_redis``). Importing this module runs the NAT registration
9+
decorators for the Redis Agent Memory backend and auto-memory wrapper.
910
"""
1011

1112
from . import memory as _memory # noqa: F401

0 commit comments

Comments
 (0)