scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-04 02:31:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 69

Cover explicit tool choice with parallel calls

Browse source
This commit is contained in:
Warren 2026-05-21 15:52:33 -07:00
parent 91cdb0e66b
commit 1172a18211
1 changed files with 14 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
tests/test_litellm/llms/bedrock/chat/test_converse_transformation.py
View file

@ -3794,29 +3794,35 @@ def test_parallel_tool_calls_true_includes_type_field():
assert "parallel_tool_calls" not in request_data["additionalModelRequestFields"]
@pytest.mark.parametrize(
("parallel_tool_calls", "disable_parallel_tool_use"),
[(True, False), (False, True)],
ids=["parallel-enabled", "parallel-disabled"],
)
@pytest.mark.parametrize(
("openai_tool_choice", "expected_anthropic_tool_choice"),
[
(
"auto",
{"type": "auto", "disable_parallel_tool_use": True},
{"type": "auto"},
),
(
"required",
{"type": "any", "disable_parallel_tool_use": True},
{"type": "any"},
),
(
{"type": "function", "function": {"name": "get_weather"}},
{
"type": "tool",
"name": "get_weather",
"disable_parallel_tool_use": True,
},
),
],
ids=["auto", "required", "named-tool"],
)
def test_parallel_tool_calls_moves_explicit_tool_choice_into_anthropic_extension(
parallel_tool_calls,
disable_parallel_tool_use,
openai_tool_choice,
expected_anthropic_tool_choice,
):
@ -3829,7 +3835,7 @@ def test_parallel_tool_calls_moves_explicit_tool_choice_into_anthropic_extension
non_default_params={
"tools": _TOOL_PARAM,
"tool_choice": openai_tool_choice,
"parallel_tool_calls": False,
"parallel_tool_calls": parallel_tool_calls,
},
optional_params={},
model=model,
@ -3844,6 +3850,10 @@ def test_parallel_tool_calls_moves_explicit_tool_choice_into_anthropic_extension
headers={},
)
expected_anthropic_tool_choice = {
**expected_anthropic_tool_choice,
"disable_parallel_tool_use": disable_parallel_tool_use,
}
assert (
request_data["additionalModelRequestFields"]["tool_choice"]
== expected_anthropic_tool_choice