yujunjun/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2025-08-05 07:48:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

r2

Browse Source
This commit is contained in:
jyong 2025-06-10 17:59:14 +08:00
parent eb7479b1ea
commit aeb1d1946c
2 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/fields/dataset_fields.py
View File

@ -88,6 +88,8 @@ dataset_detail_fields = {
"chunk_structure": fields.String, "chunk_structure": fields.String,
"icon_info": fields.Nested(icon_info_fields), "icon_info": fields.Nested(icon_info_fields),
"is_published": fields.Boolean, "is_published": fields.Boolean,
"total_documents": fields.Integer,
"total_available_documents": fields.Integer,
} }
dataset_query_detail_fields = { dataset_query_detail_fields = {

13
api/models/dataset.py
View File

@ -312,6 +312,19 @@ class DatasetProcessRule(Base):
except JSONDecodeError: except JSONDecodeError:
return None 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): class Document(Base):
__tablename__ = "documents" __tablename__ = "documents"