From 620d39822031c661f7dd4ecd7e183a06565d0ba2 Mon Sep 17 00:00:00 2001 From: Quentin Machu Date: Fri, 29 May 2026 00:11:37 -0400 Subject: [PATCH] fix(lint): extract _apply_parallel_tool_use_config to fix PLR0915 in _prepare_request_params --- .../bedrock/chat/converse_transformation.py | 36 ++++++++++--------- 1 file changed, 20 insertions(+), 16 deletions(-) diff --git a/litellm/llms/bedrock/chat/converse_transformation.py b/litellm/llms/bedrock/chat/converse_transformation.py index 35cefd1c0ff..cec98d58d96 100644 --- a/litellm/llms/bedrock/chat/converse_transformation.py +++ b/litellm/llms/bedrock/chat/converse_transformation.py @@ -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(