diff --git a/libs/shared-entity/src/dto/billing_dto.rs b/libs/shared-entity/src/dto/billing_dto.rs
index ef36ccfc..9794371e 100644
--- a/libs/shared-entity/src/dto/billing_dto.rs
+++ b/libs/shared-entity/src/dto/billing_dto.rs
@@ -117,6 +117,10 @@ pub struct WorkspaceUsageAndLimit {
   pub single_upload_unlimited: bool,
   pub ai_responses_count: i64,
   pub ai_responses_count_limit: i64,
+  #[serde(default)]
+  pub ai_image_responses_count: i64,
+  #[serde(default)]
+  pub ai_image_responses_count_limit: i64,
 
   pub local_ai: bool,
   pub ai_responses_unlimited: bool,