Merge branch 'feat/parent-child-retrieval' of https://github.com/langgenius/dify into feat/parent-child-retrieval
twwu
1 year ago
commit
db01c5c89d
@ -0,0 +1,30 @@
|
|||||||
|
'use client'
|
||||||
|
import type { FC } from 'react'
|
||||||
|
import React from 'react'
|
||||||
|
import { SliceContent } from '../../formatted-text/flavours/shared'
|
||||||
|
import Score from './score'
|
||||||
|
import type { HitTestingChildChunk } from '@/models/datasets'
|
||||||
|
|
||||||
|
type Props = {
|
||||||
|
payload: HitTestingChildChunk
|
||||||
|
isShowAll: boolean
|
||||||
|
}
|
||||||
|
|
||||||
|
const ChildChunks: FC<Props> = ({
|
||||||
|
payload,
|
||||||
|
isShowAll,
|
||||||
|
}) => {
|
||||||
|
const { id, score, content } = payload
|
||||||
|
return (
|
||||||
|
<div
|
||||||
|
className={!isShowAll ? 'line-clamp-2' : ''}
|
||||||
|
>
|
||||||
|
<div className='inline-flex items-center relative top-[-3px]'>
|
||||||
|
<div className='flex items-center h-[24px] bg-state-accent-solid system-2xs-semibold-uppercase text-text-primary-on-surface px-1'>C-{id}</div>
|
||||||
|
<Score value={score} besideChunkName />
|
||||||
|
</div>
|
||||||
|
<SliceContent className='bg-state-accent-hover group-hover:bg-state-accent-hover text-text-secondary font-normal'>{content}</SliceContent>
|
||||||
|
</div>
|
||||||
|
)
|
||||||
|
}
|
||||||
|
export default React.memo(ChildChunks)
|
||||||
@ -0,0 +1,89 @@
|
|||||||
|
'use client'
|
||||||
|
import type { FC } from 'react'
|
||||||
|
import React from 'react'
|
||||||
|
import { useTranslation } from 'react-i18next'
|
||||||
|
import { SegmentIndexTag } from '../../documents/detail/completed'
|
||||||
|
import Dot from '../../documents/detail/completed/common/dot'
|
||||||
|
import Score from './score'
|
||||||
|
import ChildChunksItem from './child-chunks-item'
|
||||||
|
import Modal from '@/app/components/base/modal'
|
||||||
|
import type { HitTesting } from '@/models/datasets'
|
||||||
|
import FileIcon from '@/app/components/base/file-uploader/file-type-icon'
|
||||||
|
import type { FileAppearanceTypeEnum } from '@/app/components/base/file-uploader/types'
|
||||||
|
import cn from '@/utils/classnames'
|
||||||
|
import Tag from '@/app/components/datasets/documents/detail/completed/common/tag'
|
||||||
|
|
||||||
|
const i18nPrefix = 'datasetHitTesting'
|
||||||
|
|
||||||
|
type Props = {
|
||||||
|
payload: HitTesting
|
||||||
|
onHide: () => void
|
||||||
|
}
|
||||||
|
|
||||||
|
const ChunkDetailModal: FC<Props> = ({
|
||||||
|
payload,
|
||||||
|
onHide,
|
||||||
|
}) => {
|
||||||
|
const { t } = useTranslation()
|
||||||
|
const { segment, score, child_chunks } = payload
|
||||||
|
const { position, content, keywords, document } = segment
|
||||||
|
const isParentChildRetrieval = !!(child_chunks && child_chunks.length > 0)
|
||||||
|
const extension = document.name.split('.').slice(-1)[0] as FileAppearanceTypeEnum
|
||||||
|
const maxHeighClassName = 'max-h-[min(752px,_80vh)] overflow-y-auto'
|
||||||
|
return (
|
||||||
|
<Modal
|
||||||
|
title={t(`${i18nPrefix}.chunkDetail`)}
|
||||||
|
isShow
|
||||||
|
closable
|
||||||
|
onClose={onHide}
|
||||||
|
className={cn(isParentChildRetrieval ? '!min-w-[1200px]' : '!min-w-[720px]')}
|
||||||
|
>
|
||||||
|
<div className='mt-4 flex pb-6'>
|
||||||
|
<div>
|
||||||
|
{/* Meta info */}
|
||||||
|
<div className='flex justify-between items-center'>
|
||||||
|
<div className='grow flex items-center space-x-2'>
|
||||||
|
<SegmentIndexTag
|
||||||
|
isParentChildRetrieval={isParentChildRetrieval}
|
||||||
|
positionId={position}
|
||||||
|
className={cn('w-fit group-hover:opacity-100')}
|
||||||
|
/>
|
||||||
|
<Dot />
|
||||||
|
<div className='grow flex items-center space-x-1'>
|
||||||
|
<FileIcon type={extension} size='sm' />
|
||||||
|
<span className='grow w-0 truncate text-text-secondary text-[13px] font-normal'>{document.name}</span>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
<Score value={score} />
|
||||||
|
</div>
|
||||||
|
<div className={cn('mt-2 body-md-regular text-text-secondary', maxHeighClassName)}>
|
||||||
|
{content}
|
||||||
|
</div>
|
||||||
|
{!isParentChildRetrieval && keywords && keywords.length > 0 && (
|
||||||
|
<div className='mt-6'>
|
||||||
|
<div className='font-medium text-xs text-text-tertiary uppercase'>{t(`${i18nPrefix}.keyword`)}</div>
|
||||||
|
<div className='mt-1 flex flex-wrap'>
|
||||||
|
{keywords.map(keyword => (
|
||||||
|
<Tag key={keyword} text={keyword} className='mr-2' />
|
||||||
|
))}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
|
||||||
|
{isParentChildRetrieval && (
|
||||||
|
<div className='shrink-0 w-[552px] px-6'>
|
||||||
|
<div className='system-xs-semibold-uppercase text-text-secondary'>{t(`${i18nPrefix}.hitChunks`, { num: child_chunks.length })}</div>
|
||||||
|
<div className={cn('mt-1 space-y-2', maxHeighClassName)}>
|
||||||
|
{child_chunks.map(item => (
|
||||||
|
<ChildChunksItem key={item.id} payload={item} isShowAll />
|
||||||
|
))}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
</Modal>
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
export default React.memo(ChunkDetailModal)
|
||||||
@ -0,0 +1,25 @@
|
|||||||
|
'use client'
|
||||||
|
import type { FC } from 'react'
|
||||||
|
import React from 'react'
|
||||||
|
import cn from '@/utils/classnames'
|
||||||
|
|
||||||
|
type Props = {
|
||||||
|
value: number
|
||||||
|
besideChunkName?: boolean
|
||||||
|
}
|
||||||
|
|
||||||
|
const Score: FC<Props> = ({
|
||||||
|
value,
|
||||||
|
besideChunkName,
|
||||||
|
}) => {
|
||||||
|
return (
|
||||||
|
<div className={cn('relative items-center px-[5px] border border-components-progress-bar-border overflow-hidden', besideChunkName ? 'border-l-0 h-[25px]' : 'h-[20px] rounded-md')}>
|
||||||
|
<div className={cn('absolute top-0 left-0 h-full bg-util-colors-blue-brand-blue-brand-100 border-r-[1.5px] border-components-progress-brand-progress', value === 1 && 'border-r-0')} style={{ width: `${value * 100}%` }} />
|
||||||
|
<div className={cn('relative flex items-center h-full space-x-0.5 text-util-colors-blue-brand-blue-brand-700')}>
|
||||||
|
<div className='system-2xs-medium-uppercase'>score</div>
|
||||||
|
<div className='system-xs-semibold'>{value.toFixed(2)}</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
)
|
||||||
|
}
|
||||||
|
export default React.memo(Score)
|
||||||
Loading…
Reference in New Issue