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)