Explorar el Código

feat: enhance index type handling and add error notification for missing embedding model (#16836)

tags/1.2.0
Wu Tianwei hace 6 meses
padre
commit
cd7ac20d80
No account linked to committer's email address

+ 16
- 13
web/app/components/datasets/create/step-two/index.tsx Ver fichero

const [rules, setRules] = useState<PreProcessingRule[]>([]) const [rules, setRules] = useState<PreProcessingRule[]>([])
const [defaultConfig, setDefaultConfig] = useState<Rules>() const [defaultConfig, setDefaultConfig] = useState<Rules>()
const hasSetIndexType = !!indexingType const hasSetIndexType = !!indexingType
const [indexType, setIndexType] = useState<IndexingType>(
(indexingType
|| isAPIKeySet)
? IndexingType.QUALIFIED
: IndexingType.ECONOMICAL,
)
const [indexType, setIndexType] = useState<IndexingType>(() => {
if (hasSetIndexType)
return indexingType
return isAPIKeySet ? IndexingType.QUALIFIED : IndexingType.ECONOMICAL
})


const [previewFile, setPreviewFile] = useState<DocumentItem>( const [previewFile, setPreviewFile] = useState<DocumentItem>(
(datasetId && documentDetail) (datasetId && documentDetail)
} }
else { // create else { // create
const indexMethod = getIndexing_technique() const indexMethod = getIndexing_technique()
if (indexMethod === IndexingType.QUALIFIED && (!embeddingModel.model || !embeddingModel.provider)) {
Toast.notify({
type: 'error',
message: t('appDebug.datasetConfig.embeddingModelRequired'),
})
return
}
if ( if (
!isReRankModelSelected({ !isReRankModelSelected({
rerankModelList, rerankModelList,
// get indexing type by props // get indexing type by props
if (indexingType) if (indexingType)
setIndexType(indexingType as IndexingType) setIndexType(indexingType as IndexingType)

else else
setIndexType(isAPIKeySet ? IndexingType.QUALIFIED : IndexingType.ECONOMICAL) setIndexType(isAPIKeySet ? IndexingType.QUALIFIED : IndexingType.ECONOMICAL)
}, [isAPIKeySet, indexingType, datasetId]) }, [isAPIKeySet, indexingType, datasetId])
description={t('datasetCreation.stepTwo.qualifiedTip')} description={t('datasetCreation.stepTwo.qualifiedTip')}
icon={<Image src={indexMethodIcon.high_quality} alt='' />} icon={<Image src={indexMethodIcon.high_quality} alt='' />}
isActive={!hasSetIndexType && indexType === IndexingType.QUALIFIED} isActive={!hasSetIndexType && indexType === IndexingType.QUALIFIED}
disabled={!isAPIKeySet || hasSetIndexType}
disabled={hasSetIndexType}
onSwitched={() => { onSwitched={() => {
if (isAPIKeySet)
setIndexType(IndexingType.QUALIFIED)
setIndexType(IndexingType.QUALIFIED)
}} }}
/> />
)} )}
description={t('datasetCreation.stepTwo.economicalTip')} description={t('datasetCreation.stepTwo.economicalTip')}
icon={<Image src={indexMethodIcon.economical} alt='' />} icon={<Image src={indexMethodIcon.economical} alt='' />}
isActive={!hasSetIndexType && indexType === IndexingType.ECONOMICAL} isActive={!hasSetIndexType && indexType === IndexingType.ECONOMICAL}
disabled={!isAPIKeySet || hasSetIndexType || docForm !== ChunkingMode.text}
disabled={hasSetIndexType || docForm !== ChunkingMode.text}
ref={economyDomRef} ref={economyDomRef}
onSwitched={() => { onSwitched={() => {
if (isAPIKeySet && docForm === ChunkingMode.text)
setIndexType(IndexingType.ECONOMICAL)
setIndexType(IndexingType.ECONOMICAL)
}} }}
/> />
</PortalToFollowElemTrigger> </PortalToFollowElemTrigger>

+ 1
- 0
web/i18n/en-US/app-debug.ts Ver fichero

title: 'Multi-path retrieval', title: 'Multi-path retrieval',
description: 'Based on user intent, queries across all Knowledge, retrieves relevant text from multi-sources, and selects the best results matching the user query after reranking.', description: 'Based on user intent, queries across all Knowledge, retrieves relevant text from multi-sources, and selects the best results matching the user query after reranking.',
}, },
embeddingModelRequired: 'A configured Embedding Model is required',
rerankModelRequired: 'A configured Rerank Model is required', rerankModelRequired: 'A configured Rerank Model is required',
params: 'Params', params: 'Params',
top_k: 'Top K', top_k: 'Top K',

+ 1
- 0
web/i18n/zh-Hans/app-debug.ts Ver fichero

title: '多路召回', title: '多路召回',
description: '根据用户意图同时匹配所有知识库,从多路知识库查询相关文本片段,经过重排序步骤,从多路查询结果中选择匹配用户问题的最佳结果。', description: '根据用户意图同时匹配所有知识库,从多路知识库查询相关文本片段,经过重排序步骤,从多路查询结果中选择匹配用户问题的最佳结果。',
}, },
embeddingModelRequired: '未配置 Embedding 模型',
rerankModelRequired: '未配置 Rerank 模型', rerankModelRequired: '未配置 Rerank 模型',
params: '参数设置', params: '参数设置',
top_k: 'Top K', top_k: 'Top K',

Cargando…
Cancelar
Guardar