v4.6-4 (#473)

projects/app/src/pages/dataset/list/component/CreateModal.tsx

@@ -15,10 +15,12 @@ import { postCreateDataset } from '@/web/core/dataset/api';
 import type { CreateDatasetParams } from '@/global/core/dataset/api.d';
 import MySelect from '@/components/Select';
 import { QuestionOutlineIcon } from '@chakra-ui/icons';
-import { vectorModelList } from '@/web/common/system/staticData';
+import { vectorModelList, qaModelList } from '@/web/common/system/staticData';
 import Tag from '@/components/Tag';
+import { useTranslation } from 'next-i18next';
 
 const CreateModal = ({ onClose, parentId }: { onClose: () => void; parentId?: string }) => {
+  const { t } = useTranslation();
   const [refresh, setRefresh] = useState(false);
   const { toast } = useToast();
   const router = useRouter();
@@ -29,6 +31,7 @@ const CreateModal = ({ onClose, parentId }: { onClose: () => void; parentId?: st
       name: '',
       tags: '',
       vectorModel: vectorModelList[0].model,
+      agentModel: qaModelList[0].model,
       type: 'dataset',
       parentId
     }
@@ -76,7 +79,7 @@ const CreateModal = ({ onClose, parentId }: { onClose: () => void; parentId?: st
   });
 
   return (
-    <MyModal isOpen onClose={onClose} isCentered={!isPc} w={'400px'}>
+    <MyModal isOpen onClose={onClose} isCentered={!isPc} w={'450px'}>
       <ModalHeader fontSize={'2xl'}>创建一个知识库</ModalHeader>
       <ModalBody>
         <Box color={'myGray.800'} fontWeight={'bold'}>
@@ -106,7 +109,7 @@ const CreateModal = ({ onClose, parentId }: { onClose: () => void; parentId?: st
           />
         </Flex>
         <Flex mt={6} alignItems={'center'}>
-          <Box flex={'0 0 80px'}>索引模型</Box>
+          <Box flex={'0 0 100px'}>索引模型</Box>
           <Box flex={1}>
             <MySelect
               w={'100%'}
@@ -122,8 +125,25 @@ const CreateModal = ({ onClose, parentId }: { onClose: () => void; parentId?: st
             />
           </Box>
         </Flex>
+        <Flex mt={6} alignItems={'center'}>
+          <Box flex={'0 0 100px'}>{t('dataset.Agent Model')}</Box>
+          <Box flex={1}>
+            <MySelect
+              w={'100%'}
+              value={getValues('agentModel')}
+              list={qaModelList.map((item) => ({
+                label: item.name,
+                value: item.model
+              }))}
+              onchange={(e) => {
+                setValue('agentModel', e);
+                setRefresh((state) => !state);
+              }}
+            />
+          </Box>
+        </Flex>
         <Flex mt={6} alignItems={'center'} w={'100%'}>
-          <Box flex={'0 0 80px'}>
+          <Box flex={'0 0 100px'}>
             标签
             <MyTooltip label={'用空格隔开多个标签，便于搜索'} forceShow>
               <QuestionOutlineIcon ml={1} />

projects/app/src/pages/dataset/list/index.tsx

@@ -20,7 +20,8 @@ import {
   delDatasetById,
   getDatasetPaths,
   putDatasetById,
-  postCreateDataset
+  postCreateDataset,
+  getCheckExportLimit
 } from '@/web/core/dataset/api';
 import { useTranslation } from 'next-i18next';
 import Avatar from '@/components/Avatar';
@@ -38,6 +39,7 @@ import { useDrag } from '@/web/common/hooks/useDrag';
 import { useUserStore } from '@/web/support/user/useUserStore';
 import PermissionIconText from '@/components/support/permission/IconText';
 import { PermissionTypeEnum } from '@fastgpt/global/support/permission/constant';
+import { DatasetItemType } from '@fastgpt/global/core/dataset/type';
 
 const CreateModal = dynamic(() => import('./component/CreateModal'), { ssr: false });
 const MoveModal = dynamic(() => import('./component/MoveModal'), { ssr: false });
@@ -89,6 +91,23 @@ const Kb = () => {
     successToast: t('common.Delete Success'),
     errorToast: t('dataset.Delete Dataset Error')
   });
+  // check export limit
+  const { mutate: exportDataset } = useRequest({
+    mutationFn: async (dataset: DatasetItemType) => {
+      setLoading(true);
+      await getCheckExportLimit(dataset._id);
+      const a = document.createElement('a');
+      a.href = `/api/core/dataset/exportAll?datasetId=${dataset._id}`;
+      a.download = `${dataset.name}.csv`;
+      document.body.appendChild(a);
+      a.click();
+      document.body.removeChild(a);
+    },
+    onSettled() {
+      setLoading(false);
+    },
+    errorToast: t('dataset.Export Dataset Limit Error')
+  });
 
   const { data, refetch } = useQuery(['loadDataset', parentId], () => {
     return Promise.all([loadDatasets(parentId), getDatasetPaths(parentId)]);
@@ -371,12 +390,7 @@ const Kb = () => {
                       </Flex>
                     ),
                     onClick: () => {
-                      const a = document.createElement('a');
-                      a.href = `/api/core/dataset/exportAll?datasetId=${dataset._id}`;
-                      a.download = `${dataset.name}.csv`;
-                      document.body.appendChild(a);
-                      a.click();
-                      document.body.removeChild(a);
+                      exportDataset(dataset);
                     }
                   },
                   {