nvidia-nat-redis 1.3.0a20250928__py3-none-any.whl → 1.4.0a20251111__py3-none-any.whl

nat/plugins/redis/memory.py

@@ -17,6 +17,8 @@ from pydantic import Field
 
 from nat.builder.builder import Builder
 from nat.cli.register_workflow import register_memory
+from nat.data_models.common import OptionalSecretStr
+from nat.data_models.common import get_secret_value
 from nat.data_models.component_ref import EmbedderRef
 from nat.data_models.memory import MemoryBaseConfig
 
@@ -25,6 +27,7 @@ class RedisMemoryClientConfig(MemoryBaseConfig, name="redis_memory"):
     host: str = Field(default="localhost", description="Redis server host")
     db: int = Field(default=0, description="Redis DB")
     port: int = Field(default=6379, description="Redis server port")
+    password: OptionalSecretStr = Field(default=None, description="Password for the Redis server")
     key_prefix: str = Field(default="nat", description="Key prefix to use for redis keys")
     embedder: EmbedderRef = Field(description=("Instance name of the memory client instance from the workflow "
                                                "configuration object."))
@@ -43,6 +46,7 @@ async def redis_memory_client(config: RedisMemoryClientConfig, builder: Builder)
     redis_client = redis.Redis(host=config.host,
                                port=config.port,
                                db=config.db,
+                               password=get_secret_value(config.password),
                                decode_responses=True,
                                socket_timeout=5.0,
                                socket_connect_timeout=5.0)

nat/plugins/redis/object_store.py

@@ -14,21 +14,32 @@
 # limitations under the License.
 
 from pydantic import Field
+from pydantic import field_validator
 
 from nat.builder.builder import Builder
 from nat.cli.register_workflow import register_object_store
+from nat.data_models.common import OptionalSecretStr
 from nat.data_models.object_store import ObjectStoreBaseConfig
 
 
 class RedisObjectStoreClientConfig(ObjectStoreBaseConfig, name="redis"):
     """
-    Object store that stores objects in a Redis database.
+    Object store that stores objects in a Redis database with optional TTL.
     """
 
     host: str = Field(default="localhost", description="The host of the Redis server")
     db: int = Field(default=0, description="The Redis logical database number")
     port: int = Field(default=6379, description="The port of the Redis server")
     bucket_name: str = Field(description="The name of the bucket to use for the object store")
+    password: OptionalSecretStr = Field(default=None, description="The password for the Redis server")
+    ttl: int | None = Field(default=None, description="TTL in seconds for objects (None = no expiration)")
+
+    @field_validator("ttl")
+    @classmethod
+    def validate_ttl(cls, v: int | None) -> int | None:
+        if v is not None and v <= 0:
+            raise ValueError("TTL must be a positive integer greater than 0")
+        return v
 
 
 @register_object_store(config_type=RedisObjectStoreClientConfig)

nat/plugins/redis/redis_object_store.py

@@ -28,12 +28,22 @@ logger = logging.getLogger(__name__)
 
 class RedisObjectStore(ObjectStore):
     """
-    Implementation of ObjectStore that stores objects in Redis.
+    Implementation of ObjectStore that stores objects in Redis with optional TTL.
 
     Each object is stored as a single binary value at key "nat/object_store/{bucket_name}/{object_key}".
+    When TTL is configured, keys will automatically expire after the specified duration in seconds.
     """
 
-    def __init__(self, *, bucket_name: str, host: str, port: int, db: int):
+    def __init__(
+        self,
+        *,
+        bucket_name: str,
+        host: str,
+        port: int,
+        db: int,
+        password: str | None = None,
+        ttl: int | None = None,
+    ):
 
         super().__init__()
 
@@ -41,6 +51,8 @@ class RedisObjectStore(ObjectStore):
         self._host = host
         self._port = port
         self._db = db
+        self._password = password
+        self._ttl = ttl
         self._client: redis.Redis | None = None
 
     async def __aenter__(self) -> "RedisObjectStore":
@@ -52,6 +64,7 @@ class RedisObjectStore(ObjectStore):
             host=self._host,
             port=self._port,
             db=self._db,
+            password=self._password,
             socket_timeout=5.0,
             socket_connect_timeout=5.0,
         )
@@ -86,7 +99,7 @@ class RedisObjectStore(ObjectStore):
 
         item_json = item.model_dump_json()
         # Redis SET with NX ensures we do not overwrite existing keys
