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")