From 6a8d518e441815aaa2b7baa5a8896a39fc1ee204 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Wed, 10 Jan 2024 22:15:01 +0530 Subject: [PATCH] test(test_lowest_latency_routing.py): use the correct cache key --- litellm/tests/test_lowest_latency_routing.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/litellm/tests/test_lowest_latency_routing.py b/litellm/tests/test_lowest_latency_routing.py index 7fa1b031f63..7d93f0e6089 100644 --- a/litellm/tests/test_lowest_latency_routing.py +++ b/litellm/tests/test_lowest_latency_routing.py @@ -91,7 +91,8 @@ def test_latency_updated_custom_ttl(): start_time=start_time, end_time=end_time, ) - latency_key = f"{model_group}_latency_map" + latency_key = f"{model_group}_map" + print(f"cache: {test_cache.get_cache(key=latency_key)}") assert isinstance(test_cache.get_cache(key=latency_key), dict) time.sleep(cache_time) assert test_cache.get_cache(key=latency_key) is None