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

test: update anthropic http image test for url-forwarding behavior

Browse source 
The async completion test asserted the old behavior where http:// image
URLs were downloaded and inlined as base64. Since http:// URLs are now
forwarded as url sources like https://, the test asserts nothing is
downloaded and both images arrive as url sources.

Signed-off-by: Junyi Yao <j.yao@wustl.edu>
This commit is contained in:
Junyi Yao 2026-09-24 20:53:54 -07:00 • committed by Kylinny
parent 13f9b82552
commit 43b39ff72e
1 changed files with 8 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
tests/unit/llms/anthropic/chat/test_anthropic_chat_handler.py
View file

@ -207,7 +207,12 @@ def test_anthropic_tool_call_error_is_mapped_from_the_provider_response():
assert len(route.calls) == 1
async def test_anthropic_async_completion_inlines_http_images_off_the_event_loop(async_only_image_fetch):
async def test_anthropic_async_completion_forwards_http_images_as_url_sources(async_only_image_fetch):
"""http:// image URLs are forwarded as URL sources, never downloaded.
Regression test for https://github.com/BerriAI/litellm/issues/43098: the
proxy used to download http:// URLs and send them as base64 without notice.
"""
http_image_url = f"http://img.example/{uuid.uuid4()}.png"
https_image_url = f"https://img.example/{uuid.uuid4()}.png"
captured = {}
@ -247,10 +252,10 @@ async def test_anthropic_async_completion_inlines_http_images_off_the_event_loop
)
assert response.choices[0].message.content == "Green"
assert async_only_image_fetch.fetched == [http_image_url]
assert async_only_image_fetch.fetched == []
sources = [part["source"] for part in captured["body"]["messages"][0]["content"] if part["type"] == "image"]
assert sources == [
{"type": "base64", "media_type": "image/png", "data": async_only_image_fetch.base64_png},
{"type": "url", "url": http_image_url},
{"type": "url", "url": https_image_url},
]