scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-11-12 08:42:18 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 207

test(router): cover deployment resolution for aliases, routing groups and ids

Browse source
This commit is contained in:
ryan-crabbe-berri 2026-10-20 12:04:13 -07:00
parent 9f621bc324
commit a7f1d42459
1 changed files with 25 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
tests/unit/router_utils/test_model_access_group_grants.py
View file

@ -127,3 +127,28 @@ async def test_the_default_fallback_does_not_reach_a_tagged_deployment() -> None
)
def test_only_names_that_grant_one_thing_count_as_explicit(models: list[str], grants: frozenset[str]) -> None:
assert explicit_model_grants(models) == grants
@pytest.mark.parametrize(
("model", "resolved_name", "deployment_ids"),
[
pytest.param("alias-to-private", "private-model", ("private-id",), id="alias"),
pytest.param("mixed-group", "mixed-group", ("private-id", "public-id"), id="routing-group"),
],
)
def test_resolution_finds_tagged_deployments_for_any_caller(
model: str, resolved_name: str, deployment_ids: tuple[str, ...]
) -> None:
name, deployments = _router()._resolve_available_deployments(
model=model, specific_deployment=False, request_kwargs={"metadata": _metadata(frozenset())}
)
assert isinstance(deployments, list)
assert (name, tuple(d["model_info"]["id"] for d in deployments)) == (resolved_name, deployment_ids)
def test_resolution_finds_a_tagged_deployment_by_id_for_any_caller() -> None:
name, deployment = _router()._resolve_available_deployments(
model="private-id", specific_deployment=False, request_kwargs={"metadata": _metadata(frozenset())}
)
assert isinstance(deployment, dict)
assert (name, deployment["model_info"]["id"]) == ("openai/private-id-upstream", "private-id")