scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-08 03:08:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 114

test: update tests

Browse source
This commit is contained in:
Krrish Dholakia 2025-08-13 23:33:32 -07:00
parent b53962dee2
commit 0288ed35da
1 changed files with 7 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
tests/local_testing/test_router.py
View file

@ -2023,14 +2023,13 @@ def test_router_get_model_info(model, base_model, llm_provider):
deployment=deployment.to_json(), received_model_name=model
)
else:
try:
router.get_router_model_info(
deployment=deployment.to_json(), received_model_name=model
)
pytest.fail("Expected this to raise model not mapped error")
except Exception as e:
if "This model isn't mapped yet" in str(e):
pass
# Azure models without base_model now fallback to using the original model name
# instead of raising an exception. This should succeed but log a warning.
model_info = router.get_router_model_info(
deployment=deployment.to_json(), received_model_name=model
)
# Verify that model_info is returned (even if it may have default values)
assert model_info is not None
@pytest.mark.parametrize(