scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-11 03:38:38 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 84

test(bedrock): expect text+document in toolResult content for PDFs

Browse source
This commit is contained in:
mubashir1osmani 2026-07-16 18:20:44 -07:00
parent d5200c96c0
commit 8913a0b001
1 changed files with 7 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
tests/test_litellm/llms/bedrock/chat/test_converse_transformation.py
View file

@ -5293,8 +5293,10 @@ def test_bedrock_tool_message_openai_file_pdf_becomes_document():
tool_result = translated_msg[-1]["content"][-1]["toolResult"]
assert tool_result["toolUseId"] == "tooluse_pdf_1"
assert len(tool_result["content"]) == 1
block = tool_result["content"][0]
# Bedrock requires a sibling text block when documents are present.
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"]["format"] == "pdf"
assert block["document"]["source"]["bytes"] == pdf_b64
@ -5337,8 +5339,9 @@ def test_bedrock_tool_message_image_url_pdf_data_uri_becomes_document():
tool_result = translated_msg[-1]["content"][-1]["toolResult"]
assert tool_result["toolUseId"] == "tooluse_pdf_img_1"
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"]["format"] == "pdf"
assert block["document"]["source"]["bytes"] == pdf_b64