support connection args on async redis store

Author: tylerhutcherson

langgraph/store/redis/aio.py

@@ -65,6 +65,7 @@ def __init__(
         *,
         redis_client: Optional[AsyncRedis] = None,
         index: Optional[IndexConfig] = None,
+        connection_args: Optional[dict[str, Any]] = None,
     ) -> None:
         """Initialize store with Redis connection and optional index config."""
         if redis_url is None and redis_client is None:
@@ -94,7 +95,11 @@ def __init__(
             ]
 
         # Configure client
-        self.configure_client(redis_url=redis_url, redis_client=redis_client)
+        self.configure_client(
+            redis_url=redis_url,
+            redis_client=redis_client,
+            connection_args=connection_args or {}
+        )
 
         # Create store index
         self.store_index = AsyncSearchIndex.from_dict(
@@ -149,11 +154,12 @@ def configure_client(
         self,
         redis_url: Optional[str] = None,
         redis_client: Optional[AsyncRedis] = None,
+        connection_args: Optional[dict[str, Any]] = None,
     ) -> None:
         """Configure the Redis client."""
         self._owns_its_client = redis_client is None
         self._redis = redis_client or RedisConnectionFactory.get_async_redis_connection(
-            redis_url
+            redis_url, **connection_args
         )
 
     async def setup(self) -> None: