diff --git a/litellm/proxy/auth/master_key_policy.py b/litellm/proxy/auth/master_key_policy.py index 1fec63b43fd..20b66f15f67 100644 --- a/litellm/proxy/auth/master_key_policy.py +++ b/litellm/proxy/auth/master_key_policy.py @@ -138,7 +138,7 @@ def master_key_lockout_action( return None -def master_key_lockout_exception(action: LockoutAction) -> ProxyException: +def master_key_lockout_exception() -> ProxyException: return ProxyException( message=MASTER_KEY_LOCKOUT_MESSAGE, type=ProxyErrorTypes.auth_error, diff --git a/litellm/proxy/auth/user_api_key_auth.py b/litellm/proxy/auth/user_api_key_auth.py index f49c7272921..6ba0e26f6cc 100644 --- a/litellm/proxy/auth/user_api_key_auth.py +++ b/litellm/proxy/auth/user_api_key_auth.py @@ -1354,15 +1354,13 @@ async def _user_api_key_auth_builder( _lockout_reason: Final = insecure_master_key_reason( master_key, alternative_auth_enabled=alternative_auth_enabled(general_settings) ) - if _lockout_reason is not None: - _lockout_action: Final = master_key_lockout_action( - route=route, - method=request.method, - reason=_lockout_reason, - stored_credentials_present=stored_credentials_present(), - ) - if _lockout_action is not None: - raise master_key_lockout_exception(_lockout_action) + if _lockout_reason is not None and master_key_lockout_action( + route=route, + method=request.method, + reason=_lockout_reason, + stored_credentials_present=stored_credentials_present(), + ) is not None: + raise master_key_lockout_exception() with tracer.trace("litellm.proxy.auth.pre_db_read_auth_checks"): await pre_db_read_auth_checks( request_data=request_data,