feat: 知识库匹配模式选择

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

@@ -12,12 +12,13 @@ import {
   SliderThumb,
   SliderMark,
   Tooltip,
-  Button
+  Button,
+  Select
 } from '@chakra-ui/react';
 import { QuestionOutlineIcon } from '@chakra-ui/icons';
 import type { ModelSchema } from '@/types/mongoSchema';
 import { UseFormReturn } from 'react-hook-form';
-import { modelList } from '@/constants/model';
+import { modelList, ModelVectorSearchModeMap } from '@/constants/model';
 import { formatPrice } from '@/utils/user';
 import { useConfirm } from '@/hooks/useConfirm';
 
@@ -89,15 +90,6 @@ const ModelEditForm = ({
             删除模型
           </Button>
         </Flex>
-        {/* <FormControl mt={4}>
-          <Box mb={1}>介绍:</Box>
-          <Textarea
-            rows={5}
-            maxLength={500}
-            {...register('intro')}
-            placeholder={'模型的介绍，仅做展示，不影响模型的效果'}
-          />
-        </FormControl> */}
       </Card>
       <Card p={4}>
         <Box fontWeight={'bold'}>模型效果</Box>
@@ -143,6 +135,20 @@ const ModelEditForm = ({
             </Slider>
           </Flex>
         </FormControl>
+        {canTrain && (
+          <FormControl mt={4}>
+            <Flex alignItems={'center'}>
+              <Box flex={'0 0 70px'}>搜索模式</Box>
+              <Select {...register('search.mode', { required: '搜索模式不能为空' })}>
+                {Object.entries(ModelVectorSearchModeMap).map(([key, { text }]) => (
+                  <option key={key} value={key}>
+                    {text}
+                  </option>
+                ))}
+              </Select>
+            </Flex>
+          </FormControl>
+        )}
         <Box mt={4}>
           <Box mb={1}>系统提示词</Box>
           <Textarea

src/pages/model/detail/index.tsx

@@ -143,6 +143,7 @@ const ModelDetail = ({ modelId }: { modelId: string }) => {
           systemPrompt: data.systemPrompt,
           intro: data.intro,
           temperature: data.temperature,
+          search: data.search,
           service: data.service,
           security: data.security
         });