ai-workspace-services/litellm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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(