nvidia-nat-redis 1.4.0a20251105__py3-none-any.whl → 1.4.0a20251107__py3-none-any.whl

nat/plugins/redis/object_store.py

@@ -14,6 +14,7 @@
 # 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
@@ -23,7 +24,7 @@ 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")
@@ -31,6 +32,14 @@ class RedisObjectStoreClientConfig(ObjectStoreBaseConfig, name="redis"):
     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, password: str | None = None):
+    def __init__(
+        self,
+        *,
+        bucket_name: str,
+        host: str,
+        port: int,
+        db: int,
+        password: str | None = None,
+        ttl: int | None = None,
+    ):
 
         super().__init__()
 
@@ -42,6 +52,7 @@ class RedisObjectStore(ObjectStore):
         self._port = port
         self._db = db
         self._password = password
+        self._ttl = ttl
         self._client: redis.Redis | None = None
 
     async def __aenter__(self) -> "RedisObjectStore":
@@ -88,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}")
 
@@ -100,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.4.0a20251105.dist-info → nvidia_nat_redis-1.4.0a20251107.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nvidia-nat-redis
-Version: 1.4.0a20251105
+Version: 1.4.0a20251107
 Summary: Subpackage for Redis integration in NeMo Agent toolkit
 Author: NVIDIA Corporation
 Maintainer: NVIDIA Corporation
@@ -16,7 +16,7 @@ Requires-Python: <3.14,>=3.11
 Description-Content-Type: text/markdown
 License-File: LICENSE-3rd-party.txt
 License-File: LICENSE.md
-Requires-Dist: nvidia-nat==v1.4.0a20251105
+Requires-Dist: nvidia-nat==v1.4.0a20251107
 Requires-Dist: redis~=4.3.4
 Dynamic: license-file
 

{nvidia_nat_redis-1.4.0a20251105.dist-info → nvidia_nat_redis-1.4.0a20251107.dist-info}/RECORD

@@ -1,15 +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=njBHSO5r4eT0NF8vWhKOVkelAO_ZTzoAZZM6b_DD-Ig,1837
+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=wRKB3HGfywoHp_XF2Rc4JqnAmJtGLb5usAfXeqNzINY,4402
+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.0a20251105.dist-info/licenses/LICENSE-3rd-party.txt,sha256=fOk5jMmCX9YoKWyYzTtfgl-SUy477audFC5hNY4oP7Q,284609
-nvidia_nat_redis-1.4.0a20251105.dist-info/licenses/LICENSE.md,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
-nvidia_nat_redis-1.4.0a20251105.dist-info/METADATA,sha256=v35m3K9xZktsOG1MjEcpAlppRBxdwcpM977wDYh1PUg,1894
-nvidia_nat_redis-1.4.0a20251105.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-nvidia_nat_redis-1.4.0a20251105.dist-info/entry_points.txt,sha256=nyS8t8L9CbRFIMlE70RQBtJXrflBP4Ltl5zAkIl44So,56
-nvidia_nat_redis-1.4.0a20251105.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
-nvidia_nat_redis-1.4.0a20251105.dist-info/RECORD,,
+nvidia_nat_redis-1.4.0a20251107.dist-info/licenses/LICENSE-3rd-party.txt,sha256=fOk5jMmCX9YoKWyYzTtfgl-SUy477audFC5hNY4oP7Q,284609
+nvidia_nat_redis-1.4.0a20251107.dist-info/licenses/LICENSE.md,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+nvidia_nat_redis-1.4.0a20251107.dist-info/METADATA,sha256=xmluwY3xsXLl46lGjsE4ybL5PlmS7rTwzlecSpK_xVc,1894
+nvidia_nat_redis-1.4.0a20251107.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+nvidia_nat_redis-1.4.0a20251107.dist-info/entry_points.txt,sha256=nyS8t8L9CbRFIMlE70RQBtJXrflBP4Ltl5zAkIl44So,56
+nvidia_nat_redis-1.4.0a20251107.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
+nvidia_nat_redis-1.4.0a20251107.dist-info/RECORD,,