From a50fe0ee29749cb00177bac62e51aff567b573fb Mon Sep 17 00:00:00 2001 From: "lengjs@wsidomidata.com" Date: Mon, 30 Jun 2025 14:46:38 +0800 Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E6=9F=A5=E8=AF=A2=E8=AF=B4?= =?UTF-8?q?=E6=9C=89=E6=96=87=E4=BB=B6=E6=8E=A5=E5=8F=A3?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- api/controllers/service_api/dataset/document.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/api/controllers/service_api/dataset/document.py b/api/controllers/service_api/dataset/document.py index 1a2d46140e..6d061f13de 100644 --- a/api/controllers/service_api/dataset/document.py +++ b/api/controllers/service_api/dataset/document.py @@ -372,6 +372,18 @@ class DocumentListApi(DatasetApiResource): return response +class DocumentListApi2(DatasetApiResource): + def get(self, tenant_id, dataset_id): + dataset_id = str(dataset_id) + tenant_id = str(tenant_id) + dataset = db.session.query(Dataset).filter(Dataset.tenant_id == tenant_id, Dataset.id == dataset_id).first() + if not dataset: + raise NotFound("未找到数据集") + documents = db.session.query(Document).filter(Document.dataset_id == dataset_id, Document.tenant_id == tenant_id).all() + + return { + "data": marshal(documents, document_fields) + } class DocumentIndexingStatusApi(DatasetApiResource): def get(self, tenant_id, dataset_id, batch): @@ -444,4 +456,5 @@ api.add_resource( ) api.add_resource(DocumentDeleteApi, "/datasets//documents/") api.add_resource(DocumentListApi, "/datasets//documents") +api.add_resource(DocumentListApi2, "/datasets//documents/list") api.add_resource(DocumentIndexingStatusApi, "/datasets//documents//indexing-status")