nvidia-nat-redis 1.2.0rc5__py3-none-any.whl → 1.2.0rc6__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (14) hide show
  1. {aiq → nat}/meta/pypi.md +2 -2
  2. {aiq → nat}/plugins/redis/memory.py +7 -7
  3. {aiq → nat}/plugins/redis/redis_editor.py +3 -3
  4. {nvidia_nat_redis-1.2.0rc5.dist-info → nvidia_nat_redis-1.2.0rc6.dist-info}/METADATA +5 -5
  5. nvidia_nat_redis-1.2.0rc6.dist-info/RECORD +11 -0
  6. nvidia_nat_redis-1.2.0rc6.dist-info/entry_points.txt +2 -0
  7. nvidia_nat_redis-1.2.0rc6.dist-info/top_level.txt +1 -0
  8. nvidia_nat_redis-1.2.0rc5.dist-info/RECORD +0 -11
  9. nvidia_nat_redis-1.2.0rc5.dist-info/entry_points.txt +0 -2
  10. nvidia_nat_redis-1.2.0rc5.dist-info/top_level.txt +0 -1
  11. {aiq → nat}/plugins/redis/__init__.py +0 -0
  12. {aiq → nat}/plugins/redis/register.py +0 -0
  13. {aiq → nat}/plugins/redis/schema.py +0 -0
  14. {nvidia_nat_redis-1.2.0rc5.dist-info → nvidia_nat_redis-1.2.0rc6.dist-info}/WHEEL +0 -0
{aiq → nat}/meta/pypi.md RENAMED
@@ -15,9 +15,9 @@ See the License for the specific language governing permissions and
15
15
  limitations under the License.
16
16
  -->
17
17
 
