From a35bab276d4f4231d42443f19e76d77e1160d8ba Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Fri, 28 Feb 2025 08:58:29 -0800 Subject: [PATCH] basic logging errors in DB --- litellm/proxy/hooks/proxy_track_cost_callback.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/litellm/proxy/hooks/proxy_track_cost_callback.py b/litellm/proxy/hooks/proxy_track_cost_callback.py index 1a7659fe01d..88ceef1d06b 100644 --- a/litellm/proxy/hooks/proxy_track_cost_callback.py +++ b/litellm/proxy/hooks/proxy_track_cost_callback.py @@ -28,11 +28,19 @@ class _ProxyDBLogger(CustomLogger): original_exception: Exception, user_api_key_dict: UserAPIKeyAuth, ): - await self._PROXY_track_cost_callback( + from litellm.proxy.proxy_server import update_database + + await update_database( + token=user_api_key_dict.api_key, + response_cost=0.0, + user_id=user_api_key_dict.user_id, + end_user_id=user_api_key_dict.end_user_id, + team_id=user_api_key_dict.team_id, kwargs=request_data, completion_response=original_exception, start_time=datetime.now(), end_time=datetime.now(), + org_id=user_api_key_dict.org_id, ) @log_db_metrics