diff --git a/litellm/tests/test_prometheus_service.py b/litellm/tests/test_prometheus_service.py index ec2ffb5eb64..1f562aa71ba 100644 --- a/litellm/tests/test_prometheus_service.py +++ b/litellm/tests/test_prometheus_service.py @@ -65,13 +65,14 @@ async def test_completion_with_caching_bad_call(): - Assert failure callback gets called """ litellm.set_verbose = True - sl = ServiceLogging(mock_testing=True) + try: from litellm.caching import RedisCache litellm.service_callback = ["prometheus_system"] + sl = ServiceLogging(mock_testing=True) - RedisCache(host="hello-world", **{"service_logger_obj": sl}) + RedisCache(host="hello-world", service_logger_obj=sl) except Exception as e: print(f"Receives exception = {str(e)}") diff --git a/litellm/tests/test_router_caching.py b/litellm/tests/test_router_caching.py index 3bf68595eeb..ebace161c98 100644 --- a/litellm/tests/test_router_caching.py +++ b/litellm/tests/test_router_caching.py @@ -33,7 +33,7 @@ async def test_router_async_caching_with_ssl_url(): "rpm": 10000, }, ], - redis_url=os.getenv("REDIS_URL"), + redis_url=os.getenv("REDIS_SSL_URL"), ) response = await router.cache.redis_cache.ping() @@ -60,7 +60,7 @@ def test_router_sync_caching_with_ssl_url(): "rpm": 10000, }, ], - redis_url=os.getenv("REDIS_URL"), + redis_url=os.getenv("REDIS_SSL_URL"), ) response = router.cache.redis_cache.sync_ping()