diff --git a/litellm/proxy/auth/auth_checks.py b/litellm/proxy/auth/auth_checks.py index b7f8a916b30..60a40b3ede9 100644 --- a/litellm/proxy/auth/auth_checks.py +++ b/litellm/proxy/auth/auth_checks.py @@ -2462,7 +2462,6 @@ class ExperimentalUIJWTToken: key_name=session_alias, key_alias=session_alias, expires=expires, - max_budget=litellm.max_ui_session_budget, user_id=user_info.user_id, team_id=_team_id, team_alias=team_alias, diff --git a/tests/test_litellm/proxy/auth/test_auth_checks.py b/tests/test_litellm/proxy/auth/test_auth_checks.py index 4721bf346c9..6d343af5b15 100644 --- a/tests/test_litellm/proxy/auth/test_auth_checks.py +++ b/tests/test_litellm/proxy/auth/test_auth_checks.py @@ -538,18 +538,6 @@ def test_get_cli_jwt_auth_token_unique_per_session(valid_sso_user_defined_values assert first["key_name"] == second["key_name"] == expected_alias -def test_get_cli_jwt_auth_token_enforces_max_ui_session_budget( - valid_sso_user_defined_values, -): - token = ExperimentalUIJWTToken.get_cli_jwt_auth_token(valid_sso_user_defined_values) - - decrypted = decrypt_value_helper(token, key="ui_hash_key", exception_type="debug") - assert decrypted is not None - token_data = json.loads(decrypted) - - assert token_data.get("max_budget") == litellm.max_ui_session_budget - - @pytest.mark.asyncio async def test_default_internal_user_params_with_get_user_object(monkeypatch): """Test that default_internal_user_params is used when creating a new user via get_user_object"""