diff --git a/build/plugins.ts b/build/plugins.ts
index 3e93ad2..3da638f 100644
--- a/build/plugins.ts
+++ b/build/plugins.ts
@@ -29,7 +29,22 @@ export function createVitePlugins(env: ImportMetaEnv) {
 
     // auto import api of lib
     AutoImport({
-      imports: ['vue', 'vue-router', 'pinia', '@vueuse/core', 'vue-i18n'],
+      imports: [
+        'vue',
+        'vue-router',
+        'pinia',
+        '@vueuse/core',
+        'vue-i18n',
+        {
+          'naive-ui': [
+            'useDialog',
+            'useMessage',
+            'useNotification',
+            'useLoadingBar',
+            'useModal',
+          ],
+        },
+      ],
       include: [
         /\.[tj]sx?$/,
         /\.vue$/,