refactor hardcode string into enum

11 months ago · d742e97dc6
parent 97e0e0bce0
commit d742e97dc6
1 changed files with 1 additions and 1 deletions
--- a/web/app/components/datasets/create/step-two/index.tsx
+++ b/web/app/components/datasets/create/step-two/index.tsx
@ -507,7 +507,7 @@ const StepTwo = ({
      const separator = rules.segmentation.separator
      const max = rules.segmentation.max_tokens
      const overlap = rules.segmentation.chunk_overlap
-      const isHierarchicalDocument = documentDetail.doc_form === 'hierarchical_model'
+      const isHierarchicalDocument = documentDetail.doc_form === ChunkingMode.parentChild
                              || (rules.parent_mode && rules.subchunk_segmentation)
      setSegmentIdentifier(separator)
      setMaxChunkLength(max)