perf: system model list

src/pages/api/system/getModels.ts

@@ -0,0 +1,23 @@
+import type { NextApiRequest, NextApiResponse } from 'next';
+import { jsonRes } from '@/service/response';
+import type { ChatModelItemType } from '@/constants/model';
+import { ChatModelMap, OpenAiChatEnum, ClaudeEnum } from '@/constants/model';
+
+// get the models available to the system
+export default async function handler(req: NextApiRequest, res: NextApiResponse) {
+  const chatModelList: ChatModelItemType[] = [];
+
+  if (process.env.OPENAIKEY) {
+    chatModelList.push(ChatModelMap[OpenAiChatEnum.GPT35]);
+  }
+  if (process.env.GPT4KEY) {
+    chatModelList.push(ChatModelMap[OpenAiChatEnum.GPT4]);
+  }
+  if (process.env.CLAUDE_KEY) {
+    chatModelList.push(ChatModelMap[ClaudeEnum.Claude]);
+  }
+
+  jsonRes(res, {
+    data: chatModelList
+  });
+}

src/pages/model/components/detail/components/ModelDataCard.tsx

@@ -138,7 +138,7 @@ const ModelDataCard = ({ modelId, isOwner }: { modelId: string; isOwner: boolean
   });
 
   return (
-    <>
+    <Box position={'relative'}>
       <Flex>
         <Box fontWeight={'bold'} fontSize={'lg'} flex={1} mr={2}>
           知识库数据: {total}组
@@ -303,7 +303,7 @@ const ModelDataCard = ({ modelId, isOwner }: { modelId: string; isOwner: boolean
       {isOpenSelectCsvModal && (
         <SelectCsvModal modelId={modelId} onClose={onCloseSelectCsvModal} onSuccess={refetchData} />
       )}
-    </>
+    </Box>
   );
 };
 

src/pages/model/components/detail/components/ModelEditForm.tsx

@@ -21,12 +21,13 @@ import {
 import { QuestionOutlineIcon } from '@chakra-ui/icons';
 import type { ModelSchema } from '@/types/mongoSchema';
 import { UseFormReturn } from 'react-hook-form';
-import { ChatModelMap, ModelVectorSearchModeMap, chatModelList } from '@/constants/model';
+import { ChatModelMap, ModelVectorSearchModeMap, getChatModelList } from '@/constants/model';
 import { formatPrice } from '@/utils/user';
 import { useConfirm } from '@/hooks/useConfirm';
 import { useSelectFile } from '@/hooks/useSelectFile';
 import { useToast } from '@/hooks/useToast';
 import { compressImg } from '@/utils/file';
+import { useQuery } from '@tanstack/react-query';
 
 const ModelEditForm = ({
   formHooks,
@@ -70,6 +71,8 @@ const ModelEditForm = ({
     [setValue, toast]
   );
 
+  const { data: chatModelList = [] } = useQuery(['init'], getChatModelList);
+
   return (
     <>
       <Card p={4}>
@@ -299,104 +302,6 @@ const ModelEditForm = ({
         </Card>
       )}
       <File onSelect={onSelectFile} />
-
-      {/* <Card p={4}>
-        <Box fontWeight={'bold'}>安全策略</Box>
-        <FormControl mt={2}>
-          <Flex alignItems={'center'}>
-            <Box flex={'0 0 120px'} w={0}>
-              单句最大长度:
-            </Box>
-            <Input
-              flex={1}
-              type={'number'}
-              {...register('security.contentMaxLen', {
-                required: '单句长度不能为空',
-                min: {
-                  value: 0,
-                  message: '单句长度最小为0'
-                },
-                max: {
-                  value: 4000,
-                  message: '单句长度最长为4000'
-                },
-                valueAsNumber: true
-              })}
-            ></Input>
-          </Flex>
-        </FormControl>
-        <FormControl mt={5}>
-          <Flex alignItems={'center'}>
-            <Box flex={'0 0 120px'} w={0}>
-              上下文最大长度:
-            </Box>
-            <Input
-              flex={1}
-              type={'number'}
-              {...register('security.contextMaxLen', {
-                required: '上下文长度不能为空',
-                min: {
-                  value: 1,
-                  message: '上下文长度最小为5'
-                },
-                max: {
-                  value: 400000,
-                  message: '上下文长度最长为 400000'
-                },
-                valueAsNumber: true
-              })}
-            ></Input>
-          </Flex>
-        </FormControl>
-        <FormControl mt={5}>
-          <Flex alignItems={'center'}>
-            <Box flex={'0 0 120px'} w={0}>
-              聊天过期时间:
-            </Box>
-            <Input
-              flex={1}
-              type={'number'}
-              {...register('security.expiredTime', {
-                required: '聊天过期时间不能为空',
-                min: {
-                  value: 0.1,
-                  message: '聊天过期时间最小为0.1小时'
-                },
-                max: {
-                  value: 999999,
-                  message: '聊天过期时间最长为 999999 小时'
-                },
-                valueAsNumber: true
-              })}
-            ></Input>
-            <Box ml={3}>小时</Box>
-          </Flex>
-        </FormControl>
-        <FormControl mt={5} pb={5}>
-          <Flex alignItems={'center'}>
-            <Box flex={'0 0 130px'} w={0}>
-              聊天最大加载次数:
-            </Box>
-            <Box flex={1}>
-              <Input
-                type={'number'}
-                {...register('security.maxLoadAmount', {
-                  required: '聊天最大加载次数不能为空',
-                  max: {
-                    value: 999999,
-                    message: '聊天最大加载次数最小为 999999 次'
-                  },
-                  valueAsNumber: true
-                })}
-              ></Input>
-              <Box fontSize={'sm'} color={'blackAlpha.400'} position={'absolute'}>
-                设置为-1代表不限制次数
-              </Box>
-            </Box>
-            <Box ml={3}>次</Box>
-          </Flex>
-        </FormControl>
-      </Card> */}
       <ConfirmChild />
     </>
   );

src/pages/model/index.tsx

@@ -34,7 +34,7 @@ const Model = ({ modelId, isPcDevice }: { modelId: string; isPcDevice: boolean }
           <ModelList modelId={modelId} />
         </Box>
       )}
-      <Box flex={1} h={'100%'}>
+      <Box flex={1} h={'100%'} position={'relative'}>
         {modelId && <ModelDetail modelId={modelId} isPc={isPc} />}
       </Box>
     </Flex>