From dc3068da2c50b10e2199c5f638d534d9bd941d04 Mon Sep 17 00:00:00 2001
From: jyong <718720800@qq.com>
Date: Tue, 3 Sep 2024 18:02:16 +0800
Subject: [PATCH] uncompleted and disabled documents can't create segment with
 api

---
 api/controllers/service_api/dataset/segment.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/api/controllers/service_api/dataset/segment.py b/api/controllers/service_api/dataset/segment.py
index 5e10f3b48c..570af339bd 100644
--- 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: