phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix redis sentinel - use sentinel password

Browse source
This commit is contained in:
Ishaan Jaff 2024-10-11 23:55:14 +05:30
parent 671ac7c043
commit 4506fa952b
1 changed files with 3 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

8
litellm/_redis.py
View file

@ -225,10 +225,6 @@ def _init_async_redis_sentinel(redis_kwargs) -> async_redis.Redis:
sentinel_password = redis_kwargs.get("sentinel_password") sentinel_password = redis_kwargs.get("sentinel_password")
service_name = redis_kwargs.get("service_name") service_name = redis_kwargs.get("service_name")
sentinel_kwargs: Optional[Dict] = None
if sentinel_password:
sentinel_kwargs = {"password": sentinel_password}
if not sentinel_nodes or not service_name: if not sentinel_nodes or not service_name:
raise ValueError( raise ValueError(
"Both 'sentinel_nodes' and 'service_name' are required for Redis Sentinel." "Both 'sentinel_nodes' and 'service_name' are required for Redis Sentinel."
@ -238,7 +234,9 @@ def _init_async_redis_sentinel(redis_kwargs) -> async_redis.Redis:
# Set up the Sentinel client # Set up the Sentinel client
sentinel = async_redis.Sentinel( sentinel = async_redis.Sentinel(
sentinel_nodes, socket_timeout=0.1, sentinel_kwargs=sentinel_kwargs sentinel_nodes,
socket_timeout=0.1,
password=sentinel_password,
) )
# Return the master instance for the given service # Return the master instance for the given service