From 85e6a0f38d8494f05cf7e5069bba40976d017d48 Mon Sep 17 00:00:00 2001
From: archer <545436317@qq.com>
Date: Wed, 14 Jun 2023 10:01:00 +0800
Subject: [PATCH] fix: token limit

---
 client/src/pages/api/openapi/kb/pushData.ts |  4 ++--
 client/src/pages/kb/components/DataCard.tsx | 14 +++++++++++++-
 2 files changed, 15 insertions(+), 3 deletions(-)

diff --git a/client/src/pages/api/openapi/kb/pushData.ts b/client/src/pages/api/openapi/kb/pushData.ts
index 6a913941a..39b10598f 100644
--- a/client/src/pages/api/openapi/kb/pushData.ts
+++ b/client/src/pages/api/openapi/kb/pushData.ts
@@ -24,8 +24,8 @@ export type Response = {
 };
 
 const modeMaxToken = {
-  [TrainingModeEnum.index]: 700,
-  [TrainingModeEnum.qa]: 3300
+  [TrainingModeEnum.index]: 6000,
+  [TrainingModeEnum.qa]: 10000
 };
 
 export default withNextCors(async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
diff --git a/client/src/pages/kb/components/DataCard.tsx b/client/src/pages/kb/components/DataCard.tsx
index 2d98b17c9..d4bf9e404 100644
--- a/client/src/pages/kb/components/DataCard.tsx
+++ b/client/src/pages/kb/components/DataCard.tsx
@@ -21,7 +21,7 @@ import {
   delOneKbDataByDataId,
   getTrainingData
 } from '@/api/plugins/kb';
-import { DeleteIcon } from '@chakra-ui/icons';
+import { DeleteIcon, RepeatIcon } from '@chakra-ui/icons';
 import { fileDownload } from '@/utils/file';
 import { useMutation, useQuery } from '@tanstack/react-query';
 import { useToast } from '@/hooks/useToast';
@@ -146,6 +146,18 @@ const DataCard = ({ kbId }: { kbId: string }) => {
           知识库数据: {total}组
         </Box>
         <Box>
+          <IconButton
+            icon={<RepeatIcon />}
+            aria-label={'refresh'}
+            variant={'base'}
+            isLoading={isLoading}
+            mr={[2, 4]}
+            size={'sm'}
+            onClick={() => {
+              refetchData(pageNum);
+              getTrainingData({ kbId, init: true });
+            }}
+          />
           <Button
             variant={'base'}
             mr={2}