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

Litellm enforce enterprise features (#7357)

Browse source 
* fix(proxy_server.py): enforce team id based model add only works if enterprise user

* fix(auth_checks.py): enforce common_checks can only be imported by user_api_key_auth.py

* fix(auth_checks.py): insert not premium user error message on failed common checks run
This commit is contained in:
Krish Dholakia 2024-12-21 19:14:13 -08:00 • committed by GitHub
parent 152056375a
commit ae7f54498f
3 changed files with 62 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

25
tests/local_testing/test_auth_checks.py
View file

@ -199,3 +199,28 @@ async def test_can_team_call_model(model, expect_to_work):
assert model_in_access_group(**args)
else:
assert not model_in_access_group(**args)
def test_common_checks_import():
"""
Enforce that common_checks can only be imported by the 'user_api_key_auth()' function.
"""
try:
from litellm.proxy.auth.user_api_key_auth import common_checks
from litellm.proxy._types import CommonProxyErrors
common_checks(
request_body={},
team_object=None,
user_object=None,
end_user_object=None,
global_proxy_spend=None,
general_settings={},
route="",
llm_router=None,
)
pytest.fail(
"common_checks can only be imported by the 'user_api_key_auth()' function."
)
except Exception as e:
assert CommonProxyErrors.not_premium_user.value in str(e)