From 2b395a9754391c2504855c0e1ac7b73ee2dc10ee Mon Sep 17 00:00:00 2001
From: Hannah Smith <64043506+hannahmadison@users.noreply.github.com>
Date: Tue, 21 Apr 2026 23:07:48 -0400
Subject: [PATCH] chore: tighten wildcard test per Greptile feedback

---
 tests/test_litellm/proxy/test_health_check_max_tokens.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

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