scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-10 22:41:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 82

style: apply black formatting

Browse source 
Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
Chris Hunter 2026-03-20 14:23:27 +01:00
parent 28c2873dae
commit 645599f5a0
1 changed files with 13 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
litellm/llms/bedrock/messages/invoke_transformations/anthropic_claude3_transformation.py
View file

@ -373,13 +373,10 @@ class AmazonAnthropicClaudeMessagesConfig(
schema_text = {"type": "text", "text": json.dumps(schema)}
content.append(schema_text)
_VALID_TOOL_ID_PATTERN = re.compile(r"^[a-zA-Z0-9_-]+$")
_INVALID_TOOL_ID_CHARS = re.compile(r"[^a-zA-Z0-9_-]")
def _sanitize_tool_use_ids(
self, anthropic_messages_request: Dict
) -> None:
def _sanitize_tool_use_ids(self, anthropic_messages_request: Dict) -> None:
"""
Sanitize tool_use IDs to match Bedrock's required pattern.
@ -409,12 +406,18 @@ class AmazonAnthropicClaudeMessagesConfig(
block_type = block.get("type")
if block_type == "tool_use" and "id" in block:
tool_id = block["id"]
if isinstance(tool_id, str) and not self._VALID_TOOL_ID_PATTERN.match(tool_id):
if isinstance(
tool_id, str
) and not self._VALID_TOOL_ID_PATTERN.match(tool_id):
block["id"] = self._INVALID_TOOL_ID_CHARS.sub("_", tool_id)
elif block_type == "tool_result" and "tool_use_id" in block:
tool_use_id = block["tool_use_id"]
if isinstance(tool_use_id, str) and not self._VALID_TOOL_ID_PATTERN.match(tool_use_id):
block["tool_use_id"] = self._INVALID_TOOL_ID_CHARS.sub("_", tool_use_id)
if isinstance(
tool_use_id, str
) and not self._VALID_TOOL_ID_PATTERN.match(tool_use_id):
block["tool_use_id"] = self._INVALID_TOOL_ID_CHARS.sub(
"_", tool_use_id
)
def transform_anthropic_messages_request(
self,
@ -438,9 +441,9 @@ class AmazonAnthropicClaudeMessagesConfig(
# 1. anthropic_version is required for all claude models
if "anthropic_version" not in anthropic_messages_request:
anthropic_messages_request[
"anthropic_version"
] = self.DEFAULT_BEDROCK_ANTHROPIC_API_VERSION
anthropic_messages_request["anthropic_version"] = (
self.DEFAULT_BEDROCK_ANTHROPIC_API_VERSION
)
# 2. `stream` is not allowed in request body for bedrock invoke
if "stream" in anthropic_messages_request: