scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-30 01:52:18 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 70

fix(lint): extract _apply_parallel_tool_use_config to fix PLR0915 in _prepare_request_params

Browse source
This commit is contained in:
Quentin Machu 2026-05-29 00:11:37 -04:00
parent 61118c9ab1
commit 620d398220
No known key found for this signature in database
1 changed files with 20 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

36
litellm/llms/bedrock/chat/converse_transformation.py
View file

@ -1231,6 +1231,25 @@ class AmazonConverseConfig(BaseConfig):
return {}
def _apply_parallel_tool_use_config(
self, model: str, additional_request_params: dict
) -> None:
"""Merge _parallel_tool_use_config into additional_request_params for Claude 4.5+ models."""
parallel_tool_use_config = additional_request_params.pop(
"_parallel_tool_use_config", None
)
if parallel_tool_use_config is not None and is_claude_4_5_on_bedrock(model):
for key, value in parallel_tool_use_config.items():
if (
key in additional_request_params
and isinstance(additional_request_params[key], dict)
and isinstance(value, dict)
):
additional_request_params[key].update(value)
else:
additional_request_params[key] = value
additional_request_params.pop("parallel_tool_calls", None)
def _prepare_request_params(
self, optional_params: dict, model: str
) -> Tuple[dict, dict, dict, Optional[OutputConfigBlock]]:
@ -1312,22 +1331,7 @@ class AmazonConverseConfig(BaseConfig):
k: v for k, v in inference_params.items() if k in total_supported_params
}
# Handle parallel_tool_calls configuration
parallel_tool_use_config = additional_request_params.pop(
"_parallel_tool_use_config", None
)
if parallel_tool_use_config is not None and is_claude_4_5_on_bedrock(model):
for key, value in parallel_tool_use_config.items():
if (
key in additional_request_params
and isinstance(additional_request_params[key], dict)
and isinstance(value, dict)
):
additional_request_params[key].update(value)
else:
additional_request_params[key] = value
additional_request_params.pop("parallel_tool_calls", None)
self._apply_parallel_tool_use_config(model, additional_request_params)
# Only set the topK value in for models that support it
additional_request_params.update(