18
- ![NVIDIA NeMo Agent Toolkit](https://media.githubusercontent.com/media/NVIDIA/NeMo-Agent-Toolkit/refs/heads/main/docs/source/_static/aiqtoolkit_banner.png "NeMo Agent toolkit banner image")
18
+ ![NVIDIA NeMo Agent Toolkit](https://media.githubusercontent.com/media/NVIDIA/NeMo-Agent-Toolkit/refs/heads/main/docs/source/_static/banner.png "NeMo Agent toolkit banner image")
19
19
 
20
20
  # NVIDIA NeMo Agent Toolkit Subpackage
21
21
  This is a subpackage for Redis memory integration in NeMo Agent toolkit.
22
22
 
23
- For more information about NeMo Agent toolkit, please visit the [NeMo Agent toolkit package](https://pypi.org/project/aiqtoolkit/).
23
+ For more information about NeMo Agent toolkit, please visit the [NeMo Agent toolkit package](https://pypi.org/project/nvidia-nat/).
{aiq → nat}/plugins/redis/memory.py RENAMED
@@ -16,18 +16,18 @@
16
16
  import redis.asyncio as redis
17
17
  from pydantic import Field
18
18
 
19
- from aiq.builder.builder import Builder
20
- from aiq.builder.framework_enum import LLMFrameworkEnum
21
- from aiq.cli.register_workflow import register_memory
22
- from aiq.data_models.component_ref import EmbedderRef
23
- from aiq.data_models.memory import MemoryBaseConfig
19
+ from nat.builder.builder import Builder
20
+ from nat.builder.framework_enum import LLMFrameworkEnum
21
+ from nat.cli.register_workflow import register_memory
22
+ from nat.data_models.component_ref import EmbedderRef
23
+ from nat.data_models.memory import MemoryBaseConfig
24
24
 
25
25
 
26
26
  class RedisMemoryClientConfig(MemoryBaseConfig, name="redis_memory"):
27
27
  host: str | None = Field(default="localhost", description="Redis server host")
28
28
  db: str | None = Field(default="0", description="Redis DB")
29
29
  port: str | None = Field(default="6379", description="Redis server port")
30
- key_prefix: str | None = Field(default="aiq", description="Key prefix to use for redis keys")
30
+ key_prefix: str | None = Field(default="nat", description="Key prefix to use for redis keys")
31
31
  embedder: EmbedderRef = Field(description=("Instance name of the memory client instance from the workflow "
32
32
  "configuration object."))
33
33
 
@@ -35,7 +35,7 @@ class RedisMemoryClientConfig(MemoryBaseConfig, name="redis_memory"):
35
35
  @register_memory(config_type=RedisMemoryClientConfig)
36
36
  async def redis_memory_client(config: RedisMemoryClientConfig, builder: Builder):
37
37
 
38
- from aiq.plugins.redis.redis_editor import RedisEditor
38
+ from nat.plugins.redis.redis_editor import RedisEditor
39
39
 
40
40
  from .schema import ensure_index_exists
41
41
 
{aiq → nat}/plugins/redis/redis_editor.py RENAMED
@@ -22,8 +22,8 @@ import redis.exceptions as redis_exceptions
22
22
  from langchain_core.embeddings import Embeddings
23
23
  from redis.commands.search.query import Query
24
24
 
25
- from aiq.memory.interfaces import MemoryEditor
26
- from aiq.memory.models import MemoryItem
25
+ from nat.memory.interfaces import MemoryEditor
26
+ from nat.memory.models import MemoryItem
27
27
 
28
28
  logger = logging.getLogger(__name__)
29
29
 
@@ -32,7 +32,7 @@ INDEX_NAME = "memory_idx"
32
32
 
33
33
  class RedisEditor(MemoryEditor):
34
34
  """
35
- Wrapper class that implements AIQ Toolkit Interfaces for Redis memory storage.
35
+ Wrapper class that implements NAT interfaces for Redis memory storage.
36
36
  """
37
37
 
38
38
  def __init__(self, redis_client: redis.Redis, key_prefix: str, embedder: Embeddings):
{nvidia_nat_redis-1.2.0rc5.dist-info → nvidia_nat_redis-1.2.0rc6.dist-info}/METADATA RENAMED
@@ -1,12 +1,12 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: nvidia-nat-redis
3
- Version: 1.2.0rc5
4
- Summary: Subpackage for Redis memory integration in AIQtoolkit
3
+ Version: 1.2.0rc6
4
+ Summary: Subpackage for Redis integration in NeMo Agent toolkit
5
5
  Keywords: ai,agents,memory
6
6
  Classifier: Programming Language :: Python
7
7
  Requires-Python: <3.13,>=3.11
8
8
  Description-Content-Type: text/markdown
9
- Requires-Dist: nvidia-nat~=1.2
9
+ Requires-Dist: nvidia-nat==v1.2.0-rc6
10
10
  Requires-Dist: redis~=4.3.4
11
11
 
12
12
  <!--
@@ -26,9 +26,9 @@ See the License for the specific language governing permissions and
26
26
  limitations under the License.
27
27
  -->
28
28
 
29
- ![NVIDIA NeMo Agent Toolkit](https://media.githubusercontent.com/media/NVIDIA/NeMo-Agent-Toolkit/refs/heads/main/docs/source/_static/aiqtoolkit_banner.png "NeMo Agent toolkit banner image")
29
+ ![NVIDIA NeMo Agent Toolkit](https://media.githubusercontent.com/media/NVIDIA/NeMo-Agent-Toolkit/refs/heads/main/docs/source/_static/banner.png "NeMo Agent toolkit banner image")
30
30
 
31
31
  # NVIDIA NeMo Agent Toolkit Subpackage
32
32
  This is a subpackage for Redis memory integration in NeMo Agent toolkit.
33
33
 
34
- For more information about NeMo Agent toolkit, please visit the [NeMo Agent toolkit package](https://pypi.org/project/aiqtoolkit/).
34
+ For more information about NeMo Agent toolkit, please visit the [NeMo Agent toolkit package](https://pypi.org/project/nvidia-nat/).
nvidia_nat_redis-1.2.0rc6.dist-info/RECORD ADDED
@@ -0,0 +1,11 @@
1
+ nat/meta/pypi.md,sha256=TpeNbVZJxzvEf0Gh3BGvLHPYsKnXjgM_KQVCayBPXso,1090
2
+ nat/plugins/redis/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
3
+ nat/plugins/redis/memory.py,sha256=2AbG29r6tul4RVvGnSVrQnT96Hvm725i1X23zvDGWTg,2569
4
+ nat/plugins/redis/redis_editor.py,sha256=BskJN5R0h1xzz7w5Ic8IVvd4z-qdBp9ZBIpGYNiSwE8,10080
5
+ nat/plugins/redis/register.py,sha256=_ffKNKnMfkB2HzX4Nk_9EW0pwebg3GuzAE-iB-CoC3E,839
6
+ nat/plugins/redis/schema.py,sha256=lcazZKzWEoJEJmGEt6rk135zJ6kGqw6v2b1B5l1xg7o,5494
7
+ nvidia_nat_redis-1.2.0rc6.dist-info/METADATA,sha256=rX8jEER2SgR2nB3PvXoX7Vt1CHowv77jRzo7TNRAsoU,1424
8
+ nvidia_nat_redis-1.2.0rc6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
9
+ nvidia_nat_redis-1.2.0rc6.dist-info/entry_points.txt,sha256=nyS8t8L9CbRFIMlE70RQBtJXrflBP4Ltl5zAkIl44So,56
10
+ nvidia_nat_redis-1.2.0rc6.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
11
+ nvidia_nat_redis-1.2.0rc6.dist-info/RECORD,,
nvidia_nat_redis-1.2.0rc6.dist-info/entry_points.txt ADDED
@@ -0,0 +1,2 @@
1
+ [nat.components]
2
+ nat_redis = nat.plugins.redis.register
nvidia_nat_redis-1.2.0rc5.dist-info/RECORD DELETED
@@ -1,11 +0,0 @@
1
- aiq/meta/pypi.md,sha256=tfhFKDhfJtxBVT6J5pnvrIdaBN_Uy5Dy6in6CCZpYpU,1101
2
- aiq/plugins/redis/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
3
- aiq/plugins/redis/memory.py,sha256=rz-cbcSOe1knd795783leY4vm0n1C4UtUJtZOHmINcU,2569
4
- aiq/plugins/redis/redis_editor.py,sha256=2hb5BGSTjWId5sb6dNAA3MLzO6WSVAU-Px-iv7Fzy2Q,10088
5
- aiq/plugins/redis/register.py,sha256=_ffKNKnMfkB2HzX4Nk_9EW0pwebg3GuzAE-iB-CoC3E,839
6
- aiq/plugins/redis/schema.py,sha256=lcazZKzWEoJEJmGEt6rk135zJ6kGqw6v2b1B5l1xg7o,5494
7
- nvidia_nat_redis-1.2.0rc5.dist-info/METADATA,sha256=PrUfA8u5_hdV53lCkhonCJtaZILr-LCmAsxw7ywx5i4,1427
8
- nvidia_nat_redis-1.2.0rc5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
9
- nvidia_nat_redis-1.2.0rc5.dist-info/entry_points.txt,sha256=JyY7zDx7SZTvETPEEZuTgDVhe3EN23s1Glyp8RHIoWY,56
10
- nvidia_nat_redis-1.2.0rc5.dist-info/top_level.txt,sha256=fo7AzYcNhZ_tRWrhGumtxwnxMew4xrT1iwouDy_f0Kc,4
11
- nvidia_nat_redis-1.2.0rc5.dist-info/RECORD,,
nvidia_nat_redis-1.2.0rc5.dist-info/entry_points.txt DELETED
@@ -1,2 +0,0 @@
1
- [aiq.components]
2
- aiq_redis = aiq.plugins.redis.register
nvidia_nat_redis-1.2.0rc5.dist-info/top_level.txt DELETED
@@ -1 +0,0 @@
1
- aiq
{aiq → nat}/plugins/redis/__init__.py RENAMED
File without changes
{aiq → nat}/plugins/redis/register.py RENAMED
File without changes
{aiq → nat}/plugins/redis/schema.py RENAMED
File without changes