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) { 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}组 + } + aria-label={'refresh'} + variant={'base'} + isLoading={isLoading} + mr={[2, 4]} + size={'sm'} + onClick={() => { + refetchData(pageNum); + getTrainingData({ kbId, init: true }); + }} + />