diff --git a/web/app/components/datasets/create-from-pipeline/create-options/create-from-scratch.tsx b/web/app/components/datasets/create-from-pipeline/create-options/create-from-scratch.tsx index f82b5c6370..c8828edf0c 100644 --- a/web/app/components/datasets/create-from-pipeline/create-options/create-from-scratch.tsx +++ b/web/app/components/datasets/create-from-pipeline/create-options/create-from-scratch.tsx @@ -17,8 +17,7 @@ import { useCreateDataset } from '@/service/knowledge/use-create-dataset' import type { Member } from '@/models/common' type CreateFromScratchProps = { - onClose?: () => void - onCreate?: () => void + onClose: () => void } const DEFAULT_APP_ICON: AppIconSelection = { @@ -29,7 +28,6 @@ const DEFAULT_APP_ICON: AppIconSelection = { const CreateFromScratch = ({ onClose, - onCreate, }: CreateFromScratchProps) => { const { t } = useTranslation() const [name, setName] = useState('') @@ -79,7 +77,7 @@ const CreateFromScratch = ({ const { mutateAsync: createEmptyDataset } = useCreateDataset() - const handleCreate = useCallback(() => { + const handleCreate = useCallback(async () => { if (!name) { Toast.notify({ type: 'error', @@ -108,10 +106,12 @@ const CreateFromScratch = ({ }) request.partial_member_list = selectedMemberList } - createEmptyDataset(request) - onCreate?.() - onClose?.() - }, [name, permission, appIcon, description, createEmptyDataset, memberList, selectedMemberIDs, onCreate, onClose]) + await createEmptyDataset(request, { + onSettled: () => { + onClose?.() + }, + }) + }, [name, permission, appIcon, description, createEmptyDataset, memberList, selectedMemberIDs, onClose]) return (
@@ -132,12 +132,12 @@ const CreateFromScratch = ({