scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-14 23:21:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 102

test(factory): add map_system_message_pt content-block regression to core-utils mapped test

Browse source
This commit is contained in:
Devin AI 2026-07-11 17:28:24 +00:00
parent a760798283
commit a78b03ad8d
2 changed files with 64 additions and 63 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

63
tests/llm_translation/test_optional_params.py
View file

@ -55,69 +55,6 @@ def test_supports_system_message():
assert isinstance(response, litellm.ModelResponse)
def test_map_system_message_pt_content_block_lists():
"""
Regression for https://github.com/BerriAI/litellm/issues/32904
When system and/or user content is a list of content blocks (Anthropic-style),
merging must not string-concatenate a list. Merge as a block list instead.
"""
messages = [
{"role": "system", "content": [{"type": "text", "text": "Be brief."}]},
{"role": "user", "content": [{"type": "text", "text": "Hi"}]},
]
new_messages = map_system_message_pt(messages=messages)
assert len(new_messages) == 1
assert new_messages[0]["role"] == "user"
assert new_messages[0]["content"] == [
{"type": "text", "text": "Be brief."},
{"type": "text", "text": "Hi"},
]
def test_map_system_message_pt_mixed_string_and_content_blocks():
"""
Mixed forms: a plain string on one side and a content-block list on the other
should merge into a block list rather than raising.
"""
system_list_user_str = map_system_message_pt(
messages=[
{"role": "system", "content": [{"type": "text", "text": "Be brief."}]},
{"role": "user", "content": "Hi"},
]
)
assert system_list_user_str[0]["content"] == [
{"type": "text", "text": "Be brief."},
{"type": "text", "text": "Hi"},
]
system_str_user_list = map_system_message_pt(
messages=[
{"role": "system", "content": "Be brief."},
{"role": "user", "content": [{"type": "text", "text": "Hi"}]},
]
)
assert system_str_user_list[0]["content"] == [
{"type": "text", "text": "Be brief."},
{"type": "text", "text": "Hi"},
]
def test_map_system_message_pt_string_content_unchanged():
"""
Plain string content on both sides keeps the original space-joined behavior.
"""
new_messages = map_system_message_pt(
messages=[
{"role": "system", "content": "Be brief."},
{"role": "user", "content": "Hi"},
]
)
assert new_messages[0]["content"] == "Be brief. Hi"
@pytest.mark.parametrize(
"stop_sequence, expected_count", [("\n", 0), (["\n"], 0), (["finish_reason"], 1)]
)

64
tests/test_litellm/litellm_core_utils/prompt_templates/test_litellm_core_utils_prompt_templates_factory.py
View file

@ -18,6 +18,7 @@ from litellm.litellm_core_utils.prompt_templates.factory import (
anthropic_messages_pt,
convert_to_gemini_tool_call_result,
make_valid_bedrock_tool_name,
map_system_message_pt,
ollama_pt,
sanitize_messages_for_tool_calling,
)
@ -3166,3 +3167,66 @@ async def test_bedrock_converse_message_level_cache_point_preserves_ttl_async():
)
assert _collect_cache_points(result) == [{"type": "default", "ttl": "1h"}]
def test_map_system_message_pt_content_block_lists():
"""
Regression for https://github.com/BerriAI/litellm/issues/32904
When system and/or user content is a list of content blocks (Anthropic-style),
merging must not string-concatenate a list. Merge as a block list instead.
"""
messages = [
{"role": "system", "content": [{"type": "text", "text": "Be brief."}]},
{"role": "user", "content": [{"type": "text", "text": "Hi"}]},
]
new_messages = map_system_message_pt(messages=messages)
assert len(new_messages) == 1
assert new_messages[0]["role"] == "user"
assert new_messages[0]["content"] == [
{"type": "text", "text": "Be brief."},
{"type": "text", "text": "Hi"},
]
def test_map_system_message_pt_mixed_string_and_content_blocks():
"""
Mixed forms: a plain string on one side and a content-block list on the other
should merge into a block list rather than raising.
"""
system_list_user_str = map_system_message_pt(
messages=[
{"role": "system", "content": [{"type": "text", "text": "Be brief."}]},
{"role": "user", "content": "Hi"},
]
)
assert system_list_user_str[0]["content"] == [
{"type": "text", "text": "Be brief."},
{"type": "text", "text": "Hi"},
]
system_str_user_list = map_system_message_pt(
messages=[
{"role": "system", "content": "Be brief."},
{"role": "user", "content": [{"type": "text", "text": "Hi"}]},
]
)
assert system_str_user_list[0]["content"] == [
{"type": "text", "text": "Be brief."},
{"type": "text", "text": "Hi"},
]
def test_map_system_message_pt_string_content_unchanged():
"""
Plain string content on both sides keeps the original space-joined behavior.
"""
new_messages = map_system_message_pt(
messages=[
{"role": "system", "content": "Be brief."},
{"role": "user", "content": "Hi"},
]
)
assert new_messages[0]["content"] == "Be brief. Hi"