scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-06 08:16:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 178

test(router): cover configured mode lookup

Browse source
This commit is contained in:
moe-berri 2026-09-03 13:43:30 -07:00
parent 828d561fcc
commit a264c62b04
1 changed files with 65 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

65
tests/test_litellm/test_router.py
View file

@ -7358,6 +7358,71 @@ def test_get_configured_token_limits_coerces_numeric_strings():
assert router.get_configured_token_limits("quoted-limits-model") == (32000, 8000)
def test_get_configured_mode_reads_deployment_model_info():
router = litellm.Router(
model_list=[
{
"model_name": "chat-model",
"litellm_params": {"model": "openai/some-unmapped-model"},
"model_info": {"mode": "chat"},
}
]
)
assert router.get_configured_mode("chat-model") == "chat"
def test_get_configured_mode_returns_none_for_unset_or_unknown():
router = litellm.Router(
model_list=[
{
"model_name": "no-mode-model",
"litellm_params": {"model": "openai/some-unmapped-model"},
}
]
)
assert router.get_configured_mode("no-mode-model") is None
assert router.get_configured_mode("not-a-real-model") is None
def test_get_configured_mode_skips_wildcard_pattern_matching():
router = litellm.Router(
model_list=[
{
"model_name": "bedrock/*",
"litellm_params": {"model": "bedrock/*"},
"model_info": {"mode": "chat"},
}
]
)
with patch.object(
router.pattern_router, "route", side_effect=AssertionError("pattern route called")
):
assert (
router.get_configured_mode("bedrock/anthropic.claude-3-5-sonnet-20240620-v1:0")
is None
)
def test_get_configured_mode_treats_malformed_values_as_absent():
malformed = ["", " ", 12345, ["chat"], {"mode": "chat"}, True]
router = litellm.Router(
model_list=[
{
"model_name": f"bad-mode-{i}",
"litellm_params": {"model": "openai/some-unmapped-model"},
"model_info": {"mode": bad},
}
for i, bad in enumerate(malformed)
]
)
for i in range(len(malformed)):
assert router.get_configured_mode(f"bad-mode-{i}") is None
def test_get_configured_display_name_reads_deployment_model_info():
router = litellm.Router(
model_list=[