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 154

test(proxy): cover team alias retrieve metadata

Browse source
This commit is contained in:
Hayden Moulds 2026-09-10 16:28:48 +10:00
parent 49809a814b
commit e664500003
No known key found for this signature in database
1 changed files with 10 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -27,6 +27,7 @@ from litellm.proxy.proxy_server import (
_get_proxy_model_info,
_translate_model_name_for_response,
)
from litellm.types.router import DeploymentModelListingInfo
def _team_row() -> dict:
@ -1427,8 +1428,13 @@ async def test_retrieve_model_by_public_name_returns_200(monkeypatch):
team_row = _team_row()
router = _public_named_router(team_row)
deployment = MagicMock()
deployment.litellm_params.model = "azure/gpt-5.2-low-rpm-testing"
deployment.litellm_params.model = "azure/gpt-4.1"
router.get_deployment_by_model_group_name.return_value = deployment
router.get_model_listing_info.return_value = DeploymentModelListingInfo(
cost_map_keys=("azure/gpt-4.1",),
max_input_tokens=16384,
max_output_tokens=4096,
)
monkeypatch.setattr(ps, "llm_router", router)
monkeypatch.setattr(ps, "general_settings", {})
@ -1445,6 +1451,9 @@ async def test_retrieve_model_by_public_name_returns_200(monkeypatch):
resp = await ps.model_info(model_id="team-claude-sonnet", user_api_key_dict=key)
assert resp["id"] == "team-claude-sonnet"
assert resp.get("mode") == "chat"
assert resp.get("max_input_tokens") == 16384
assert resp.get("max_output_tokens") == 4096
# lookup happened by the internal routing key, not the public name
router.get_deployment_by_model_group_name.assert_called_once_with(
"model_name_team-abc-123_4a6b8"