mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-22 00:31:44 +00:00
test: add adaptive_routers alias resolution test
Cover the adaptive_routers branch in async_pre_routing_hook, completing test coverage for all four router map types.
This commit is contained in:
parent
4f17351d31
commit
55d393976c
1 changed files with 22 additions and 0 deletions
|
|
@ -55,6 +55,8 @@ def _make_router_with_alias(
|
|||
router.auto_routers[target_group] = mock_sub_router
|
||||
elif router_type == "quality":
|
||||
router.quality_routers[target_group] = mock_sub_router
|
||||
elif router_type == "adaptive":
|
||||
router.adaptive_routers[target_group] = mock_sub_router
|
||||
|
||||
return router
|
||||
|
||||
|
|
@ -123,6 +125,26 @@ class TestPreRoutingHookAliasResolution:
|
|||
mock = router.quality_routers["quality-router-group"]
|
||||
mock.async_pre_routing_hook.assert_awaited_once()
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_alias_resolves_to_adaptive_router(self):
|
||||
"""An aliased model name should match the adaptive_router for the
|
||||
resolved group name."""
|
||||
router = _make_router_with_alias(
|
||||
alias_name="adaptive",
|
||||
target_group="adaptive-router-group",
|
||||
router_type="adaptive",
|
||||
)
|
||||
|
||||
result = await router.async_pre_routing_hook(
|
||||
model="adaptive",
|
||||
request_kwargs={},
|
||||
messages=[{"role": "user", "content": "Hello"}],
|
||||
)
|
||||
|
||||
assert result is not None
|
||||
mock = router.adaptive_routers["adaptive-router-group"]
|
||||
mock.async_pre_routing_hook.assert_awaited_once()
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_non_alias_model_still_works(self):
|
||||
"""A model name that is NOT an alias should still match the router
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue