diff --git a/react-ui/src/pages/Dataset/components/AddDatasetModal/index.tsx b/react-ui/src/pages/Dataset/components/AddDatasetModal/index.tsx
index 177bf2a0..b55aa08c 100644
--- a/react-ui/src/pages/Dataset/components/AddDatasetModal/index.tsx
+++ b/react-ui/src/pages/Dataset/components/AddDatasetModal/index.tsx
@@ -7,7 +7,6 @@ import { getDictSelectOption } from '@/services/system/dict';
 import { to } from '@/utils/promise';
 import { getFileListFromEvent, validateUploadFiles } from '@/utils/ui';
 import {
-  App,
   Button,
   Form,
   Input,
@@ -15,6 +14,7 @@ import {
   Select,
   Upload,
   UploadFile,
+  message,
   type ModalProps,
   type UploadProps,
 } from 'antd';
@@ -32,7 +32,6 @@ interface AddDatasetModalProps extends Omit<ModalProps, 'onOk'> {
 function AddDatasetModal({ typeList, tagList, onOk, ...rest }: AddDatasetModalProps) {
   const [uuid] = useState(Date.now());
   const [clusterOptions, setClusterOptions] = useState<DictValueEnumObj[]>([]);
-  const { message } = App.useApp();
 
   useEffect(() => {
     getClusterOptions();
diff --git a/react-ui/src/pages/Dataset/components/AddModelModal/index.tsx b/react-ui/src/pages/Dataset/components/AddModelModal/index.tsx
index f4d68c53..f132fb05 100644
--- a/react-ui/src/pages/Dataset/components/AddModelModal/index.tsx
+++ b/react-ui/src/pages/Dataset/components/AddModelModal/index.tsx
@@ -6,13 +6,13 @@ import { addModel } from '@/services/dataset/index.js';
 import { to } from '@/utils/promise';
 import { getFileListFromEvent, validateUploadFiles } from '@/utils/ui';
 import {
-  App,
   Button,
   Form,
   Input,
   Select,
   Upload,
   UploadFile,
+  message,
   type ModalProps,
   type UploadProps,
 } from 'antd';
@@ -28,7 +28,6 @@ interface AddModelModalProps extends Omit<ModalProps, 'onOk'> {
 
 function AddModelModal({ typeList, tagList, onOk, ...rest }: AddModelModalProps) {
   const [uuid] = useState(Date.now());
-  const { message } = App.useApp();
 
   // 上传组件参数
   const uploadProps: UploadProps = {
diff --git a/react-ui/src/pages/Dataset/components/AddVersionModal/index.tsx b/react-ui/src/pages/Dataset/components/AddVersionModal/index.tsx
index c00569f7..839c8e20 100644
--- a/react-ui/src/pages/Dataset/components/AddVersionModal/index.tsx
+++ b/react-ui/src/pages/Dataset/components/AddVersionModal/index.tsx
@@ -5,12 +5,12 @@ import { ResourceType, resourceConfig } from '@/pages/Dataset/types';
 import { to } from '@/utils/promise';
 import { getFileListFromEvent, validateUploadFiles } from '@/utils/ui';
 import {
-  App,
   Button,
   Form,
   Input,
   Upload,
   UploadFile,
+  message,
   type ModalProps,
   type UploadProps,
 } from 'antd';
@@ -33,7 +33,6 @@ function AddVersionModal({
   ...rest
 }: AddVersionModalProps) {
   const [uuid] = useState(Date.now());
-  const { message } = App.useApp();
 
   // 上传组件参数
   const uploadProps: UploadProps = {