nvidia-nat-redis 1.4.0a20251028__py3-none-any.whl → 1.4.0a20260127__py3-none-any.whl

nat/meta/pypi.md

@@ -1,5 +1,5 @@
 <!--
-SPDX-FileCopyrightText: Copyright (c) 2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+SPDX-FileCopyrightText: Copyright (c) 2025-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 SPDX-License-Identifier: Apache-2.0
 
 Licensed under the Apache License, Version 2.0 (the "License");

nat/plugins/redis/memory.py

@@ -1,4 +1,4 @@
-# SPDX-FileCopyrightText: Copyright (c) 2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-FileCopyrightText: Copyright (c) 2025-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: Apache-2.0
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
@@ -14,10 +14,11 @@
 # limitations under the License.
 
 from pydantic import Field
-from pydantic import SecretStr
 
 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
 
@@ -26,7 +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: SecretStr | None = Field(default=None, description="Password for the Redis server")
+    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."))
@@ -42,14 +43,10 @@ async def redis_memory_client(config: RedisMemoryClientConfig, builder: Builder)
 
     from .schema import ensure_index_exists
 
-    redis_password = None
-    if config.password is not None:
-        redis_password = config.password.get_secret_value()
-
     redis_client = redis.Redis(host=config.host,
                                port=config.port,
                                db=config.db,
-                               password=redis_password,
+                               password=get_secret_value(config.password),
                                decode_responses=True,
                                socket_timeout=5.0,
                                socket_connect_timeout=5.0)

nat/plugins/redis/object_store.py

@@ -1,4 +1,4 @@
-# SPDX-FileCopyrightText: Copyright (c) 2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-FileCopyrightText: Copyright (c) 2025-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: Apache-2.0
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
@@ -14,30 +14,32 @@
 # limitations under the License.
 
 from pydantic import Field
-from pydantic import SecretStr
-from pydantic import field_serializer
+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: SecretStr | None = Field(default=None, description="The password for the Redis server")
-
-    @field_serializer('password')
-    def dump_secret(self, v: SecretStr | None) -> str | None:
-        if v is None:
-            return None
-        return v.get_secret_value()
+    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_editor.py

@@ -1,4 +1,4 @@
-# SPDX-FileCopyrightText: Copyright (c) 2024-2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-FileCopyrightText: Copyright (c) 2024-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: Apache-2.0
 #
 # Licensed under the Apache License, Version 2.0 (the "License");

nat/plugins/redis/redis_object_store.py

@@ -1,4 +1,4 @@
-# SPDX-FileCopyrightText: Copyright (c) 2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-FileCopyrightText: Copyright (c) 2025-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: Apache-2.0
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
@@ -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:

nat/plugins/redis/register.py

@@ -1,4 +1,4 @@
-# SPDX-FileCopyrightText: Copyright (c) 2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-FileCopyrightText: Copyright (c) 2025-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: Apache-2.0
 #
 # Licensed under the Apache License, Version 2.0 (the "License");

nat/plugins/redis/schema.py

@@ -1,4 +1,4 @@
-# SPDX-FileCopyrightText: Copyright (c) 2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-FileCopyrightText: Copyright (c) 2025-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: Apache-2.0
 #
 # Licensed under the Apache License, Version 2.0 (the "License");

