From 4ebb4570be37076e3eaacae56a4882072f45740c Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Mon, 1 Apr 2024 17:08:48 -0700 Subject: [PATCH] (fix) handle urls correctly --- litellm/caching.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/litellm/caching.py b/litellm/caching.py index 75c9f0d09..301acd593 100644 --- a/litellm/caching.py +++ b/litellm/caching.py @@ -109,6 +109,7 @@ class RedisCache(BaseCache): **kwargs, ): from ._redis import get_redis_client, get_redis_connection_pool + import redis redis_kwargs = {} if host is not None: @@ -122,6 +123,14 @@ class RedisCache(BaseCache): self.redis_client = get_redis_client(**redis_kwargs) self.redis_kwargs = redis_kwargs self.async_redis_conn_pool = get_redis_connection_pool(**redis_kwargs) + + if "url" in redis_kwargs and redis_kwargs["url"] is not None: + parsed_kwargs = redis.connection.parse_url(redis_kwargs["url"]) + redis_kwargs.update(parsed_kwargs) + self.redis_kwargs.update(parsed_kwargs) + # pop url + self.redis_kwargs.pop("url") + # redis namespaces self.namespace = namespace # for high traffic, we store the redis results in memory and then batch write to redis