From 11e9d07226cfa99624689b463c39aa36c1d48f01 Mon Sep 17 00:00:00 2001
From: aqiu <819110812@qq.com>
Date: Sun, 29 Jun 2025 14:10:57 +0800
Subject: [PATCH] fix: correct segment settings when creating documents via API

---
 web/app/components/datasets/create/step-two/index.tsx | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/web/app/components/datasets/create/step-two/index.tsx b/web/app/components/datasets/create/step-two/index.tsx
index 9a76bc6834..cc97dcd6c4 100644
--- a/web/app/components/datasets/create/step-two/index.tsx
+++ b/web/app/components/datasets/create/step-two/index.tsx
@@ -513,7 +513,10 @@ const StepTwo = ({
       setRules(rules.pre_processing_rules)
       setDefaultConfig(rules)
 
-      if (documentDetail.dataset_process_rule.mode === 'hierarchical') {
+      const isHierarchicalDocument = documentDetail.doc_form === 'hierarchical_model'
+                              || (rules.parent_mode && rules.subchunk_segmentation)
+
+      if (isHierarchicalDocument) {
         setParentChildConfig({
           chunkForContext: rules.parent_mode || 'paragraph',
           parent: {