dify/web/hooks/use-knowledge.ts

import type { I18nKeysByPrefix } from '@/types/i18n'
import { useCallback } from 'react'
import { useTranslation } from 'react-i18next'

type IndexingTechnique = I18nKeysByPrefix<'dataset', 'indexingTechnique.'>
type IndexingMethod = I18nKeysByPrefix<'dataset', 'indexingMethod.'>

export const useKnowledge = () => {
  const { t } = useTranslation()

  const formatIndexingTechnique = useCallback(
    (indexingTechnique: IndexingTechnique) => {
      return t(($) => $[`indexingTechnique.${indexingTechnique}`], { ns: 'dataset' }) as string
    },
    [t],
  )

  const formatIndexingMethod = useCallback(
    (indexingMethod: IndexingMethod, isEco?: boolean) => {
      if (isEco) return t(($) => $['indexingMethod.invertedIndex'], { ns: 'dataset' })

      return t(($) => $[`indexingMethod.${indexingMethod}`], { ns: 'dataset' }) as string
    },
    [t],
  )

  const formatIndexingTechniqueAndMethod = useCallback(
    (indexingTechnique: IndexingTechnique, indexingMethod: IndexingMethod) => {
      let result = formatIndexingTechnique(indexingTechnique)

      if (indexingMethod)
        result += ` · ${formatIndexingMethod(indexingMethod, indexingTechnique === 'economy')}`

      return result
    },
    [formatIndexingTechnique, formatIndexingMethod],
  )

  return {
    formatIndexingTechnique,
    formatIndexingMethod,
    formatIndexingTechniqueAndMethod,
  }
}