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

test: add tests to catch stop block indentation bug

Browse source
This commit is contained in:
Monesh Ram 2026-02-24 09:39:26 +05:30 • committed by GitHub
parent 4a024957ec
commit f9bb09352c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 18 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
tests/litellm/llms/bedrock/test_bedrock_converse_transformation.py
View file

@ -64,3 +64,21 @@ def test_map_openai_params_stop_filtering():
drop_params=False
)
assert optional_params["stopSequences"] == ["valid"]
# Test valid string stop - this would fail if stop block is outside the for loop
config.map_openai_params(
non_default_params={"stop": "END"},
optional_params=optional_params,
model="bedrock/anthropic.claude-3",
drop_params=False
)
assert optional_params["stopSequences"] == ["END"]
# Test valid list of stop strings
config.map_openai_params(
non_default_params={"stop": ["STOP1", "STOP2"]},
optional_params=optional_params,
model="bedrock/anthropic.claude-3",
drop_params=False
)
assert optional_params["stopSequences"] == ["STOP1", "STOP2"]