youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-05 09:58:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: finish add chunk mode

Browse Source
This commit is contained in:
Joel
2024-12-05 14:08:45 +08:00
parent 1578dc50ef
commit 8a8fc7ab50
6 changed files with 55 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
web/app/components/datasets/common/chunking-mode-label.tsx Normal file
View File

@ -0,0 +1,29 @@
'use client'
import type { FC } from 'react'
import React from 'react'
import { useTranslation } from 'react-i18next'
import Badge from '@/app/components/base/badge'
import { GeneralType, ParentChildType } from '@/app/components/base/icons/src/public/knowledge'
type Props = {
isGeneralMode: boolean
isQAMode: boolean
}
const ChunkingModeLabel: FC<Props> = ({
isGeneralMode,
isQAMode,
}) => {
const { t } = useTranslation()
const TypeIcon = isGeneralMode ? GeneralType : ParentChildType
return (
<Badge>
<div className='flex items-center h-full space-x-0.5 text-text-tertiary'>
<TypeIcon className='w-3 h-3' />
<span className='system-2xs-medium-uppercase'>{isGeneralMode ? `${t('dataset.chunkingMode.general')}${isQAMode ? ' · QA' : ''}` : t('dataset.chunkingMode.parentChild')}</span>
</div>
</Badge>
)
}
export default React.memo(ChunkingModeLabel)

9
web/app/components/datasets/common/document-picker/index.tsx
View File

@ -3,8 +3,10 @@ import type { FC } from 'react'
import React, { useState } from 'react'
import { useBoolean } from 'ahooks'
import { RiArrowDownSLine, RiArrowUpSLine } from '@remixicon/react'
import { useTranslation } from 'react-i18next'
import FileIcon from '../document-file-icon'
import type { ParentMode, ProcessMode, SimpleDocumentDetail } from '@/models/datasets'
import type { ParentMode, SimpleDocumentDetail } from '@/models/datasets'
import { ProcessMode } from '@/models/datasets'
import {
PortalToFollowElem,
PortalToFollowElemContent,
@ -32,6 +34,7 @@ const DocumentPicker: FC<Props> = ({
value,
onChange,
}) => {
const { t } = useTranslation()
const {
name,
extension,
@ -49,7 +52,7 @@ const DocumentPicker: FC<Props> = ({
},
})
const documentsList = data?.data
const isParentChild = processMode === 'hierarchical'
const isParentChild = processMode === ProcessMode.parentChild
const TypeIcon = isParentChild ? ParentChildType : GeneralType
const [open, {
@ -75,7 +78,7 @@ const DocumentPicker: FC<Props> = ({
<div className='flex items-center h-3 text-text-tertiary space-x-0.5'>
<TypeIcon className='w-3 h-3' />
<span className={cn('system-2xs-medium-uppercase', isParentChild && 'mt-0.5' /* to icon problem cause not ver align */)}>
{isParentChild ? 'Parent-Child' : 'General'}
{isParentChild ? t('dataset.chunkingMode.parentChild') : t('dataset.chunkingMode.general')}
{isParentChild && ` · ${parentMode || '--'}`}
</span>
</div>