Spaces:
Paused
Paused
| import EntityTypesItem from '@/components/entity-types-item'; | |
| import LayoutRecognize from '@/components/layout-recognize'; | |
| import MaxTokenNumber from '@/components/max-token-number'; | |
| import ParseConfiguration, { | |
| showRaptorParseConfiguration, | |
| } from '@/components/parse-configuration'; | |
| import { useTranslate } from '@/hooks/common-hooks'; | |
| import { normFile } from '@/utils/file-util'; | |
| import { PlusOutlined } from '@ant-design/icons'; | |
| import { Button, Form, Input, Radio, Select, Space, Upload } from 'antd'; | |
| import { FormInstance } from 'antd/lib'; | |
| import { | |
| useFetchKnowledgeConfigurationOnMount, | |
| useSubmitKnowledgeConfiguration, | |
| } from './hooks'; | |
| import styles from './index.less'; | |
| const { Option } = Select; | |
| const ConfigurationForm = ({ form }: { form: FormInstance }) => { | |
| const { submitKnowledgeConfiguration, submitLoading, navigateToDataset } = | |
| useSubmitKnowledgeConfiguration(form); | |
| const { parserList, embeddingModelOptions, disabled } = | |
| useFetchKnowledgeConfigurationOnMount(form); | |
| const { t } = useTranslate('knowledgeConfiguration'); | |
| return ( | |
| <Form form={form} name="validateOnly" layout="vertical" autoComplete="off"> | |
| <Form.Item name="name" label={t('name')} rules={[{ required: true }]}> | |
| <Input /> | |
| </Form.Item> | |
| <Form.Item | |
| name="avatar" | |
| label={t('photo')} | |
| valuePropName="fileList" | |
| getValueFromEvent={normFile} | |
| > | |
| <Upload | |
| listType="picture-card" | |
| maxCount={1} | |
| beforeUpload={() => false} | |
| showUploadList={{ showPreviewIcon: false, showRemoveIcon: false }} | |
| > | |
| <button style={{ border: 0, background: 'none' }} type="button"> | |
| <PlusOutlined /> | |
| <div style={{ marginTop: 8 }}>{t('upload')}</div> | |
| </button> | |
| </Upload> | |
| </Form.Item> | |
| <Form.Item name="description" label={t('description')}> | |
| <Input /> | |
| </Form.Item> | |
| <Form.Item | |
| label={t('language')} | |
| name="language" | |
| initialValue={'English'} | |
| rules={[{ required: true, message: t('languageMessage') }]} | |
| > | |
| <Select placeholder={t('languagePlaceholder')}> | |
| <Option value="English">{t('english')}</Option> | |
| <Option value="Chinese">{t('chinese')}</Option> | |
| </Select> | |
| </Form.Item> | |
| <Form.Item | |
| name="permission" | |
| label={t('permissions')} | |
| tooltip={t('permissionsTip')} | |
| rules={[{ required: true }]} | |
| > | |
| <Radio.Group> | |
| <Radio value="me">{t('me')}</Radio> | |
| <Radio value="team">{t('team')}</Radio> | |
| </Radio.Group> | |
| </Form.Item> | |
| <Form.Item | |
| name="embd_id" | |
| label={t('embeddingModel')} | |
| rules={[{ required: true }]} | |
| tooltip={t('embeddingModelTip')} | |
| > | |
| <Select | |
| placeholder={t('embeddingModelPlaceholder')} | |
| options={embeddingModelOptions} | |
| disabled={disabled} | |
| ></Select> | |
| </Form.Item> | |
| <Form.Item | |
| name="parser_id" | |
| label={t('chunkMethod')} | |
| tooltip={t('chunkMethodTip')} | |
| rules={[{ required: true }]} | |
| > | |
| <Select placeholder={t('chunkMethodPlaceholder')} disabled={disabled}> | |
| {parserList.map((x) => ( | |
| <Option value={x.value} key={x.value}> | |
| {x.label} | |
| </Option> | |
| ))} | |
| </Select> | |
| </Form.Item> | |
| <Form.Item noStyle dependencies={['parser_id']}> | |
| {({ getFieldValue }) => { | |
| const parserId = getFieldValue('parser_id'); | |
| return ( | |
| <> | |
| {parserId === 'knowledge_graph' && ( | |
| <EntityTypesItem></EntityTypesItem> | |
| )} | |
| {parserId === 'naive' && ( | |
| <> | |
| <MaxTokenNumber></MaxTokenNumber> | |
| <LayoutRecognize></LayoutRecognize> | |
| </> | |
| )} | |
| {showRaptorParseConfiguration(parserId) && ( | |
| <ParseConfiguration></ParseConfiguration> | |
| )} | |
| </> | |
| ); | |
| }} | |
| </Form.Item> | |
| <Form.Item> | |
| <div className={styles.buttonWrapper}> | |
| <Space> | |
| <Button size={'middle'} onClick={navigateToDataset}> | |
| {t('cancel')} | |
| </Button> | |
| <Button | |
| type="primary" | |
| size={'middle'} | |
| loading={submitLoading} | |
| onClick={submitKnowledgeConfiguration} | |
| > | |
| {t('save')} | |
| </Button> | |
| </Space> | |
| </div> | |
| </Form.Item> | |
| </Form> | |
| ); | |
| }; | |
| export default ConfigurationForm; | |