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

nat/plugins/redis/memory.py

@@ -14,6 +14,7 @@
 # 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
@@ -25,6 +26,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")
     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."))
@@ -40,9 +42,14 @@ 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,
                                decode_responses=True,
                                socket_timeout=5.0,
                                socket_connect_timeout=5.0)

nat/plugins/redis/object_store.py

@@ -14,6 +14,8 @@
 # limitations under the License.
 
 from pydantic import Field
+from pydantic import SecretStr
+from pydantic import field_serializer
 
 from nat.builder.builder import Builder
 from nat.cli.register_workflow import register_object_store
@@ -29,6 +31,13 @@ class RedisObjectStoreClientConfig(ObjectStoreBaseConfig, name="redis"):
     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()
 
 
 @register_object_store(config_type=RedisObjectStoreClientConfig)

nat/plugins/redis/redis_object_store.py

@@ -33,7 +33,7 @@ class RedisObjectStore(ObjectStore):
     Each object is stored as a single binary value at key "nat/object_store/{bucket_name}/{object_key}".
     """
 
-    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):
 
         super().__init__()
 
@@ -41,6 +41,7 @@ class RedisObjectStore(ObjectStore):
         self._host = host
         self._port = port
         self._db = db
+        self._password = password
         self._client: redis.Redis | None = None
 
     async def __aenter__(self) -> "RedisObjectStore":
@@ -52,6 +53,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,
         )

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nvidia-nat-redis
-Version: 1.4.0a20251023
+Version: 1.4.0a20251028
 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.0a20251023
+Requires-Dist: nvidia-nat==v1.4.0a20251028
 Requires-Dist: redis~=4.3.4
 Dynamic: license-file
 

nvidia_nat_redis-1.4.0a20251028.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=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,,

nvidia_nat_redis-1.4.0a20251023.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=wqj_UYqnllc4kTCtFDz3eu-OZnyPWXxNZ3e6G992OGQ,2546
-nat/plugins/redis/object_store.py,sha256=f_GtCTZ3KHfxE4f0lAzKaoAoInAZZki4Dfo7hrKCAHA,1681
-nat/plugins/redis/redis_editor.py,sha256=nkSIWi1HUPald088fXTuF0rmZ0uS_3V65Vxy20vLSgk,9746
-nat/plugins/redis/redis_object_store.py,sha256=DX46GEQl4H1Ivf2wrRaSimNcq6EfvRsm-xhyiECudoQ,4302
-nat/plugins/redis/register.py,sha256=dJBKi-7W72ipkmZTOIo1E3ETffmJIlYhQTOlrkiFH3A,834
-nat/plugins/redis/schema.py,sha256=Zcas3hIIqG7wuR94baYRFycmnccB3CuGmTs4p8Vv4mA,5589
-nvidia_nat_redis-1.4.0a20251023.dist-info/licenses/LICENSE-3rd-party.txt,sha256=fOk5jMmCX9YoKWyYzTtfgl-SUy477audFC5hNY4oP7Q,284609
-nvidia_nat_redis-1.4.0a20251023.dist-info/licenses/LICENSE.md,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
-nvidia_nat_redis-1.4.0a20251023.dist-info/METADATA,sha256=Md3PEWPowCCPp1NXc5494cPg4pKF2atc9Qz8IounRfI,1894
-nvidia_nat_redis-1.4.0a20251023.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-nvidia_nat_redis-1.4.0a20251023.dist-info/entry_points.txt,sha256=nyS8t8L9CbRFIMlE70RQBtJXrflBP4Ltl5zAkIl44So,56
-nvidia_nat_redis-1.4.0a20251023.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
-nvidia_nat_redis-1.4.0a20251023.dist-info/RECORD,,