From d2346b525912a5c006c70bbf32b2cceeaba0c8bb Mon Sep 17 00:00:00 2001 From: Sameer Kankute Date: Fri, 1 May 2026 12:09:19 +0530 Subject: [PATCH] Fix prompt factory --- .../litellm_core_utils/prompt_templates/factory.py | 11 ++++------- 1 file changed, 4 insertions(+), 7 deletions(-) diff --git a/litellm/litellm_core_utils/prompt_templates/factory.py b/litellm/litellm_core_utils/prompt_templates/factory.py index 4ef08b0a01..ba840bc3d8 100644 --- a/litellm/litellm_core_utils/prompt_templates/factory.py +++ b/litellm/litellm_core_utils/prompt_templates/factory.py @@ -4042,13 +4042,9 @@ def _convert_to_bedrock_tool_call_result( tool_result_content_blocks.append( BedrockToolResultContentBlock(image=_block["image"]) ) - elif content["type"] == "document": - _doc_block = BedrockConverseMessagesProcessor._process_document_message( - content - ) - if "document" in _doc_block: - tool_result_content_blocks.append( - BedrockToolResultContentBlock(document=_doc_block["document"]) + elif "document" in _block: + tool_result_content_blocks.append( + BedrockToolResultContentBlock(document=_block["document"]) ) else: verbose_logger.warning( @@ -4094,6 +4090,7 @@ def _convert_to_bedrock_tool_call_result( "%s for file tool-result block %s; dropping.", list(_file_block.keys()), content, + ) message.get("name", "") id = str(message.get("tool_call_id", str(uuid.uuid4())))