diff --git a/tests/unit/test_router_weighted_failover.py b/tests/unit/test_router_weighted_failover.py index c878b79b8e6..1f797b02dd4 100644 --- a/tests/unit/test_router_weighted_failover.py +++ b/tests/unit/test_router_weighted_failover.py @@ -1126,7 +1126,7 @@ async def test_a_failed_pick_stamps_no_deployment() -> None: ], ) - with pytest.raises(Exception) as raised: + with pytest.raises(litellm.BadRequestError, match="no healthy deployments") as raised: await router._aembedding(input=["hi"], model="no-such-model") assert getattr(raised.value, "failed_deployment_id", None) is None