scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-12 23:01:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 182

Update spend logs writer to add organization_id

Browse source
This commit is contained in:
yuneng-jiang 2025-11-26 10:32:49 -08:00
parent 2794153012
commit 5b645d5f2f
1 changed files with 15 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
litellm/proxy/db/db_spend_update_writer.py
View file

@ -118,6 +118,21 @@ class DBSpendUpdateWriter:
payload["startTime"] = payload["startTime"].isoformat()
if isinstance(payload["endTime"], datetime):
payload["endTime"] = payload["endTime"].isoformat()
# Override organization_id and team_id from parameters if provided
# This ensures the correct values are set even if metadata is missing them
# Convert empty strings to None for optional fields to avoid database issues
if org_id is not None and org_id != "":
payload["organization_id"] = org_id
elif payload.get("organization_id") == "":
# Convert empty string to None for optional field
payload["organization_id"] = None
if team_id is not None and team_id != "":
payload["team_id"] = team_id
elif payload.get("team_id") == "":
# Convert empty string to None for optional field
payload["team_id"] = None
asyncio.create_task(
self._update_user_db(