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

test_get_enforced_params_for_service_account_settings

Browse source
This commit is contained in:
Ishaan Jaff 2025-04-01 08:39:41 -07:00
parent 01bd418e0b
commit f805e15f7b
1 changed files with 8 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

14
tests/litellm/proxy/test_litellm_pre_call_utils.py
View file

@ -44,18 +44,20 @@ def test_get_enforced_params_for_service_account_settings():
service_account_token = UserAPIKeyAuth(
api_key="test-key", metadata={"service_account_id": "test-service-account"}
)
general_settings_with_service = {
"enforced_params": ["user"],
general_settings_with_service_account_settings = {
"service_account_settings": {"enforced_params": ["metadata.service"]},
}
result = _get_enforced_params(
general_settings=general_settings_with_service,
general_settings=general_settings_with_service_account_settings,
user_api_key_dict=service_account_token,
)
assert result == ["user", "metadata.service"]
assert result == ["metadata.service"]
regular_token = UserAPIKeyAuth(api_key="test-key", metadata={})
regular_token = UserAPIKeyAuth(
api_key="test-key", metadata={"enforced_params": ["user"]}
)
result = _get_enforced_params(
general_settings=general_settings_with_service, user_api_key_dict=regular_token
general_settings=general_settings_with_service_account_settings,
user_api_key_dict=regular_token,
)
assert result == ["user"]