diff --git a/frontend/src/components/TaskFlowQueryDialog.vue b/frontend/src/components/TaskFlowQueryDialog.vue
index c78c373..ba9d1fd 100644
--- a/frontend/src/components/TaskFlowQueryDialog.vue
+++ b/frontend/src/components/TaskFlowQueryDialog.vue
@@ -153,7 +153,7 @@
 
 <script setup lang="ts">
 import { ref, watch } from 'vue'
-import { ElMessage } from 'element-plus'
+import { message } from 'ant-design-vue'
 import { getTaskFlowStatus, type TaskFlowStatusVO } from '@/api/task'
 import Icon from '@/components/Icon.vue'
 
@@ -188,7 +188,7 @@ watch(dialogVisible, (val) => {
 // 处理搜索
 const handleSearch = async () => {
   if (!searchInput.value.trim()) {
-    ElMessage.warning('请输入任务编号或关键字')
+    message.warning('请输入任务编号或关键字')
     return
   }
 
@@ -210,7 +210,7 @@ const handleSearch = async () => {
       result.value = res.data
     }
   } catch (error: any) {
-    ElMessage.error(error.message || '查询失败')
+    message.error(error.message || '查询失败')
   } finally {
     loading.value = false
   }