prompt adapter

This commit is contained in:
Xi Yan 2025-01-30 14:01:18 -08:00
parent 393d94429c
commit ff73f0a17b

View file

@ -185,8 +185,10 @@ async def localize_image_content(media: ImageContentItem) -> Tuple[bytes, str]:
return content, format
else:
pil_image = PIL_Image.open(io.BytesIO(image.data))
return image.data, pil_image.format
# data is a base64 encoded string, decode it to bytes first
data_bytes = base64.b64decode(image.data)
pil_image = PIL_Image.open(io.BytesIO(data_bytes))
return data_bytes, pil_image.format
async def convert_image_content_to_url(