fix(proxy/auth): copy end_user_max_budget in update_valid_token_with_end_user_params

This commit is contained in:
Sricharan Anugandula 2026-09-09 01:04:22 +05:30
parent 168a0055a2
commit 8d0fa9465e

View file

@ -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