From f62b2b5b45d00ed66971c55b7b7b9b6a06a1e813 Mon Sep 17 00:00:00 2001 From: "Charlie.Wei" Date: Fri, 26 Apr 2024 11:47:23 +0800 Subject: [PATCH] optimize the knowledge failed documents query (#3870) Co-authored-by: luowei Co-authored-by: crazywoola <427733928@qq.com> Co-authored-by: crazywoola <100913391+crazywoola@users.noreply.github.com> --- api/services/dataset_service.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/api/services/dataset_service.py b/api/services/dataset_service.py index aaf6076d1..fa2ed8088 100644 --- a/api/services/dataset_service.py +++ b/api/services/dataset_service.py @@ -418,9 +418,8 @@ class DocumentService: def get_error_documents_by_dataset_id(dataset_id: str) -> list[Document]: documents = db.session.query(Document).filter( Document.dataset_id == dataset_id, - Document.indexing_status == 'error' or Document.indexing_status == 'paused' + Document.indexing_status.in_(['error', 'paused']) ).all() - return documents @staticmethod