scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-08 03:08:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 114

fix(proxy): include tag budgets in the idempotent booking transaction

Browse source
This commit is contained in:
todayim 2026-08-06 01:07:18 +08:00
parent 73f31beb0d
commit d4fb8e9a1b
1 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
litellm/proxy/spend_tracking/usage_ingestion_endpoints.py
View file

@ -237,7 +237,9 @@ async def reserve_spend_log_atomic(
if ProxyUpdateSpend.disable_spend_updates() is True:
return "disabled"
payload: Final = prisma_client.jsonify_object(build_spend_log_payload(record, request_id, hashed_token, key, cost))
spend_payload: Final = build_spend_log_payload(record, request_id, hashed_token, key, cost)
payload: Final = prisma_client.jsonify_object(spend_payload)
request_tags: Final = spend_payload.get("request_tags")
from prisma.errors import UniqueViolationError
writer: Final = proxy_logging_obj.db_spend_update_writer
@ -258,6 +260,7 @@ async def reserve_spend_log_atomic(
response_cost=cost, team_id=key.team_id, user_id=key.user_id, prisma_client=shim
)
await writer._update_org_db(response_cost=cost, org_id=key.organization_id, prisma_client=shim)
await writer._update_tag_db(response_cost=cost, request_tags=request_tags, prisma_client=shim)
except UniqueViolationError:
return "duplicate"
return "reserved"