test(bedrock): update remaining toolResult document length asserts

This commit is contained in:
mubashir1osmani 2026-07-16 18:21:03 -07:00
parent 8913a0b001
commit 5b080b865a

View file

@ -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=="