phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

add coverage

Browse source
This commit is contained in:
Ishaan Jaff 2024-11-25 15:48:40 -08:00
parent a18aeaa2fb
commit 632283a371
1 changed files with 3 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

3
tests/proxy_unit_tests/test_user_api_key_auth.py
View file

@ -409,6 +409,8 @@ def test_is_api_route_allowed(route, user_role, expected_result):
("custom_engine/model-123", ["custom_engine/*"], True),
("custom_engine/model-123", ["custom_engine/*", "azure/*"], True),
("custom-engine/model-123", ["custom_engine/*", "azure/*"], False),
("openai/gpt-4o", ["openai/*"], True),
("openai/gpt-12", ["openai/*"], True),
("gpt-4", ["gpt-*"], True),
("gpt-4", ["claude-*"], False),
# Mixed scenarios
@ -430,6 +432,7 @@ def test_model_is_within_list_of_allowed_models(model, allowed_models, expected_
("gpt-4", ["gpt-*"], True),
("azure/gpt-4", ["azure/*"], True),
("custom_engine/model-123", ["custom_engine/*"], True),
("openai/my-fake-model", ["openai/*"], True),
("custom_engine/model-123", ["custom_engine/*", "azure/*"], True),
("custom-engine/model-123", ["custom_engine/*", "azure/*"], False),
# Multiple patterns