Fix/segment create with api (langgenius#7928)

cuiks · Sep 26, 2024 · 98f3efc · 98f3efc
1 parent 77bdda0
commit 98f3efc
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: