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

Merge pull request #21680 from BerriAI/fix/converse-transformation-too-many-statements

Browse source 
fix(lint): extract service_tier mapping to fix PLR0915 in converse_transformation.py
This commit is contained in:
jquinter 2026-02-20 13:19:02 -03:00 • committed by GitHub
commit a8d806cdf0
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 13 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -932,14 +932,7 @@ class AmazonConverseConfig(BaseConfig):
self._validate_request_metadata(value) # type: ignore
optional_params["requestMetadata"] = value
if param == "service_tier" and isinstance(value, str):
# Map OpenAI service_tier (string) to Bedrock serviceTier (object)
# OpenAI values: "auto", "default", "flex", "priority"
# Bedrock values: "default", "flex", "priority" (no "auto")
bedrock_tier = value
if value == "auto":
bedrock_tier = "default" # Bedrock doesn't support "auto"
if bedrock_tier in ("default", "flex", "priority"):
optional_params["serviceTier"] = {"type": bedrock_tier}
self._map_service_tier_param(value, optional_params)
if param == "web_search_options" and isinstance(value, dict):
# Note: we use `isinstance(value, dict)` instead of `value and isinstance(value, dict)`
@ -970,6 +963,18 @@ class AmazonConverseConfig(BaseConfig):
return optional_params
def _map_service_tier_param(self, value: str, optional_params: dict) -> None:
"""Map OpenAI service_tier (string) to Bedrock serviceTier (object).
OpenAI values: "auto", "default", "flex", "priority"
Bedrock values: "default", "flex", "priority" (no "auto")
"""
bedrock_tier = value
if value == "auto":
bedrock_tier = "default" # Bedrock doesn't support "auto"
if bedrock_tier in ("default", "flex", "priority"):
optional_params["serviceTier"] = {"type": bedrock_tier}
def _translate_response_format_param(
self,
value: dict,