phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

(v0) fix

Browse source
This commit is contained in:
ishaan-jaff 2024-01-16 19:42:34 -08:00
parent d1897ca156
commit 179860863a
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

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