fix: dataset select check (#3048)

labring · Oct 31, 2024 · 4061b11 · 4061b11
1 parent bc171db
commit 4061b11
Showing 1 changed file with 13 additions and 7 deletions.
diff --git a/projects/app/src/components/core/app/DatasetSelectModal.tsx b/projects/app/src/components/core/app/DatasetSelectModal.tsx
@@ -55,6 +55,9 @@ export const DatasetSelectModal = ({
       )
     };
   }, [datasets, allDatasets, selectedDatasets]);
+  const activeVectorModel = allDatasets.find(
+    (dataset) => dataset._id === selectedDatasets[0]?.datasetId
+  )?.vectorModel?.model;
 
   return (
     <DatasetSelectContainer
@@ -141,11 +144,7 @@ export const DatasetSelectModal = ({
                         if (item.type === DatasetTypeEnum.folder) {
                           setParentId(item._id);
                         } else {
-                          const vectorModel = datasets.find(
-                            (dataset) => dataset._id === selectedDatasets[0]?.datasetId
-                          )?.vectorModel?.model;
-
-                          if (vectorModel && vectorModel !== item.vectorModel.model) {
+                          if (activeVectorModel && activeVectorModel !== item.vectorModel.model) {
                             return toast({
                               status: 'warning',
                               title: t('common:dataset.Select Dataset Tips')
@@ -168,13 +167,20 @@ export const DatasetSelectModal = ({
                           {item.name}
                         </Box>
                       </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 ? (
                           <Box color={'myGray.500'}>{t('common:Folder')}</Box>
                         ) : (
                           <>
                             <MyIcon mr={1} name="kbTest" w={'12px'} />
-                            <Box color={'myGray.500'}>{item.vectorModel.name}</Box>
+                            <Box>{item.vectorModel.name}</Box>
                           </>
                         )}
                       </Flex>