mirror of https://github.com/langgenius/dify.git
49 lines
1.4 KiB
TypeScript
49 lines
1.4 KiB
TypeScript
import type { Option } from './types'
|
|
import { useTranslation } from 'react-i18next'
|
|
import {
|
|
GeneralChunk,
|
|
ParentChildChunk,
|
|
QuestionAndAnswer,
|
|
} from '@/app/components/base/icons/src/vender/knowledge'
|
|
import { ChunkingMode } from '@/models/datasets'
|
|
import { EffectColor } from './types'
|
|
|
|
export const useChunkStructure = () => {
|
|
const { t } = useTranslation()
|
|
|
|
const GeneralOption: Option = {
|
|
id: ChunkingMode.text,
|
|
icon: <GeneralChunk className="size-[18px]" />,
|
|
iconActiveColor: 'text-util-colors-indigo-indigo-600',
|
|
title: 'General',
|
|
description: t('stepTwo.generalTip', { ns: 'datasetCreation' }),
|
|
effectColor: EffectColor.indigo,
|
|
showEffectColor: true,
|
|
}
|
|
const ParentChildOption: Option = {
|
|
id: ChunkingMode.parentChild,
|
|
icon: <ParentChildChunk className="size-[18px]" />,
|
|
iconActiveColor: 'text-util-colors-blue-light-blue-light-500',
|
|
title: 'Parent-Child',
|
|
description: t('stepTwo.parentChildTip', { ns: 'datasetCreation' }),
|
|
effectColor: EffectColor.blueLight,
|
|
showEffectColor: true,
|
|
}
|
|
const QuestionAnswerOption: Option = {
|
|
id: ChunkingMode.qa,
|
|
icon: <QuestionAndAnswer className="size-[18px]" />,
|
|
title: 'Q&A',
|
|
description: t('stepTwo.qaTip', { ns: 'datasetCreation' }),
|
|
}
|
|
|
|
const options = [
|
|
GeneralOption,
|
|
ParentChildOption,
|
|
QuestionAnswerOption,
|
|
]
|
|
|
|
return {
|
|
options,
|
|
}
|
|
}
|