AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-15 20:25:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix: add loading for dataset creation (#1214)

Browse Source
This commit is contained in:
KVOJJJin 2023-09-24 14:35:20 +08:00 committed by GitHub
parent 9df0dcedae
commit 841e967d48
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
web/app/components/datasets/create/step-two/index.tsx
View File

@ -107,6 +107,7 @@ const StepTwo = ({
const fileIndexingEstimate = (() => { const fileIndexingEstimate = (() => {
return segmentationType === SegmentType.AUTO ? automaticFileIndexingEstimate : customFileIndexingEstimate return segmentationType === SegmentType.AUTO ? automaticFileIndexingEstimate : customFileIndexingEstimate
})() })()
const [isCreating, setIsCreating] = useState(false)
const scrollHandle = (e: Event) => { const scrollHandle = (e: Event) => {
if ((e.target as HTMLDivElement).scrollTop > 0) if ((e.target as HTMLDivElement).scrollTop > 0)
@ -277,7 +278,7 @@ const StepTwo = ({
} as CreateDocumentReq } as CreateDocumentReq
if (dataSourceType === DataSourceType.FILE) { if (dataSourceType === DataSourceType.FILE) {
params.data_source.info_list.file_info_list = { params.data_source.info_list.file_info_list = {
file_ids: files.map(file => file.id), file_ids: files.map(file => file.id || '').filter(Boolean),
} }
} }
if (dataSourceType === DataSourceType.NOTION) if (dataSourceType === DataSourceType.NOTION)
@ -321,6 +322,7 @@ const StepTwo = ({
try { try {
let res let res
const params = getCreationParams() const params = getCreationParams()
setIsCreating(true)
if (!datasetId) { if (!datasetId) {
res = await createFirstDocument({ res = await createFirstDocument({
body: params, body: params,
@ -347,6 +349,9 @@ const StepTwo = ({
message: `${err}`, message: `${err}`,
}) })
} }
finally {
setIsCreating(false)
}
} }
const handleSwitch = (state: boolean) => { const handleSwitch = (state: boolean) => {
@ -622,7 +627,7 @@ const StepTwo = ({
<> <>
<div className='mb-2 text-xs font-medium text-gray-500'>{t('datasetCreation.stepTwo.fileSource')}</div> <div className='mb-2 text-xs font-medium text-gray-500'>{t('datasetCreation.stepTwo.fileSource')}</div>
<div className='flex items-center text-sm leading-6 font-medium text-gray-800'> <div className='flex items-center text-sm leading-6 font-medium text-gray-800'>
<span className={cn(s.fileIcon, files.length && s[files[0].extension])} /> <span className={cn(s.fileIcon, files.length && s[files[0].extension || ''])} />
{getFileName(files[0].name || '')} {getFileName(files[0].name || '')}
{files.length > 1 && ( {files.length > 1 && (
<span className={s.sourceCount}> <span className={s.sourceCount}>
@ -676,12 +681,12 @@ const StepTwo = ({
<div className='flex items-center mt-8 py-2'> <div className='flex items-center mt-8 py-2'>
<Button onClick={() => onStepChange && onStepChange(-1)}>{t('datasetCreation.stepTwo.lastStep')}</Button> <Button onClick={() => onStepChange && onStepChange(-1)}>{t('datasetCreation.stepTwo.lastStep')}</Button>
<div className={s.divider} /> <div className={s.divider} />
<Button type='primary' onClick={createHandle}>{t('datasetCreation.stepTwo.nextStep')}</Button> <Button loading={isCreating} type='primary' onClick={createHandle}>{t('datasetCreation.stepTwo.nextStep')}</Button>
</div> </div>
) )
: ( : (
<div className='flex items-center mt-8 py-2'> <div className='flex items-center mt-8 py-2'>
<Button type='primary' onClick={createHandle}>{t('datasetCreation.stepTwo.save')}</Button> <Button loading={isCreating} type='primary' onClick={createHandle}>{t('datasetCreation.stepTwo.save')}</Button>
<Button className='ml-2' onClick={onCancel}>{t('datasetCreation.stepTwo.cancel')}</Button> <Button className='ml-2' onClick={onCancel}>{t('datasetCreation.stepTwo.cancel')}</Button>
</div> </div>
)} )}