scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-11 22:51:28 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 123

test(proxy): use deployment listing metadata in alias coverage

Browse source
This commit is contained in:
Hayden Moulds 2026-09-11 12:23:48 +10:00
parent 2f8f3d3d21
commit 5b9244105c
No known key found for this signature in database
1 changed files with 10 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
tests/test_litellm/proxy/proxy_server/test_team_model_name_translation.py
View file

@ -1109,7 +1109,11 @@ async def test_v1_models_team_alias_inherits_token_limits_and_chat_mode(monkeypa
router.get_model_names.return_value = ["model_name_teamX_terra_uuid"]
router.get_model_access_groups.return_value = {"grp-a": ["model_name_teamX_terra_uuid"]}
router.get_fully_blocked_model_names.return_value = set()
router.get_configured_token_limits.return_value = (876000, 128000)
router.get_model_listing_info.return_value = DeploymentModelListingInfo(
cost_map_keys=("azure/gpt-4.1",),
max_input_tokens=876000,
max_output_tokens=128000,
)
router.get_configured_mode.return_value = "chat"
router.model_list = [team_dep]
router.get_model_list.return_value = [team_dep]
@ -1153,7 +1157,11 @@ async def test_v1_models_team_image_alias_inherits_image_generation_mode(monkeyp
router.get_model_names.return_value = ["model_name_teamX_image_uuid"]
router.get_model_access_groups.return_value = {"grp-a": ["model_name_teamX_image_uuid"]}
router.get_fully_blocked_model_names.return_value = set()
router.get_configured_token_limits.return_value = (None, None)
router.get_model_listing_info.return_value = DeploymentModelListingInfo(
cost_map_keys=("openai/gpt-image-1",),
max_input_tokens=None,
max_output_tokens=None,
)
router.get_configured_mode.return_value = "image_generation"
router.model_list = [team_dep]
router.get_model_list.return_value = [team_dep]