From 5b645d5f2f7dfa6d5b398048e171749f4fee77a8 Mon Sep 17 00:00:00 2001 From: yuneng-jiang Date: Wed, 26 Nov 2025 10:32:49 -0800 Subject: [PATCH] Update spend logs writer to add organization_id --- litellm/proxy/db/db_spend_update_writer.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/litellm/proxy/db/db_spend_update_writer.py b/litellm/proxy/db/db_spend_update_writer.py index f404c640a08..b6f98393700 100644 --- a/litellm/proxy/db/db_spend_update_writer.py +++ b/litellm/proxy/db/db_spend_update_writer.py @@ -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(