scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-06 08:16:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 79

fix: add if openai_image_url guard on single-item path and update tests

Browse source 
Add consistency guard to skip creating ChatCompletionImageObject when
image URL translation returns None, matching the multi-item path.
Update two tests to assert the new list-of-image-object structure.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
Hashwanth S 2026-04-02 17:36:41 -07:00
parent 5e24cf590b
commit 7759827c13
2 changed files with 32 additions and 26 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

38
litellm/llms/anthropic/experimental_pass_through/adapters/transformation.py
View file

@ -473,26 +473,26 @@ class LiteLLMAnthropicMessagesAdapter:
self._translate_anthropic_image_to_openai(
cast(dict, source)
)
or ""
)
tool_result = ChatCompletionToolMessage(
role="tool",
tool_call_id=content.get(
"tool_use_id", ""
),
content=[
ChatCompletionImageObject(
type="image_url",
image_url=ChatCompletionImageUrlObject(
url=openai_image_url
),
)
],
)
self._add_cache_control_if_applicable(
content, tool_result, model
)
tool_message_list.append(tool_result) # type: ignore[arg-type]
if openai_image_url:
tool_result = ChatCompletionToolMessage(
role="tool",
tool_call_id=content.get(
"tool_use_id", ""
),
content=[
ChatCompletionImageObject(
type="image_url",
image_url=ChatCompletionImageUrlObject(
url=openai_image_url
),
)
],
)
self._add_cache_control_if_applicable(
content, tool_result, model
)
tool_message_list.append(tool_result) # type: ignore[arg-type]
else:
# For multiple content items, combine into a single tool message
# with list content to preserve all items while having one tool_use_id

20
tests/test_litellm/llms/anthropic/experimental_pass_through/adapters/test_anthropic_experimental_pass_through_adapters_transformation.py
View file

@ -750,10 +750,13 @@ def test_translate_anthropic_messages_to_openai_tool_result_with_base64_image():
break
assert tool_message is not None, "Tool message not found in result"
# Tool messages in OpenAI format have string content (data URL), not list
assert isinstance(tool_message["content"], str)
assert tool_message["content"].startswith("data:image/jpeg;base64,")
assert "/9j/4AAQSkZJRgABAQAAAQABAAD" in tool_message["content"]
# Tool messages with images use list content with ChatCompletionImageObject
assert isinstance(tool_message["content"], list)
assert len(tool_message["content"]) == 1
image_obj = tool_message["content"][0]
assert image_obj["type"] == "image_url"
assert image_obj["image_url"]["url"].startswith("data:image/jpeg;base64,")
assert "/9j/4AAQSkZJRgABAQAAAQABAAD" in image_obj["image_url"]["url"]
def test_translate_anthropic_messages_to_openai_tool_result_with_url_image():
@ -806,10 +809,13 @@ def test_translate_anthropic_messages_to_openai_tool_result_with_url_image():
break
assert tool_message is not None, "Tool message not found in result"
# Tool messages in OpenAI format have string content (URL), not list
assert isinstance(tool_message["content"], str)
# Tool messages with images use list content with ChatCompletionImageObject
assert isinstance(tool_message["content"], list)
assert len(tool_message["content"]) == 1
image_obj = tool_message["content"][0]
assert image_obj["type"] == "image_url"
assert (
tool_message["content"]
image_obj["image_url"]["url"]
== "https://i0.wp.com/picjumbo.com/wp-content/uploads/amazing-stone-path-in-forest-free-image.jpg"
)