phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 10:44:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix check_if_token_is_service_account

Browse source
This commit is contained in:
Ishaan Jaff 2025-04-01 07:36:35 -07:00
parent b0fa934fe3
commit f8ce30763e
4 changed files with 57 additions and 63 deletions
Download patch file Download diff file Expand all files Collapse all files

19
litellm/proxy/litellm_pre_call_utils.py
View file

@ -747,7 +747,10 @@ def _get_enforced_params(
enforced_params: Optional[list] = None
if general_settings is not None:
enforced_params = general_settings.get("enforced_params")
if "service_account_settings" in general_settings:
if (
"service_account_settings" in general_settings
and check_if_token_is_service_account(user_api_key_dict) is True
):
service_account_settings = general_settings["service_account_settings"]
if "enforced_params" in service_account_settings:
if enforced_params is None:
@ -760,6 +763,20 @@ def _get_enforced_params(
return enforced_params
def check_if_token_is_service_account(valid_token: UserAPIKeyAuth) -> bool:
"""
Checks if the token is a service account
Returns:
bool: True if token is a service account
"""
if valid_token.metadata:
if "service_account_id" in valid_token.metadata:
return True
return False
def _enforced_params_check(
request_body: dict,
general_settings: Optional[dict],