scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-10 22:41:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 82

fix(bfl): handle URL and file path inputs in image edit

Browse source 
_read_image_bytes was not handling string inputs (URLs or file paths),
causing a TypeError when passing a URL as the image source.
This commit is contained in:
Chesars 2026-03-04 18:16:21 -03:00
parent d9d39d545a
commit 00cf9550af
1 changed files with 10 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
litellm/llms/black_forest_labs/image_edit/transformation.py
View file

@ -184,6 +184,16 @@ class BlackForestLabsImageEditConfig(BaseImageEditConfig):
elif isinstance(image, list):
# If it's a list, take the first image
return self._read_image_bytes(image[0])
elif isinstance(image, str):
if image.startswith(("http://", "https://")):
# Download image from URL
import httpx as _httpx
response = _httpx.get(image)
return response.content
else:
# Assume it's a file path
with open(image, "rb") as f:
return f.read()
elif hasattr(image, "read"):
# File-like object
pos = getattr(image, "tell", lambda: 0)()