test: fix assert string on test

This commit is contained in:
Krrish Dholakia 2024-08-28 12:07:05 -07:00
parent 65097d595c
commit 60381ffcdd

View file

@ -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])