doc gpt V0.2

2023-02-19 14:35:25 +08:00
parent cc5cf99e7a
commit 0ecf576e4e
124 changed files with 11780 additions and 573 deletions
--- a/src/pages/model/components/CreateModel.tsx
+++ b/src/pages/model/components/CreateModel.tsx
@@ -0,0 +1,126 @@
+import React, { Dispatch, useState, useCallback } from 'react';
+import {
+  Modal,
+  ModalOverlay,
+  ModalContent,
+  ModalHeader,
+  ModalFooter,
+  ModalBody,
+  ModalCloseButton,
+  FormControl,
+  FormErrorMessage,
+  Button,
+  useToast,
+  Input,
+  Select
+} from '@chakra-ui/react';
+import { useForm } from 'react-hook-form';
+import { postCreateModel } from '@/api/model';
+import { ModelType } from '@/types/model';
+import { OpenAiList } from '@/constants/model';
+
+interface CreateFormType {
+  name: string;
+  serviceModelName: string;
+}
+
+const CreateModel = ({
+  isOpen,
+  setCreateModelOpen,
+  onSuccess
+}: {
+  isOpen: boolean;
+  setCreateModelOpen: Dispatch<boolean>;
+  onSuccess: Dispatch<ModelType>;
+}) => {
+  const [requesting, setRequesting] = useState(false);
+  const toast = useToast({
+    duration: 2000,
+    position: 'top'
+  });
+  const {
+    register,
+    handleSubmit,
+    formState: { errors }
+  } = useForm<CreateFormType>({
+    defaultValues: {
+      serviceModelName: OpenAiList[0].model
+    }
+  });
+
+  const handleCreateModel = useCallback(
+    async (data: CreateFormType) => {
+      setRequesting(true);
+      try {
+        const res = await postCreateModel(data);
+        toast({
+          title: '创建成功',
+          status: 'success'
+        });
+        onSuccess(res);
+        setCreateModelOpen(false);
+      } catch (err: any) {
+        toast({
+          title: typeof err === 'string' ? err : err.message || '出现了意外',
+          status: 'error'
+        });
+      }
+      setRequesting(false);
+    },
+    [onSuccess, setCreateModelOpen, toast]
+  );
+
+  return (
+    <>
+      <Modal isOpen={isOpen} onClose={() => setCreateModelOpen(false)}>
+        <ModalOverlay />
+        <ModalContent>
+          <ModalHeader>创建模型</ModalHeader>
+          <ModalCloseButton />
+
+          <ModalBody>
+            <FormControl mb={8} isInvalid={!!errors.name}>
+              <Input
+                placeholder="模型名称"
+                {...register('name', {
+                  required: '模型名不能为空'
+                })}
+              />
+              <FormErrorMessage position={'absolute'} fontSize="xs">
+                {!!errors.name && errors.name.message}
+              </FormErrorMessage>
+            </FormControl>
+            <FormControl isInvalid={!!errors.serviceModelName}>
+              <Select
+                placeholder="选择基础模型类型"
+                {...register('serviceModelName', {
+                  required: '底层模型不能为空'
+                })}
+              >
+                {OpenAiList.map((item) => (
+                  <option key={item.model} value={item.model}>
+                    {item.name}
+                  </option>
+                ))}
+              </Select>
+              <FormErrorMessage position={'absolute'} fontSize="xs">
+                {!!errors.serviceModelName && errors.serviceModelName.message}
+              </FormErrorMessage>
+            </FormControl>
+          </ModalBody>
+
+          <ModalFooter>
+            <Button mr={3} colorScheme={'gray'} onClick={() => setCreateModelOpen(false)}>
+              取消
+            </Button>
+            <Button isLoading={requesting} onClick={handleSubmit(handleCreateModel)}>
+              确认创建
+            </Button>
+          </ModalFooter>
+        </ModalContent>
+      </Modal>
+    </>
+  );
+};
+
+export default CreateModel;