fix: dataset select check (#3048)

This commit is contained in:
Archer
2024-10-31 17:25:02 +08:00
committed by shilin66
parent b11e1cf4b2
commit 9668fbba63

View File

@@ -55,6 +55,9 @@ export const DatasetSelectModal = ({
) )
}; };
}, [datasets, allDatasets, selectedDatasets]); }, [datasets, allDatasets, selectedDatasets]);
const activeVectorModel = allDatasets.find(
(dataset) => dataset._id === selectedDatasets[0]?.datasetId
)?.vectorModel?.model;
return ( return (
<DatasetSelectContainer <DatasetSelectContainer
@@ -141,11 +144,7 @@ export const DatasetSelectModal = ({
if (item.type === DatasetTypeEnum.folder) { if (item.type === DatasetTypeEnum.folder) {
setParentId(item._id); setParentId(item._id);
} else { } else {
const vectorModel = datasets.find( if (activeVectorModel && activeVectorModel !== item.vectorModel.model) {
(dataset) => dataset._id === selectedDatasets[0]?.datasetId
)?.vectorModel?.model;
if (vectorModel && vectorModel !== item.vectorModel.model) {
return toast({ return toast({
status: 'warning', status: 'warning',
title: t('common:dataset.Select Dataset Tips') title: t('common:dataset.Select Dataset Tips')
@@ -168,13 +167,20 @@ export const DatasetSelectModal = ({
{item.name} {item.name}
</Box> </Box>
</Flex> </Flex>
<Flex justifyContent={'flex-end'} alignItems={'center'} fontSize={'sm'}> <Flex
justifyContent={'flex-end'}
alignItems={'center'}
fontSize={'sm'}
color={
activeVectorModel === item.vectorModel.name ? 'primary.600' : 'myGray.500'
}
>
{item.type === DatasetTypeEnum.folder ? ( {item.type === DatasetTypeEnum.folder ? (
<Box color={'myGray.500'}>{t('common:Folder')}</Box> <Box color={'myGray.500'}>{t('common:Folder')}</Box>
) : ( ) : (
<> <>
<MyIcon mr={1} name="kbTest" w={'12px'} /> <MyIcon mr={1} name="kbTest" w={'12px'} />
<Box color={'myGray.500'}>{item.vectorModel.name}</Box> <Box>{item.vectorModel.name}</Box>
</> </>
)} )}
</Flex> </Flex>