From ca82761e4843e110d4192d586731f812d2eb3d60 Mon Sep 17 00:00:00 2001 From: oss-agent-shin Date: Wed, 13 May 2026 13:28:22 -0700 Subject: [PATCH] docs(budget_manager): add docstring to BudgetManager.reset_cost (#27867) Co-authored-by: oss-agent-shin <279349115+oss-agent-shin@users.noreply.github.com> --- litellm/budget_manager.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/litellm/budget_manager.py b/litellm/budget_manager.py index b25967579e..bbebb6042c 100644 --- a/litellm/budget_manager.py +++ b/litellm/budget_manager.py @@ -178,6 +178,18 @@ class BudgetManager: return list(self.user_dict.keys()) def reset_cost(self, user): + """ + Reset the tracked spend for a user back to zero. + + Clears both the aggregate ``current_cost`` and the per-model + ``model_cost`` breakdown stored for the given user. + + Args: + user: The user identifier whose cost should be reset. + + Returns: + dict: ``{"user": }`` reflecting the reset state. + """ self.user_dict[user]["current_cost"] = 0 self.user_dict[user]["model_cost"] = {} return {"user": self.user_dict[user]}