Fix/segment create with api (#7928)

2 years ago · 1a5116cba0
parent 01581dd35f
commit 1a5116cba0
1 changed files with 4 additions and 0 deletions
--- a/api/controllers/service_api/dataset/segment.py
+++ b/api/controllers/service_api/dataset/segment.py
@ -36,6 +36,10 @@ class SegmentApi(DatasetApiResource):
        document = DocumentService.get_document(dataset.id, document_id)
        if not document:
            raise NotFound("Document not found.")
+        if document.indexing_status != "completed":
+            raise NotFound("Document is already completed.")
+        if not document.enabled:
+            raise NotFound("Document is disabled.")
        # check embedding model setting
        if dataset.indexing_technique == "high_quality":
            try: