phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-24 18:24:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update test_optional_params.py

Browse source
This commit is contained in:
ftao 2025-03-18 16:31:49 +08:00 • committed by GitHub
parent df7b7e5ed4
commit 861422640b
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 23 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

23
tests/llm_translation/test_optional_params.py
View file

@ -53,6 +53,29 @@ def test_supports_system_message():
assert isinstance(response, litellm.ModelResponse)
def test_supports_system_message_multipart():
"""
Check if litellm.completion(...,supports_system_message=False) works with multipart user message
"""
messages = [
ChatCompletionSystemMessageParam(role="system", content="Listen here!"),
ChatCompletionUserMessageParam(role="user", content=[{"type": "text", "text": "Hello there!"}]),
]
new_messages = map_system_message_pt(messages=messages)
assert len(new_messages) == 1
assert new_messages[0]["role"] == "user"
## confirm you can make a openai call with this param
response = litellm.completion(
model="gpt-3.5-turbo", messages=new_messages, supports_system_message=False
)
assert isinstance(response, litellm.ModelResponse)
@pytest.mark.parametrize(
"stop_sequence, expected_count", [("\n", 0), (["\n"], 0), (["finish_reason"], 1)]
)