Use base64 encoded str for image data

This commit is contained in:
ciaranbor
2026-01-06 10:03:18 +00:00
parent bedb9045a0
commit 098a11b262
2 changed files with 3 additions and 7 deletions
+2 -6
View File
@@ -566,14 +566,10 @@ class API:
)
assert isinstance(chunk, ImageChunk)
# Convert image bytes to base64
import base64
b64_image = base64.b64encode(chunk.data).decode("utf-8")
# chunk.data is already base64-encoded string
images.append(
ImageData(
b64_json=b64_image
b64_json=chunk.data
if payload.response_format == "b64_json"
else None,
url=None, # URL format not implemented yet
+1 -1
View File
@@ -23,7 +23,7 @@ class TokenChunk(BaseChunk):
class ImageChunk(BaseChunk):
data: bytes
data: str # Base64-encoded image data
finish_reason: FinishReason | None = None