phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 10:44:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: fix linting error

Browse source
This commit is contained in:
Krrish Dholakia 2025-04-11 17:19:08 -07:00
parent 2ec2151537
commit 0a07e26e13
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
litellm/proxy/utils.py
View file

@ -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