From 3e729eade3f3de17d321f48f048062244f8aaf7e Mon Sep 17 00:00:00 2001 From: Sameer Kankute Date: Wed, 24 Jun 2026 23:03:50 +0530 Subject: [PATCH] revert(auth): remove max_ui_session_budget from CLI session token max_ui_session_budget defaults to $0.25 and is sized for the UI chat pane (10-min sessions). CLI sessions are 24-hour tokens for real work; capping them at that ceiling would throttle users under their actual user/team budget. Budget enforcement for CLI sessions is via the shared user and team counters as originally intended. --- litellm/proxy/auth/auth_checks.py | 1 - tests/test_litellm/proxy/auth/test_auth_checks.py | 12 ------------ 2 files changed, 13 deletions(-) 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"""