diff --git a/tests/test_litellm/proxy/management_endpoints/test_auto_router_endpoints.py b/tests/test_litellm/proxy/management_endpoints/test_auto_router_endpoints.py index 36130137c64..a5c93c41a84 100644 --- a/tests/test_litellm/proxy/management_endpoints/test_auto_router_endpoints.py +++ b/tests/test_litellm/proxy/management_endpoints/test_auto_router_endpoints.py @@ -459,7 +459,7 @@ async def test_jev_test_routing_enforces_key_budget_before_provider_invocation( user_role=LitellmUserRoles.PROXY_ADMIN, api_key="sk-jev-budget-test", user_id="admin", - models=["cheap-model"], + models=["cheap-model", "typesafe/jev-test"], max_budget=max_budget, spend=spend, )