From 03b1f8f2334bc33772e886fbc4249d9526f1b1bc Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Mon, 10 Feb 2025 11:58:44 -0800 Subject: [PATCH] update RedisCluster creation --- litellm/_redis.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/litellm/_redis.py b/litellm/_redis.py index 70c38cf7f56..b0069ba7072 100644 --- a/litellm/_redis.py +++ b/litellm/_redis.py @@ -266,7 +266,9 @@ def get_redis_client(**env_overrides): return redis.Redis(**redis_kwargs) -def get_redis_async_client(**env_overrides) -> async_redis.Redis: +def get_redis_async_client( + **env_overrides, +) -> Union[async_redis.Redis, async_redis.RedisCluster]: redis_kwargs = _get_redis_client_logic(**env_overrides) if "url" in redis_kwargs and redis_kwargs["url"] is not None: args = _get_redis_url_kwargs(client=async_redis.Redis.from_url)