feat: add and delete child chunks

web/app/components/datasets/documents/detail/completed/child-segment-list.tsx

@@ -1,6 +1,5 @@
 import { type FC, useMemo, useState } from 'react'
 import { RiArrowDownSLine, RiArrowRightSLine } from '@remixicon/react'
-import { FormattedText } from '../../../formatted-text/formatted'
 import { EditSlice } from '../../../formatted-text/flavours/edit-slice'
 import { useDocumentContext } from '../index'
 import type { ChildChunkDetail } from '@/models/datasets'
@@ -10,14 +9,22 @@ import Divider from '@/app/components/base/divider'
 
 type IChildSegmentCardProps = {
   childChunks: ChildChunkDetail[]
-  handleInputChange: (value: string) => void
+  parentChunkId: string
+  handleInputChange?: (value: string) => void
+  handleAddNewChildChunk?: (parentChunkId: string) => void
   enabled: boolean
+  onDelete?: (segId: string, childChunkId: string) => Promise<void>
+  onClickSlice?: (childChunk: ChildChunkDetail) => void
 }
 
 const ChildSegmentList: FC<IChildSegmentCardProps> = ({
   childChunks,
+  parentChunkId,
   handleInputChange,
+  handleAddNewChildChunk,
   enabled,
+  onDelete,
+  onClickSlice,
 }) => {
   const parentMode = useDocumentContext(s => s.parentMode)
 
@@ -62,6 +69,7 @@ const ChildSegmentList: FC<IChildSegmentCardProps> = ({
             className={classNames('px-1.5 py-1 text-components-button-secondary-accent-text system-xs-semibold', isParagraphMode ? 'hidden group-hover/card:inline-block' : '')}
             onClick={(event) => {
               event.stopPropagation()
+              handleAddNewChildChunk?.(parentChunkId)
             }}
           >
             ADD
@@ -73,25 +81,26 @@ const ChildSegmentList: FC<IChildSegmentCardProps> = ({
             showClearIcon
             wrapperClassName='!w-52'
             value={''}
-            onChange={e => handleInputChange(e.target.value)}
+            onChange={e => handleInputChange?.(e.target.value)}
-            onClear={() => handleInputChange('')}
+            onClear={() => handleInputChange?.('')}
           />
           : null}
       </div>
       {(isFullDocMode || !collapsed)
         ? <div className={classNames('flex gap-x-0.5', isFullDocMode ? 'grow overflow-y-auto' : '')}>
           {isParagraphMode && <Divider type='vertical' className='h-auto w-[2px] mx-[7px] bg-text-accent-secondary' />}
-          <FormattedText className={classNames('w-full !leading-5 flex flex-col', isParagraphMode ? 'gap-y-2' : 'gap-y-3')}>
+          <div className={classNames('w-full !leading-5 flex flex-col', isParagraphMode ? 'gap-y-2' : 'gap-y-3')}>
             {childChunks.map((childChunk) => {
+              const edited = childChunk.type === 'customized'
               return <EditSlice
-                key={childChunk.segment_id}
+                key={childChunk.id}
-                label={`C-${childChunk.position}`}
+                label={`C-${childChunk.position}${edited ? '·EDITED' : ''}`}
                 text={childChunk.content}
-                onDelete={() => {}}
+                onDelete={() => onDelete?.(childChunk.segment_id, childChunk.id)}
-                className=''
+                onClick={() => onClickSlice?.(childChunk)}
               />
             })}
-          </FormattedText>
+          </div>
         </div>
         : null}
     </div>

web/app/components/datasets/documents/detail/completed/common/chunk-content.tsx

@@ -4,9 +4,9 @@ import AutoHeightTextarea from '@/app/components/base/auto-height-textarea/commo
 
 type IChunkContentProps = {
   question: string
-  answer: string
+  answer?: string
   onQuestionChange: (question: string) => void
-  onAnswerChange: (answer: string) => void
+  onAnswerChange?: (answer: string) => void
   isEditMode?: boolean
   docForm: string
 }
@@ -39,7 +39,7 @@ const ChunkContent: FC<IChunkContentProps> = ({
           className='leading-6 text-md text-gray-800'
           value={answer}
           placeholder={t('datasetDocuments.segment.answerPlaceholder') || ''}
-          onChange={e => onAnswerChange(e.target.value)}
+          onChange={e => onAnswerChange?.(e.target.value)}
           disabled={!isEditMode}
           autoFocus
         />

web/app/components/datasets/documents/detail/completed/display-toggle.tsx

@@ -1,11 +1,17 @@
 import React, { type FC } from 'react'
 import { RiLineHeight } from '@remixicon/react'
-import { useSegmentListContext } from '.'
 import Tooltip from '@/app/components/base/tooltip'
 import { Collapse } from '@/app/components/base/icons/src/public/knowledge'
 
-const DisplayToggle: FC = () => {
+type DisplayToggleProps = {
-  const [isCollapsed, toggleCollapsed] = useSegmentListContext(s => [s.isCollapsed, s.toggleCollapsed])
+  isCollapsed: boolean
+  toggleCollapsed: () => void
+}
+
+const DisplayToggle: FC<DisplayToggleProps> = ({
+  isCollapsed,
+  toggleCollapsed,
+}) => {
   return (
     <Tooltip
       popupContent={isCollapsed ? 'Expand chunks' : 'Collapse chunks'}

web/app/components/datasets/documents/detail/completed/index.tsx

@@ -13,6 +13,7 @@ import BatchAction from './batch-action'
 import SegmentDetail from './segment-detail'
 import SegmentCard from './segment-card'
 import ChildSegmentList from './child-segment-list'
+import NewChildSegment from './new-child-segment'
 import FullScreenDrawer from './common/full-screen-drawer'
 import Pagination from '@/app/components/base/pagination'
 import cn from '@/utils/classnames'
@@ -27,24 +28,27 @@ import type { ChildChunkDetail, SegmentDetailModel, SegmentUpdater } from '@/mod
 import NewSegment from '@/app/components/datasets/documents/detail/new-segment'
 import { useEventEmitterContextContext } from '@/context/event-emitter'
 import Checkbox from '@/app/components/base/checkbox'
-import { useChildSegmentList, useDeleteSegment, useDisableSegment, useEnableSegment, useSegmentList, useSegmentListKey } from '@/service/knowledge/use-segment'
+import {
+  useChildSegmentList,
+  useChildSegmentListKey,
+  useDeleteChildSegment,
+  useDeleteSegment,
+  useDisableSegment,
+  useEnableSegment,
+  useSegmentList,
+  useSegmentListKey,
+} from '@/service/knowledge/use-segment'
 import { useInvalid } from '@/service/use-base'
 
 const DEFAULT_LIMIT = 10
 
 type SegmentListContextValue = {
   isCollapsed: boolean
-  toggleCollapsed: () => void
   fullScreen: boolean
-  toggleFullScreen: () => void
+  toggleFullScreen: (fullscreen?: boolean) => void
 }
 
-const SegmentListContext = createContext({
+const SegmentListContext = createContext<SegmentListContextValue>({} as SegmentListContextValue)
-  isCollapsed: true,
-  toggleCollapsed: () => { },
-  fullScreen: false,
-  toggleFullScreen: () => { },
-})
 
 export const useSegmentListContext = (selector: (value: SegmentListContextValue) => any) => {
   return useContextSelector(SegmentListContext, selector)
@@ -73,6 +77,8 @@ const Completed: FC<ICompletedProps> = ({
   const [datasetId = '', documentId = '', docForm, mode, parentMode] = useDocumentContext(s => [s.datasetId, s.documentId, s.docForm, s.mode, s.parentMode])
   // the current segment id and whether to show the modal
   const [currSegment, setCurrSegment] = useState<{ segInfo?: SegmentDetailModel; showModal: boolean; isEditMode?: boolean }>({ showModal: false })
+  const [currChildChunk, setCurrChildChunk] = useState<{ childChunkInfo?: ChildChunkDetail; showModal: boolean }>({ showModal: false })
+  const [currChunkId, setCurrChunkId] = useState('')
 
   const [inputValue, setInputValue] = useState<string>('') // the input value
   const [searchValue, setSearchValue] = useState<string>('') // the search value
@@ -86,6 +92,8 @@ const Completed: FC<ICompletedProps> = ({
   const [currentPage, setCurrentPage] = useState(1) // start from 1
   const [limit, setLimit] = useState(DEFAULT_LIMIT)
   const [fullScreen, setFullScreen] = useState(false)
+  const [showNewChildSegmentModal, setShowNewChildSegmentModal] = useState(false)
+
   const segmentListRef = useRef<HTMLDivElement>(null)
   const needScrollToBottom = useRef(false)
 
@@ -136,7 +144,7 @@ const Completed: FC<ICompletedProps> = ({
     }
   }, [segments])
 
-  const { data: childChunkListData, refetch: refreshChildSegmentList } = useChildSegmentList(
+  const { data: childChunkListData } = useChildSegmentList(
     {
       datasetId,
       documentId,
@@ -149,6 +157,7 @@ const Completed: FC<ICompletedProps> = ({
     },
     !isFullDocMode || segments.length === 0,
   )
+  const invalidChildSegmentList = useInvalid(useChildSegmentListKey)
 
   useEffect(() => {
     if (childChunkListData)
@@ -162,6 +171,12 @@ const Completed: FC<ICompletedProps> = ({
     // eslint-disable-next-line react-hooks/exhaustive-deps
   }, [])
 
+  const resetChildList = useCallback(() => {
+    setChildSegments([])
+    invalidChildSegmentList()
+    // eslint-disable-next-line react-hooks/exhaustive-deps
+  }, [])
+
   const onClickCard = (detail: SegmentDetailModel, isEditMode = false) => {
     setCurrSegment({ segInfo: detail, showModal: true, isEditMode })
   }
@@ -320,10 +335,54 @@ const Completed: FC<ICompletedProps> = ({
     // eslint-disable-next-line react-hooks/exhaustive-deps
   }, [segmentListData, limit, currentPage])
 
+  const { mutateAsync: deleteChildSegment } = useDeleteChildSegment()
+
+  const onDeleteChildChunk = useCallback(async (segmentId: string, childChunkId: string) => {
+    await deleteChildSegment(
+      { datasetId, documentId, segmentId, childChunkId },
+      {
+        onSuccess: () => {
+          notify({ type: 'success', message: t('common.actionMsg.modifiedSuccessfully') })
+          if (parentMode === 'paragraph')
+            resetList()
+          else
+            resetChildList()
+        },
+        onError: () => {
+          notify({ type: 'error', message: t('common.actionMsg.modifiedUnsuccessfully') })
+        },
+      },
+    )
+  // eslint-disable-next-line react-hooks/exhaustive-deps
+  }, [datasetId, documentId])
+
+  const handleAddNewChildChunk = useCallback((parentChunkId: string) => {
+    setShowNewChildSegmentModal(true)
+    setCurrChunkId(parentChunkId)
+  }, [])
+
+  const viewNewlyAddedChildChunk = useCallback(() => {
+    const totalPages = childChunkListData?.total_pages || 0
+    const total = childChunkListData?.total || 0
+    const newPage = Math.ceil((total + 1) / limit)
+    needScrollToBottom.current = true
+    if (newPage > totalPages) {
+      setCurrentPage(totalPages + 1)
+    }
+    else {
+      resetChildList()
+      currentPage !== totalPages && setCurrentPage(totalPages)
+    }
+    // eslint-disable-next-line react-hooks/exhaustive-deps
+  }, [childChunkListData, limit, currentPage])
+
+  const onClickSlice = useCallback((detail: ChildChunkDetail) => {
+    setCurrChildChunk({ childChunkInfo: detail, showModal: true })
+  }, [])
+
   return (
     <SegmentListContext.Provider value={{
       isCollapsed,
-      toggleCollapsed: () => setIsCollapsed(!isCollapsed),
       fullScreen,
       toggleFullScreen,
     }}>
@@ -355,7 +414,7 @@ const Completed: FC<ICompletedProps> = ({
           onClear={() => handleInputChange('')}
         />
         <Divider type='vertical' className='h-3.5 mx-3' />
-        <DisplayToggle />
+        <DisplayToggle isCollapsed={isCollapsed} toggleCollapsed={() => setIsCollapsed(!isCollapsed)} />
       </div>}
       {/* Segment list */}
       {
@@ -367,8 +426,12 @@ const Completed: FC<ICompletedProps> = ({
               loading={false}
             />
             <ChildSegmentList
+              parentChunkId={segments[0]?.id}
+              onDelete={onDeleteChildChunk}
               childChunks={childSegments}
-              handleInputChange={() => { }}
+              handleInputChange={handleInputChange}
+              handleAddNewChildChunk={handleAddNewChildChunk}
+              onClickSlice={onClickSlice}
               enabled={!archived}
             />
           </div>
@@ -383,6 +446,9 @@ const Completed: FC<ICompletedProps> = ({
             onDelete={onDelete}
             onClick={onClickCard}
             archived={archived}
+            onDeleteChildChunk={onDeleteChildChunk}
+            handleAddNewChildChunk={handleAddNewChildChunk}
+            onClickSlice={onClickSlice}
           />
       }
       {/* Pagination */}
@@ -421,6 +487,26 @@ const Completed: FC<ICompletedProps> = ({
           viewNewlyAddedChunk={viewNewlyAddedChunk}
         />
       </FullScreenDrawer>
+      {/* Create New Child Segment */}
+      <FullScreenDrawer
+        isOpen={showNewChildSegmentModal}
+        fullScreen={fullScreen}
+      >
+        <NewChildSegment
+          chunkId={currChunkId}
+          onCancel={() => {
+            setShowNewChildSegmentModal(false)
+            setFullScreen(false)
+          }}
+          onSave={() => {
+            if (parentMode === 'paragraph')
+              resetList()
+            else
+              resetChildList()
+          }}
+          viewNewlyAddedChildChunk={viewNewlyAddedChildChunk}
+        />
+      </FullScreenDrawer>
       {/* Batch Action Buttons */}
       {selectedSegmentIds.length > 0
         && <BatchAction

web/app/components/datasets/documents/detail/completed/new-child-segment.tsx

@@ -0,0 +1,151 @@
+import { memo, useRef, useState } from 'react'
+import type { FC } from 'react'
+import { useTranslation } from 'react-i18next'
+import { useContext } from 'use-context-selector'
+import { useParams } from 'next/navigation'
+import { RiCloseLine, RiExpandDiagonalLine } from '@remixicon/react'
+import { useShallow } from 'zustand/react/shallow'
+import { SegmentIndexTag } from './common/segment-index-tag'
+import ActionButtons from './common/action-buttons'
+import ChunkContent from './common/chunk-content'
+import AddAnother from './common/add-another'
+import { useSegmentListContext } from './index'
+import { useStore as useAppStore } from '@/app/components/app/store'
+import { ToastContext } from '@/app/components/base/toast'
+import type { SegmentUpdater } from '@/models/datasets'
+import classNames from '@/utils/classnames'
+import { formatNumber } from '@/utils/format'
+import Divider from '@/app/components/base/divider'
+import { useAddChildSegment } from '@/service/knowledge/use-segment'
+
+type NewChildSegmentModalProps = {
+  chunkId: string
+  onCancel: () => void
+  onSave: () => void
+  viewNewlyAddedChildChunk?: () => void
+}
+
+const NewChildSegmentModal: FC<NewChildSegmentModalProps> = ({
+  chunkId,
+  onCancel,
+  onSave,
+  viewNewlyAddedChildChunk,
+}) => {
+  const { t } = useTranslation()
+  const { notify } = useContext(ToastContext)
+  const [content, setContent] = useState('')
+  const { datasetId, documentId } = useParams<{ datasetId: string; documentId: string }>()
+  const [loading, setLoading] = useState(false)
+  const [addAnother, setAddAnother] = useState(true)
+  const [fullScreen, toggleFullScreen] = useSegmentListContext(s => [s.fullScreen, s.toggleFullScreen])
+  const { appSidebarExpand } = useAppStore(useShallow(state => ({
+    appSidebarExpand: state.appSidebarExpand,
+  })))
+  const refreshTimer = useRef<any>(null)
+
+  const CustomButton = <>
+    <Divider type='vertical' className='h-3 mx-1 bg-divider-regular' />
+    <button className='text-text-accent system-xs-semibold' onClick={() => {
+      clearTimeout(refreshTimer.current)
+      viewNewlyAddedChildChunk?.()
+    }}>
+      {t('datasetDocuments.segment.viewAddedChunk')}
+    </button>
+  </>
+
+  const handleCancel = (actionType: 'esc' | 'add' = 'esc') => {
+    if (actionType === 'esc' || !addAnother)
+      onCancel()
+    setContent('')
+  }
+
+  const { mutateAsync: addChildSegment } = useAddChildSegment()
+
+  const handleSave = async () => {
+    const params: SegmentUpdater = { content: '' }
+
+    if (!content.trim())
+      return notify({ type: 'error', message: t('datasetDocuments.segment.contentEmpty') })
+
+    params.content = content
+
+    setLoading(true)
+    try {
+      await addChildSegment({ datasetId, documentId, segmentId: chunkId, body: params })
+      notify({
+        type: 'success',
+        message: t('datasetDocuments.segment.chunkAdded'),
+        className: `!w-[296px] !bottom-0 ${appSidebarExpand === 'expand' ? '!left-[216px]' : '!left-14'}
+          !top-auto !right-auto !mb-[52px] !ml-11`,
+        customComponent: CustomButton,
+      })
+      handleCancel('add')
+      refreshTimer.current = setTimeout(() => {
+        onSave()
+      }, 3000)
+    }
+    finally {
+      setLoading(false)
+    }
+  }
+
+  return (
+    <div className={'flex flex-col h-full'}>
+      <div className={classNames('flex items-center justify-between', fullScreen ? 'py-3 pr-4 pl-6 border border-divider-subtle' : 'pt-3 pr-3 pl-4')}>
+        <div className='flex flex-col'>
+          <div className='text-text-primary system-xl-semibold'>{
+            t('datasetDocuments.segment.addChildChunk')
+          }</div>
+          <div className='flex items-center gap-x-2'>
+            <SegmentIndexTag label={'New Child Chunk'} />
+            <span className='text-text-quaternary system-xs-medium'>·</span>
+            <span className='text-text-tertiary system-xs-medium'>{formatNumber(content.length)} {t('datasetDocuments.segment.characters')}</span>
+          </div>
+        </div>
+        <div className='flex items-center'>
+          {fullScreen && (
+            <>
+              <AddAnother className='mr-3' isChecked={addAnother} onCheck={() => setAddAnother(!addAnother)} />
+              <ActionButtons
+                handleCancel={handleCancel.bind(null, 'esc')}
+                handleSave={handleSave}
+                loading={loading}
+                actionType='add'
+              />
+              <Divider type='vertical' className='h-3.5 bg-divider-regular ml-4 mr-2' />
+            </>
+          )}
+          <div className='w-8 h-8 flex justify-center items-center p-1.5 cursor-pointer mr-1' onClick={toggleFullScreen}>
+            <RiExpandDiagonalLine className='w-4 h-4 text-text-tertiary' />
+          </div>
+          <div className='w-8 h-8 flex justify-center items-center p-1.5 cursor-pointer' onClick={handleCancel.bind(null, 'esc')}>
+            <RiCloseLine className='w-4 h-4 text-text-tertiary' />
+          </div>
+        </div>
+      </div>
+      <div className={classNames('flex grow overflow-hidden', fullScreen ? 'w-full flex-row justify-center px-6 pt-6 gap-x-8' : 'flex-col gap-y-1 py-3 px-4')}>
+        <div className={classNames('break-all overflow-y-auto whitespace-pre-line', fullScreen ? 'w-1/2' : 'grow')}>
+          <ChunkContent
+            docForm=''
+            question={content}
+            onQuestionChange={content => setContent(content)}
+            isEditMode={true}
+          />
+        </div>
+      </div>
+      {!fullScreen && (
+        <div className='flex items-center justify-between p-4 pt-3 border-t-[1px] border-t-divider-subtle'>
+          <AddAnother isChecked={addAnother} onCheck={() => setAddAnother(!addAnother)} />
+          <ActionButtons
+            handleCancel={handleCancel.bind(null, 'esc')}
+            handleSave={handleSave}
+            loading={loading}
+            actionType='add'
+          />
+        </div>
+      )}
+    </div>
+  )
+}
+
+export default memo(NewChildSegmentModal)

web/app/components/datasets/documents/detail/completed/segment-card.tsx

@@ -8,7 +8,7 @@ import Tag from './common/tag'
 import Dot from './common/dot'
 import { SegmentIndexTag } from './common/segment-index-tag'
 import { useSegmentListContext } from './index'
-import type { SegmentDetailModel } from '@/models/datasets'
+import type { ChildChunkDetail, SegmentDetailModel } from '@/models/datasets'
 import Indicator from '@/app/components/header/indicator'
 import Switch from '@/app/components/base/switch'
 import Divider from '@/app/components/base/divider'
@@ -25,6 +25,9 @@ type ISegmentCardProps = {
   onClick?: () => void
   onChangeSwitch?: (enabled: boolean, segId?: string) => Promise<void>
   onDelete?: (segId: string) => Promise<void>
+  onDeleteChildChunk?: (segId: string, childChunkId: string) => Promise<void>
+  handleAddNewChildChunk?: (parentChunkId: string) => void
+  onClickSlice?: (childChunk: ChildChunkDetail) => void
   onClickEdit?: () => void
   className?: string
   archived?: boolean
@@ -36,6 +39,9 @@ const SegmentCard: FC<ISegmentCardProps> = ({
   onClick,
   onChangeSwitch,
   onDelete,
+  onDeleteChildChunk,
+  handleAddNewChildChunk,
+  onClickSlice,
   onClickEdit,
   loading = true,
   className = '',
@@ -216,9 +222,12 @@ const SegmentCard: FC<ISegmentCardProps> = ({
             {
               child_chunks.length > 0
               && <ChildSegmentList
+                parentChunkId={id}
                 childChunks={child_chunks}
-                handleInputChange={() => {}}
                 enabled={enabled}
+                onDelete={onDeleteChildChunk!}
+                handleAddNewChildChunk={handleAddNewChildChunk}
+                onClickSlice={onClickSlice}
               />
             }
           </>

web/app/components/datasets/documents/detail/completed/segment-list.tsx

@@ -1,6 +1,6 @@
 import React, { type ForwardedRef } from 'react'
 import SegmentCard from './segment-card'
-import type { SegmentDetailModel } from '@/models/datasets'
+import type { ChildChunkDetail, SegmentDetailModel } from '@/models/datasets'
 import Checkbox from '@/app/components/base/checkbox'
 import Loading from '@/app/components/base/loading'
 import Divider from '@/app/components/base/divider'
@@ -14,6 +14,9 @@ type ISegmentListProps = {
   onClick: (detail: SegmentDetailModel, isEditMode?: boolean) => void
   onChangeSwitch: (enabled: boolean, segId?: string,) => Promise<void>
   onDelete: (segId: string) => Promise<void>
+  onDeleteChildChunk: (sgId: string, childChunkId: string) => Promise<void>
+  handleAddNewChildChunk: (parentChunkId: string) => void
+  onClickSlice: (childChunk: ChildChunkDetail) => void
   archived?: boolean
   embeddingAvailable: boolean
 }
@@ -26,6 +29,9 @@ const SegmentList = React.forwardRef(({
   onClick: onClickCard,
   onChangeSwitch,
   onDelete,
+  onDeleteChildChunk,
+  handleAddNewChildChunk,
+  onClickSlice,
   archived,
   embeddingAvailable,
 }: ISegmentListProps,
@@ -54,6 +60,9 @@ ref: ForwardedRef<HTMLDivElement>,
                   onChangeSwitch={onChangeSwitch}
                   onClickEdit={() => onClickCard(segItem, true)}
                   onDelete={onDelete}
+                  onDeleteChildChunk={onDeleteChildChunk}
+                  handleAddNewChildChunk={handleAddNewChildChunk}
+                  onClickSlice={onClickSlice}
                   loading={false}
                   archived={archived}
                   embeddingAvailable={embeddingAvailable}

web/app/components/datasets/formatted-text/flavours/edit-slice.tsx

@@ -10,10 +10,11 @@ import ActionButton, { ActionButtonState } from '@/app/components/base/action-bu
 type EditSliceProps = SliceProps<{
   label: ReactNode
   onDelete: () => void
+  onClick?: () => void
 }>
 
 export const EditSlice: FC<EditSliceProps> = (props) => {
-  const { label, className, text, onDelete, ...rest } = props
+  const { label, className, text, onDelete, onClick, ...rest } = props
   const [delBtnShow, setDelBtnShow] = useState(false)
   const [isDelBtnHover, setDelBtnHover] = useState(false)
 
@@ -35,7 +36,10 @@ export const EditSlice: FC<EditSliceProps> = (props) => {
   const isDestructive = delBtnShow && isDelBtnHover
 
   return (
-    <div>
+    <div onClick={(e) => {
+      e.stopPropagation()
+      onClick?.()
+    }}>
       <SliceContainer {...rest}
         className={className}
         ref={refs.setReference}

web/service/knowledge/use-segment.ts

@@ -85,7 +85,7 @@ export const useChildSegmentList = (
   return useQuery({
     queryKey: [...useChildSegmentListKey, datasetId, documentId, segmentId, page, limit, keyword],
     queryFn: () => {
-      return get<ChildSegmentsResponse>(`/datasets/${datasetId}/documents/${documentId}/segment/${segmentId}/child_chunks`, { params })
+      return get<ChildSegmentsResponse>(`/datasets/${datasetId}/documents/${documentId}/segments/${segmentId}/child_chunks`, { params })
     },
     enabled: !disable,
     initialData: disable ? { data: [], total: 0, page: 1, total_pages: 0, limit: 10 } : undefined,
@@ -97,7 +97,7 @@ export const useDeleteChildSegment = () => {
     mutationKey: [NAME_SPACE, 'childChunk', 'delete'],
     mutationFn: (payload: { datasetId: string; documentId: string; segmentId: string; childChunkId: string }) => {
       const { datasetId, documentId, segmentId, childChunkId } = payload
-      return del<CommonResponse>(`/datasets/${datasetId}/documents/${documentId}/segment/${segmentId}/child_chunks/${childChunkId}`)
+      return del<CommonResponse>(`/datasets/${datasetId}/documents/${documentId}/segments/${segmentId}/child_chunks/${childChunkId}`)
     },
   })
 }
@@ -107,7 +107,17 @@ export const useAddChildSegment = () => {
     mutationKey: [NAME_SPACE, 'childChunk', 'add'],
     mutationFn: (payload: { datasetId: string; documentId: string; segmentId: string; body: { content: string } }) => {
       const { datasetId, documentId, segmentId, body } = payload
-      return post<{ data: ChildChunkDetail }>(`/datasets/${datasetId}/documents/${documentId}/segment/${segmentId}/child_chunks`, { body })
+      return post<{ data: ChildChunkDetail }>(`/datasets/${datasetId}/documents/${documentId}/segments/${segmentId}/child_chunks`, { body })
+    },
+  })
+}
+
+export const useUpdateChildSegment = () => {
+  return useMutation({
+    mutationKey: [NAME_SPACE, 'childChunk', 'update'],
+    mutationFn: (payload: { datasetId: string; documentId: string; segmentId: string; childChunkId: string; body: { content: string } }) => {
+      const { datasetId, documentId, segmentId, childChunkId, body } = payload
+      return patch<{ data: ChildChunkDetail }>(`/datasets/${datasetId}/documents/${documentId}/segments/${segmentId}/child_chunks/${childChunkId}`, { body })
     },
   })
 }