diff --git a/tests/test_litellm/proxy/test_health_check_max_tokens.py b/tests/test_litellm/proxy/test_health_check_max_tokens.py index ac4bc33cbf3..12b52e375ea 100644 --- a/tests/test_litellm/proxy/test_health_check_max_tokens.py +++ b/tests/test_litellm/proxy/test_health_check_max_tokens.py @@ -33,15 +33,15 @@ async def test_update_litellm_params_max_tokens_custom(): @pytest.mark.asyncio async def test_update_litellm_params_max_tokens_wildcard(): """ - Test that max_tokens does NOT default to 1 for wildcard models. + Test that max_tokens is not set for wildcard models. """ model_info = {} litellm_params = {"model": "openai/*"} updated_params = _update_litellm_params_for_health_check(model_info, litellm_params) - # Should not be set to 1 - assert "max_tokens" not in updated_params or updated_params["max_tokens"] != 1 + # Should not be set at all for wildcard models + assert "max_tokens" not in updated_params @pytest.mark.asyncio