From f2f6e958805aaf90f2a1b70cc6eac4c4628c4383 Mon Sep 17 00:00:00 2001
From: onlylhf <27225745+onlylhf@users.noreply.github.com>
Date: Mon, 23 Jun 2025 10:36:30 +0800
Subject: [PATCH] fix: Update the logic for segment type settings to support
 parent-child mode selection. (#21278)

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

diff --git a/web/app/components/datasets/create/step-two/index.tsx b/web/app/components/datasets/create/step-two/index.tsx
index c931addd1..ec053ccbe 100644
--- a/web/app/components/datasets/create/step-two/index.tsx
+++ b/web/app/components/datasets/create/step-two/index.tsx
@@ -207,7 +207,14 @@ const StepTwo = ({
     }
     if (value === ChunkingMode.parentChild && indexType === IndexingType.ECONOMICAL)
       setIndexType(IndexingType.QUALIFIED)
+
     setDocForm(value)
+
+    if (value === ChunkingMode.parentChild)
+      setSegmentationType(ProcessMode.parentChild)
+    else
+      setSegmentationType(ProcessMode.general)
+
     // eslint-disable-next-line ts/no-use-before-define
     currentEstimateMutation.reset()
   }