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

nat/plugins/redis/memory.py

@@ -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

@@ -14,11 +14,10 @@
 # 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
+from nat.data_models.common import OptionalSecretStr
 from nat.data_models.object_store import ObjectStoreBaseConfig
 
 
@@ -31,13 +30,7 @@ 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()
+    password: OptionalSecretStr = Field(default=None, description="The password for the Redis server")
 
 
 @register_object_store(config_type=RedisObjectStoreClientConfig)

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

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

{nvidia_nat_redis-1.4.0a20251028.dist-info → nvidia_nat_redis-1.4.0a20251030.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=U6cFp0QYb2Peh5f_zNMqM_uPgPC-E3jU1dgxcoPDf1k,2854
-nat/plugins/redis/object_store.py,sha256=ZcNFG6Hb2V2zA-5sRS2US18lkTwEoPV4xAce4gHsVHI,2031
+nat/plugins/redis/memory.py,sha256=up25g7p7EXUcwAvMgCJjzyeGxH0454y9rE6QPPPYjK4,2825
+nat/plugins/redis/object_store.py,sha256=njBHSO5r4eT0NF8vWhKOVkelAO_ZTzoAZZM6b_DD-Ig,1837
 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.0a20251030.dist-info/licenses/LICENSE-3rd-party.txt,sha256=fOk5jMmCX9YoKWyYzTtfgl-SUy477audFC5hNY4oP7Q,284609
+nvidia_nat_redis-1.4.0a20251030.dist-info/licenses/LICENSE.md,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+nvidia_nat_redis-1.4.0a20251030.dist-info/METADATA,sha256=AeHsvFFRjYpVQkuvTmIsUO5YlR8krltsv8L7tQHcWJY,1894
+nvidia_nat_redis-1.4.0a20251030.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+nvidia_nat_redis-1.4.0a20251030.dist-info/entry_points.txt,sha256=nyS8t8L9CbRFIMlE70RQBtJXrflBP4Ltl5zAkIl44So,56
+nvidia_nat_redis-1.4.0a20251030.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
+nvidia_nat_redis-1.4.0a20251030.dist-info/RECORD,,