From d0a75d28cf766ca405064088cb9dc4979541aebd Mon Sep 17 00:00:00 2001 From: Claude Date: Wed, 4 Nov 2026 18:13:33 +0000 Subject: [PATCH] test(router): narrow the failed-pick raise to the no-healthy-deployments error Co-Authored-By: Claude Opus 5.5 Claude-Session: https://claude.ai/code/session_013FDwtoMjnHa4xxazGJJNSm --- tests/unit/test_router_weighted_failover.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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