From 5b080b865ac11ee6a5a09cbe71e3f12e3028435f Mon Sep 17 00:00:00 2001 From: mubashir1osmani Date: Thu, 16 Jul 2026 18:21:03 -0700 Subject: [PATCH] test(bedrock): update remaining toolResult document length asserts --- .../llms/bedrock/chat/test_converse_transformation.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tests/test_litellm/llms/bedrock/chat/test_converse_transformation.py b/tests/test_litellm/llms/bedrock/chat/test_converse_transformation.py index 3df9a221dc1..85f1265bfda 100644 --- a/tests/test_litellm/llms/bedrock/chat/test_converse_transformation.py +++ b/tests/test_litellm/llms/bedrock/chat/test_converse_transformation.py @@ -5401,8 +5401,9 @@ def test_bedrock_tool_message_file_id_http_url_becomes_document(): assert mock_proc.call_args.kwargs["image_url"] == pdf_url tool_result = translated_msg[-1]["content"][-1]["toolResult"] - assert len(tool_result["content"]) == 1 - block = tool_result["content"][0] + assert len(tool_result["content"]) == 2 + assert "text" in tool_result["content"][0] + block = tool_result["content"][1] assert "document" in block, f"expected document block, got {block}" assert block["document"]["source"]["bytes"] == "ZmFrZQ=="