From 179860863a7a0dbd4670619151460c1b2930ce6b Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Tue, 16 Jan 2024 19:42:34 -0800 Subject: [PATCH] (v0) fix --- litellm/proxy/proxy_server.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index b383317b3b6..0503d2aad84 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -507,7 +507,9 @@ async def track_cost_callback( if user_api_key and ( prisma_client is not None or custom_db_client is not None ): - await update_database(token=user_api_key, response_cost=response_cost) + await update_database( + token=user_api_key, response_cost=response_cost, user_id=user_id + ) elif kwargs["stream"] == False: # for non streaming responses response_cost = litellm.completion_cost( completion_response=completion_response