diff --git a/litellm/proxy/hooks/proxy_track_cost_callback.py b/litellm/proxy/hooks/proxy_track_cost_callback.py index abe0d235af3..68ebd522cd2 100644 --- a/litellm/proxy/hooks/proxy_track_cost_callback.py +++ b/litellm/proxy/hooks/proxy_track_cost_callback.py @@ -713,7 +713,10 @@ async def _update_database_and_spend_counters( return True -async def _reconcile_budget_reservation_before_db_update(budget_reservation: dict, response_cost: float) -> None: +async def _reconcile_budget_reservation_before_db_update( + budget_reservation: dict, # mutable-ok: reconcile_budget_reservation stamps applied_adjustment on the caller's shared reservation dict + response_cost: float, +) -> None: from litellm.proxy.spend_tracking.budget_reservation import reconcile_budget_reservation try: