fix: correct segment settings when creating documents via API

11 months ago · 11e9d07226
parent 787ad5ab38
commit 11e9d07226
1 changed files with 4 additions and 1 deletions
--- 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: {