From 9c573cda0a80d3a25f6d5b2afa3732f977b6d757 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Wed, 4 Sep 2024 10:59:15 -0700 Subject: [PATCH] fix test_proxy_logging_setup --- litellm/tests/test_caching.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/litellm/tests/test_caching.py b/litellm/tests/test_caching.py index 660cf68ea95..a272d2dcf3b 100644 --- a/litellm/tests/test_caching.py +++ b/litellm/tests/test_caching.py @@ -1992,7 +1992,8 @@ async def test_proxy_logging_setup(): """ Assert always_read_redis is True when used by internal usage cache """ + from litellm.caching import DualCache from litellm.proxy.utils import ProxyLogging - pl_obj = ProxyLogging() + pl_obj = ProxyLogging(user_api_key_cache=DualCache()) assert pl_obj.internal_usage_cache.always_read_redis is True