From 8d0fa9465ee611f4593b943fb7f2b927f1f96d39 Mon Sep 17 00:00:00 2001 From: Sricharan Anugandula Date: Wed, 9 Sep 2026 01:04:22 +0530 Subject: [PATCH] fix(proxy/auth): copy end_user_max_budget in update_valid_token_with_end_user_params --- litellm/proxy/auth/user_api_key_auth.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/litellm/proxy/auth/user_api_key_auth.py b/litellm/proxy/auth/user_api_key_auth.py index 93293db24c6..9f288128fce 100644 --- a/litellm/proxy/auth/user_api_key_auth.py +++ b/litellm/proxy/auth/user_api_key_auth.py @@ -563,6 +563,8 @@ def update_valid_token_with_end_user_params(valid_token: UserAPIKeyAuth, end_use valid_token.end_user_rpm_limit = end_user_params["end_user_rpm_limit"] if end_user_params.get("allowed_model_region") is not None: valid_token.allowed_model_region = end_user_params["allowed_model_region"] + if end_user_params.get("end_user_max_budget") is not None: + valid_token.end_user_max_budget = end_user_params["end_user_max_budget"] if end_user_params.get("end_user_model_max_budget") is not None: valid_token.end_user_model_max_budget = end_user_params["end_user_model_max_budget"] return valid_token