From bc4953b0b95e45d0b499e7daad54a30f4971f908 Mon Sep 17 00:00:00 2001 From: milo Date: Sat, 23 May 2026 00:36:13 +0800 Subject: [PATCH] fix: address typing and streaming test expectations --- litellm/llms/bedrock/chat/converse_transformation.py | 11 +++++++---- litellm/llms/bedrock/chat/invoke_handler.py | 2 +- 2 files changed, 8 insertions(+), 5 deletions(-) diff --git a/litellm/llms/bedrock/chat/converse_transformation.py b/litellm/llms/bedrock/chat/converse_transformation.py index 76da11dd337..23c1965f68e 100644 --- a/litellm/llms/bedrock/chat/converse_transformation.py +++ b/litellm/llms/bedrock/chat/converse_transformation.py @@ -1979,9 +1979,12 @@ class AmazonConverseConfig(BaseConfig): message.content = None returned_finish_reason = "tool_calls" except Exception: - tool_call = self._text_content_tool_call_transformation( - message.content, tools - ) + if message.content is not None: + tool_call = self._text_content_tool_call_transformation( + message.content, tools + ) + else: + tool_call = None if tool_call is not None: message.tool_calls = [tool_call] message.content = None @@ -2280,7 +2283,7 @@ class AmazonConverseConfig(BaseConfig): text_tool_call_tools = optional_params.get("tools") if text_tool_call_tools is None: - request_data = data + request_data: Optional[Union[dict, str]] = data if isinstance(data, str): try: request_data = json.loads(data) diff --git a/litellm/llms/bedrock/chat/invoke_handler.py b/litellm/llms/bedrock/chat/invoke_handler.py index 6c6f5ef0ce0..adc1e1793df 100644 --- a/litellm/llms/bedrock/chat/invoke_handler.py +++ b/litellm/llms/bedrock/chat/invoke_handler.py @@ -1707,7 +1707,7 @@ class AWSEventStreamDecoder: if "trace" in chunk_data: trace = chunk_data.get("trace") model_response_provider_specific_fields["trace"] = trace - delta_content = text + delta_content: Optional[str] = text if tool_use is not None and delta_content == "": delta_content = None elif (