{nvidia_nat_redis-1.4.0a20251028.dist-info → nvidia_nat_redis-1.4.0a20260127.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nvidia-nat-redis
-Version: 1.4.0a20251028
+Version: 1.4.0a20260127
 Summary: Subpackage for Redis integration in NeMo Agent toolkit
 Author: NVIDIA Corporation
 Maintainer: NVIDIA Corporation
@@ -16,12 +16,12 @@ 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.0a20251028
-Requires-Dist: redis~=4.3.4
+Requires-Dist: nvidia-nat==v1.4.0a20260127
+Requires-Dist: redis<5.0.0,>=4.3.4
 Dynamic: license-file
 
 <!--
-SPDX-FileCopyrightText: Copyright (c) 2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+SPDX-FileCopyrightText: Copyright (c) 2025-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 SPDX-License-Identifier: Apache-2.0
 
 Licensed under the Apache License, Version 2.0 (the "License");

nvidia_nat_redis-1.4.0a20260127.dist-info/RECORD

@@ -0,0 +1,15 @@
+nat/meta/pypi.md,sha256=BAi1_MjQHFjNWdFul67NeSrwTwBlzvsEu9rcXUKzOho,1095
+nat/plugins/redis/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+nat/plugins/redis/memory.py,sha256=Z_xxRi2mOxnVFwN3maoQNy6KMMJC_d8RrdiDxLLY8x0,2830
+nat/plugins/redis/object_store.py,sha256=YttUfeRg-ipo--Jz8iCKbAvRgG0YG-j-5ZGpYN2Jjoc,2238
+nat/plugins/redis/redis_editor.py,sha256=wdN4Oe-AAJtlYsjxwl0SAZITPulXAG0zOjCGRug-LeA,9746
+nat/plugins/redis/redis_object_store.py,sha256=9hjvhDtehTFoP-iz2PhHw8QL1jLd0ivwtRfoTeZOZy4,4672
+nat/plugins/redis/register.py,sha256=dk9o8B_k0QzUxU8JRxpAYyDfbh0xlENz3i0dzZuZI2Q,839
+nat/plugins/redis/schema.py,sha256=fOFwnXPYCMzzEoFN8KNkU2_gMBNF0RCmeRTqccxD_aM,5594
+nvidia_nat_redis-1.4.0a20260127.dist-info/licenses/LICENSE-3rd-party.txt,sha256=fOk5jMmCX9YoKWyYzTtfgl-SUy477audFC5hNY4oP7Q,284609
+nvidia_nat_redis-1.4.0a20260127.dist-info/licenses/LICENSE.md,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+nvidia_nat_redis-1.4.0a20260127.dist-info/METADATA,sha256=Pxgw-7g09zfUhAGIFprcPkdEsGrP43Hr_JoZ-QryfRc,1906
+nvidia_nat_redis-1.4.0a20260127.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+nvidia_nat_redis-1.4.0a20260127.dist-info/entry_points.txt,sha256=nyS8t8L9CbRFIMlE70RQBtJXrflBP4Ltl5zAkIl44So,56
+nvidia_nat_redis-1.4.0a20260127.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
+nvidia_nat_redis-1.4.0a20260127.dist-info/RECORD,,

{nvidia_nat_redis-1.4.0a20251028.dist-info → nvidia_nat_redis-1.4.0a20260127.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.9.0)
+Generator: setuptools (80.10.2)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

nvidia_nat_redis-1.4.0a20251028.dist-info/RECORD

@@ -1,15 +0,0 @@
-nat/meta/pypi.md,sha256=TpeNbVZJxzvEf0Gh3BGvLHPYsKnXjgM_KQVCayBPXso,1090
-nat/plugins/redis/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-nat/plugins/redis/memory.py,sha256=U6cFp0QYb2Peh5f_zNMqM_uPgPC-E3jU1dgxcoPDf1k,2854
-nat/plugins/redis/object_store.py,sha256=ZcNFG6Hb2V2zA-5sRS2US18lkTwEoPV4xAce4gHsVHI,2031
-nat/plugins/redis/redis_editor.py,sha256=nkSIWi1HUPald088fXTuF0rmZ0uS_3V65Vxy20vLSgk,9746
-nat/plugins/redis/redis_object_store.py,sha256=wRKB3HGfywoHp_XF2Rc4JqnAmJtGLb5usAfXeqNzINY,4402
-nat/plugins/redis/register.py,sha256=dJBKi-7W72ipkmZTOIo1E3ETffmJIlYhQTOlrkiFH3A,834
-nat/plugins/redis/schema.py,sha256=Zcas3hIIqG7wuR94baYRFycmnccB3CuGmTs4p8Vv4mA,5589
-nvidia_nat_redis-1.4.0a20251028.dist-info/licenses/LICENSE-3rd-party.txt,sha256=fOk5jMmCX9YoKWyYzTtfgl-SUy477audFC5hNY4oP7Q,284609
-nvidia_nat_redis-1.4.0a20251028.dist-info/licenses/LICENSE.md,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
-nvidia_nat_redis-1.4.0a20251028.dist-info/METADATA,sha256=3Xc_Oke0BtPl0GspsqYCd9u3nFJ3k6lHJWJcUzjetI4,1894
-nvidia_nat_redis-1.4.0a20251028.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-nvidia_nat_redis-1.4.0a20251028.dist-info/entry_points.txt,sha256=nyS8t8L9CbRFIMlE70RQBtJXrflBP4Ltl5zAkIl44So,56
-nvidia_nat_redis-1.4.0a20251028.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
-nvidia_nat_redis-1.4.0a20251028.dist-info/RECORD,,