From d86a7c3702f3b9c4a57080b20ef77bd16bfc2229 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Sat, 23 Nov 2024 16:52:45 -0800 Subject: [PATCH] fix code quality check --- litellm/router_strategy/provider_budgets.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/litellm/router_strategy/provider_budgets.py b/litellm/router_strategy/provider_budgets.py index 42e63b297..65b1e3c97 100644 --- a/litellm/router_strategy/provider_budgets.py +++ b/litellm/router_strategy/provider_budgets.py @@ -279,7 +279,7 @@ class ProviderBudgetLimiting(CustomLogger): if isinstance(redis_values, dict): # Check if redis_values is a dictionary for key, value in redis_values.items(): if value is not None: - self.router_cache.in_memory_cache.set_cache( + await self.router_cache.in_memory_cache.async_set_cache( key=key, value=float(value) ) verbose_router_logger.debug( @@ -287,9 +287,6 @@ class ProviderBudgetLimiting(CustomLogger): ) except Exception as e: - import traceback - - traceback.print_exc() verbose_router_logger.error( f"Error syncing in-memory cache with Redis: {str(e)}" )