ai-workspace-services/litellm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix: test_bedrock_no_default_message

Browse Source
This commit is contained in:
Sameer Kankute 2026-02-04 17:52:28 +05:30
parent 15351a93dc
commit 1a5f2a566f
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tests/llm_translation/test_bedrock_completion.py
View File

@ -2356,9 +2356,8 @@ def test_bedrock_no_default_message():
assistant_messages = [
msg for msg in formatted_messages if msg["role"] == "assistant"
]
assert len(assistant_messages) == 2
assert assistant_messages[0]["content"][0]["text"] == "."
assert assistant_messages[1]["content"][0]["text"] == "Valid response"
assert len(assistant_messages) == 1
assert assistant_messages[0]["content"][0]["text"] == "Valid response"
@pytest.mark.parametrize("top_k_param", ["top_k", "topK"])