-        if not await self._client.set(full_key, item_json, nx=True):
+        if not await self._client.set(full_key, item_json, nx=True, ex=self._ttl):
             raise KeyAlreadyExistsError(key=key,
                                         additional_message=f"Redis bucket {self._bucket_name} already has key {key}")
 
@@ -98,7 +111,7 @@ class RedisObjectStore(ObjectStore):
 
         full_key = self._make_key(key)
         item_json = item.model_dump_json()
-        await self._client.set(full_key, item_json)
+        await self._client.set(full_key, item_json, ex=self._ttl)
 
     @override
     async def get_object(self, key: str) -> ObjectStoreItem:

{nvidia_nat_redis-1.3.0a20250928.dist-info → nvidia_nat_redis-1.4.0a20251111.dist-info}/METADATA

@@ -1,7 +1,12 @@
 Metadata-Version: 2.4
 Name: nvidia-nat-redis
-Version: 1.3.0a20250928
+Version: 1.4.0a20251111
 Summary: Subpackage for Redis integration in NeMo Agent toolkit
+Author: NVIDIA Corporation
+Maintainer: NVIDIA Corporation
+License: Apache-2.0
+Project-URL: documentation, https://docs.nvidia.com/nemo/agent-toolkit/latest/
+Project-URL: source, https://github.com/NVIDIA/NeMo-Agent-Toolkit
 Keywords: ai,agents,memory
 Classifier: Programming Language :: Python
 Classifier: Programming Language :: Python :: 3.11
@@ -9,8 +14,11 @@ Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Requires-Python: <3.14,>=3.11
 Description-Content-Type: text/markdown
-Requires-Dist: nvidia-nat==v1.3.0a20250928
+License-File: LICENSE-3rd-party.txt
+License-File: LICENSE.md
+Requires-Dist: nvidia-nat==v1.4.0a20251111
 Requires-Dist: redis~=4.3.4
+Dynamic: license-file
 
 <!--
 SPDX-FileCopyrightText: Copyright (c) 2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.

nvidia_nat_redis-1.4.0a20251111.dist-info/RECORD

@@ -0,0 +1,15 @@
+nat/meta/pypi.md,sha256=TpeNbVZJxzvEf0Gh3BGvLHPYsKnXjgM_KQVCayBPXso,1090
+nat/plugins/redis/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+nat/plugins/redis/memory.py,sha256=up25g7p7EXUcwAvMgCJjzyeGxH0454y9rE6QPPPYjK4,2825
+nat/plugins/redis/object_store.py,sha256=cRmZfyKJB2yhrbhsbFEaJ8UMjX2E5ca_eT9o41dBK7o,2233
+nat/plugins/redis/redis_editor.py,sha256=nkSIWi1HUPald088fXTuF0rmZ0uS_3V65Vxy20vLSgk,9746
+nat/plugins/redis/redis_object_store.py,sha256=AvVWCnAO6QMePE8twB03dFSe1VHLq_AOcqCcvDZscbE,4667
+nat/plugins/redis/register.py,sha256=dJBKi-7W72ipkmZTOIo1E3ETffmJIlYhQTOlrkiFH3A,834
+nat/plugins/redis/schema.py,sha256=Zcas3hIIqG7wuR94baYRFycmnccB3CuGmTs4p8Vv4mA,5589
+nvidia_nat_redis-1.4.0a20251111.dist-info/licenses/LICENSE-3rd-party.txt,sha256=fOk5jMmCX9YoKWyYzTtfgl-SUy477audFC5hNY4oP7Q,284609
+nvidia_nat_redis-1.4.0a20251111.dist-info/licenses/LICENSE.md,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+nvidia_nat_redis-1.4.0a20251111.dist-info/METADATA,sha256=nobgaoa3267c99TAk-6bs2C_Myrzs6ycWKdUnArPTew,1894
+nvidia_nat_redis-1.4.0a20251111.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+nvidia_nat_redis-1.4.0a20251111.dist-info/entry_points.txt,sha256=nyS8t8L9CbRFIMlE70RQBtJXrflBP4Ltl5zAkIl44So,56
+nvidia_nat_redis-1.4.0a20251111.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
+nvidia_nat_redis-1.4.0a20251111.dist-info/RECORD,,