diff --git a/src/arkruntime/types/images/image_data_item.py b/src/arkruntime/types/images/image_data_item.py
index ca9af24..3d9fafe 100644
--- a/src/arkruntime/types/images/image_data_item.py
+++ b/src/arkruntime/types/images/image_data_item.py
@@ -12,6 +12,7 @@
 
 from .bounding_box import BoundingBox
 from .image_error import ImageError
+from .output_format import OutputFormat
 
 
 class ImageDataItem(BaseModel):
@@ -35,6 +36,10 @@ class ImageDataItem(BaseModel):
     """
     Per-image failure (set when this slot failed while others succeeded).
     """
+    output_format: Optional[OutputFormat] = None
+    """
+    Actual container/codec used for this image.
+    """
     z_index: Optional[int] = None
     """
     Layer stacking order. The base image is `0`; layers follow bottom-up.
diff --git a/src/arkruntime/types/images/usage.py b/src/arkruntime/types/images/usage.py
index d0eae3f..0d1cad0 100644
--- a/src/arkruntime/types/images/usage.py
+++ b/src/arkruntime/types/images/usage.py
@@ -14,6 +14,10 @@
 
 
 class Usage(BaseModel):
+    input_images: int
+    """
+    Number of reference images supplied by the client.
+    """
     generated_images: int
     """
     Number of images returned in `data`.