From 0de3b7ef714d91fcfb596f0ec8a376cbffbd3432 Mon Sep 17 00:00:00 2001
From: ciaranbor <ciaranborourke-dev@proton.me>
Date: Wed, 3 Dec 2025 16:11:16 +0000
Subject: [PATCH] Add indexing info to ImageChunk

---
 src/exo/shared/types/chunks.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/exo/shared/types/chunks.py b/src/exo/shared/types/chunks.py
index cb7ff81e..aef0d399 100644
--- a/src/exo/shared/types/chunks.py
+++ b/src/exo/shared/types/chunks.py
@@ -26,7 +26,10 @@ class TokenChunk(BaseChunk):
 
 
 class ImageChunk(BaseChunk):
-    data: str  # Base64-encoded image data
+    data: str
+    chunk_index: int
+    total_chunks: int
+    image_index: int
     finish_reason: FinishReason | None = None