From 0a07e26e13e8b27d3b8e237a20890de4a89a0c2a Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Fri, 11 Apr 2025 17:19:08 -0700 Subject: [PATCH] fix: fix linting error --- litellm/proxy/utils.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/litellm/proxy/utils.py b/litellm/proxy/utils.py index d64cc9b5bc..fed47b3387 100644 --- a/litellm/proxy/utils.py +++ b/litellm/proxy/utils.py @@ -353,7 +353,7 @@ class ProxyLogging: self.db_spend_update_writer.redis_update_buffer.redis_cache = redis_cache self.db_spend_update_writer.pod_lock_manager.redis_cache = redis_cache - self.proxy_hook_mapping = {} + self.proxy_hook_mapping: Dict[str, CustomLogger] = {} def _add_proxy_hooks(self, llm_router: Optional[Router] = None): """ @@ -371,8 +371,8 @@ class ProxyLogging: passed_in_args["internal_usage_cache"] = self.internal_usage_cache if "prisma_client" in expected_args: passed_in_args["prisma_client"] = prisma_client - proxy_hook_obj = proxy_hook(**passed_in_args) - litellm.logging_callback_manager.add_litellm_callback(proxy_hook_obj) # type: ignore + proxy_hook_obj = cast(CustomLogger, proxy_hook(**passed_in_args)) + litellm.logging_callback_manager.add_litellm_callback(proxy_hook_obj) self.proxy_hook_mapping[hook] = proxy_hook_obj