From 5d9efb42433db33d593987efe07259fcff123e60 Mon Sep 17 00:00:00 2001
From: "Nathan.fooo" <86001920+appflowy@users.noreply.github.com>
Date: Mon, 20 Jan 2025 19:36:31 +0800
Subject: [PATCH] Return ai image count (#1182)

* chore: return ai image count
---
 libs/shared-entity/src/dto/billing_dto.rs | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/libs/shared-entity/src/dto/billing_dto.rs b/libs/shared-entity/src/dto/billing_dto.rs
index ef36ccfc5..9794371ef 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,