From 0288ed35da38910debd2b9e823aaf32e4dd4605d Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Wed, 13 Aug 2025 23:33:32 -0700 Subject: [PATCH] test: update tests --- tests/local_testing/test_router.py | 15 +++++++-------- 1 file changed, 7 insertions(+), 8 deletions(-) diff --git a/tests/local_testing/test_router.py b/tests/local_testing/test_router.py index 73c31f0ba4..f719f0b606 100644 --- a/tests/local_testing/test_router.py +++ b/tests/local_testing/test_router.py @@ -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(