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

(v0) fix

Browse source
This commit is contained in:
ishaan-jaff 2024-01-16 19:42:34 -08:00
parent 4c69efac8a
commit a9338d22eb
1 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/proxy/proxy_server.py
View file

@ -506,7 +506,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