diff --git a/litellm/proxy/db/db_spend_update_writer.py b/litellm/proxy/db/db_spend_update_writer.py index 98c788b6fc4..a1838c3d4d1 100644 --- a/litellm/proxy/db/db_spend_update_writer.py +++ b/litellm/proxy/db/db_spend_update_writer.py @@ -1066,7 +1066,7 @@ class DBSpendUpdateWriter: team_id=team_id, user_id=user_id, prisma_client=prisma_client, - tokens=payload["total_tokens"], + tokens=payload["total_tokens"] if "total_tokens" in payload else 0, ) except Exception: verbose_proxy_logger.debug(