Skip to content

Commit 4061b11

Browse files
authored
fix: dataset select check (#3048)
1 parent bc171db commit 4061b11

File tree

1 file changed

+13
-7
lines changed

1 file changed

+13
-7
lines changed

projects/app/src/components/core/app/DatasetSelectModal.tsx

Lines changed: 13 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -55,6 +55,9 @@ export const DatasetSelectModal = ({
5555
)
5656
};
5757
}, [datasets, allDatasets, selectedDatasets]);
58+
const activeVectorModel = allDatasets.find(
59+
(dataset) => dataset._id === selectedDatasets[0]?.datasetId
60+
)?.vectorModel?.model;
5861

5962
return (
6063
<DatasetSelectContainer
@@ -141,11 +144,7 @@ export const DatasetSelectModal = ({
141144
if (item.type === DatasetTypeEnum.folder) {
142145
setParentId(item._id);
143146
} else {
144-
const vectorModel = datasets.find(
145-
(dataset) => dataset._id === selectedDatasets[0]?.datasetId
146-
)?.vectorModel?.model;
147-
148-
if (vectorModel && vectorModel !== item.vectorModel.model) {
147+
if (activeVectorModel && activeVectorModel !== item.vectorModel.model) {
149148
return toast({
150149
status: 'warning',
151150
title: t('common:dataset.Select Dataset Tips')
@@ -168,13 +167,20 @@ export const DatasetSelectModal = ({
168167
{item.name}
169168
</Box>
170169
</Flex>
171-
<Flex justifyContent={'flex-end'} alignItems={'center'} fontSize={'sm'}>
170+
<Flex
171+
justifyContent={'flex-end'}
172+
alignItems={'center'}
173+
fontSize={'sm'}
174+
color={
175+
activeVectorModel === item.vectorModel.name ? 'primary.600' : 'myGray.500'
176+
}
177+
>
172178
{item.type === DatasetTypeEnum.folder ? (
173179
<Box color={'myGray.500'}>{t('common:Folder')}</Box>
174180
) : (
175181
<>
176182
<MyIcon mr={1} name="kbTest" w={'12px'} />
177-
<Box color={'myGray.500'}>{item.vectorModel.name}</Box>
183+
<Box>{item.vectorModel.name}</Box>
178184
</>
179185
)}
180186
</Flex>

0 commit comments

Comments
 (0)