From 60381ffcdd662fe077c86cd2894ac564644b1baf Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Wed, 28 Aug 2024 12:07:05 -0700 Subject: [PATCH] test: fix assert string on test --- litellm/tests/test_router_fallbacks.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/litellm/tests/test_router_fallbacks.py b/litellm/tests/test_router_fallbacks.py index a9a225be7c5..ec5b4373526 100644 --- a/litellm/tests/test_router_fallbacks.py +++ b/litellm/tests/test_router_fallbacks.py @@ -1225,7 +1225,9 @@ async def test_using_default_fallback(sync_mode): pytest.fail(f"Expected call to fail we passed model=openai/foo") except Exception as e: print("got exception = ", e) - assert "No healthy deployment available, passed model=very-bad-model" in str(e) + from litellm.types.router import RouterErrors + + assert RouterErrors.no_deployments_available.value in str(e) @pytest.mark.parametrize("sync_mode", [False])