scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-03 02:22:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 81

chore: tighten wildcard test per Greptile feedback

Browse source
This commit is contained in:
Hannah Smith 2026-04-21 23:07:48 -04:00
parent 043bc8e7f8
commit 2b395a9754
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
tests/test_litellm/proxy/test_health_check_max_tokens.py
View file

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