scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-25 01:02:15 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 80

test: add shorten_message regression test to codecov-covered suite

Browse source
This commit is contained in:
Dantuluri Surya Narayana Raju 2026-05-17 21:57:25 +05:30
parent ffcd98c481
commit 904537880d
1 changed files with 20 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
tests/test_litellm/test_utils.py
View file

@ -4107,3 +4107,23 @@ class TestValidateAndFixThinkingParam:
validate_and_fix_thinking_param(thinking=thinking)
assert "budgetTokens" in thinking
assert "budget_tokens" not in thinking
def test_shorten_message_content_never_grows():
"""Regression test: shorten_message_to_fit_limit must never make content longer.
When half_length==0, content[-0:] == content[0:] (entire string) due to
Python's -0 == 0 identity, making trimmed content longer than the original.
Regression: https://github.com/BerriAI/litellm/issues/28128
"""
from litellm.utils import shorten_message_to_fit_limit
content = "hello world this is a moderately long message"
message = {"role": "user", "content": content}
original_len = len(content)
result = shorten_message_to_fit_limit(message, tokens_needed=1, model=None)
assert len(result["content"]) <= original_len, (
f"Content grew from {original_len} to {len(result['content'])} chars — "
"half_length==0 guard is missing or broken"
)