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

add sentinel_password support

Browse source
This commit is contained in:
Ishaan Jaff 2024-10-10 19:11:36 +05:30
parent 1a9d9e1cad
commit 68fab2bf5d
1 changed files with 9 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

11
litellm/_redis.py
View file

@ -12,7 +12,7 @@ import json
# s/o [@Frank Colson](https://www.linkedin.com/in/frank-colson-422b9b183/) for this redis implementation # s/o [@Frank Colson](https://www.linkedin.com/in/frank-colson-422b9b183/) for this redis implementation
import os import os
from typing import List, Optional, Union from typing import Dict, List, Optional, Union
import redis # type: ignore import redis # type: ignore
import redis.asyncio as async_redis # type: ignore import redis.asyncio as async_redis # type: ignore
@ -215,8 +215,13 @@ def _init_redis_sentinel(redis_kwargs) -> redis.Redis:
def _init_async_redis_sentinel(redis_kwargs) -> async_redis.Redis: def _init_async_redis_sentinel(redis_kwargs) -> async_redis.Redis:
sentinel_nodes = redis_kwargs.get("sentinel_nodes") sentinel_nodes = redis_kwargs.get("sentinel_nodes")
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."
@ -225,7 +230,9 @@ def _init_async_redis_sentinel(redis_kwargs) -> async_redis.Redis:
verbose_logger.debug("init_redis_sentinel: sentinel nodes are being initialized.") verbose_logger.debug("init_redis_sentinel: sentinel nodes are being initialized.")
# Set up the Sentinel client # Set up the Sentinel client
sentinel = async_redis.Sentinel(sentinel_nodes, socket_timeout=0.1) sentinel = async_redis.Sentinel(
sentinel_nodes, socket_timeout=0.1, sentinel_kwargs=sentinel_kwargs
)
# Return the master instance for the given service # Return the master instance for the given service