address comments

This commit is contained in:
Xi Yan 2024-12-30 16:18:20 -08:00
parent a720928a1c
commit 790f21459a

View file

@ -121,7 +121,9 @@ async def interleaved_content_convert_to_raw(
if c.url.uri.startswith("data"): if c.url.uri.startswith("data"):
match = re.match(r"data:image/(\w+);base64,(.+)", c.url.uri) match = re.match(r"data:image/(\w+);base64,(.+)", c.url.uri)
if not match: if not match:
raise ValueError("Invalid data URL format") raise ValueError(
f"Invalid data URL format, {c.url.uri[:40]}..."
)
_, image_data = match.groups() _, image_data = match.groups()
data = base64.b64decode(image_data) data = base64.b64decode(image_data)
elif c.url.uri.startswith("file://"): elif c.url.uri.startswith("file://"):