Fix/segment create with api (#7928)

langgenius · Sep 3, 2024 · 1a5116c · 1a5116c
1 parent 01581dd
commit 1a5116c
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/api/controllers/service_api/dataset/segment.py b/api/controllers/service_api/dataset/segment.py
@@ -36,6 +36,10 @@ def post(self, tenant_id, dataset_id, document_id):
         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: