From 9eb8597957334a474b3f016bb4bbb5866867d969 Mon Sep 17 00:00:00 2001
From: jyong <718720800@qq.com>
Date: Wed, 11 Jun 2025 17:29:14 +0800
Subject: [PATCH] r2

---
 api/models/dataset.py | 27 +++++++++++++--------------
 1 file changed, 13 insertions(+), 14 deletions(-)

diff --git a/api/models/dataset.py b/api/models/dataset.py
index da6e58f113..5d18eaff49 100644
--- a/api/models/dataset.py
+++ b/api/models/dataset.py
@@ -69,6 +69,19 @@ class Dataset(Base):
     pipeline_id = db.Column(StringUUID, nullable=True)
     chunk_structure = db.Column(db.String(255), nullable=True)
 
+    @property
+    def total_documents(self):
+        return db.session.query(func.count(Document.id)).filter(Document.dataset_id == self.id).scalar()
+
+    @property
+    def total_available_documents(self):
+        return db.session.query(func.count(Document.id)).filter(
+            Document.dataset_id == self.id,
+            Document.indexing_status == "completed",
+            Document.enabled == True,
+            Document.archived == False,
+        ).scalar()
+
     @property
     def dataset_keyword_table(self):
         dataset_keyword_table = (
@@ -311,20 +324,6 @@ class DatasetProcessRule(Base):
             return json.loads(self.rules) if self.rules else None
         except JSONDecodeError:
             return None
-        
-    @property
-    def total_documents(self):
-        return db.session.query(func.count(Document.id)).filter(Document.dataset_id == self.dataset_id).scalar()
-
-    @property
-    def total_available_documents(self):
-        return db.session.query(func.count(Document.id)).filter(
-            Document.dataset_id == self.dataset_id,
-            Document.indexing_status == "completed",
-            Document.enabled == True,
-            Document.archived == False,
-        ).scalar()
-
 
 class Document(Base):
     __tablename__ = "documents"