From c4bdeb68785cfd9dc54f360b68d2da5b2f8f744d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E8=81=AA=E5=B0=8F=E9=99=88?=
 <xiaohuoni@users.noreply.github.com>
Date: Mon, 30 Dec 2024 17:06:42 +0800
Subject: [PATCH] fix: frameworkName when vite dev (#12873)

---
 .../preset-umi/src/commands/dev/plugins/ViteHtmlPlugin.ts     | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/packages/preset-umi/src/commands/dev/plugins/ViteHtmlPlugin.ts b/packages/preset-umi/src/commands/dev/plugins/ViteHtmlPlugin.ts
index 3561d51f0888..7f236f6ac9b1 100644
--- a/packages/preset-umi/src/commands/dev/plugins/ViteHtmlPlugin.ts
+++ b/packages/preset-umi/src/commands/dev/plugins/ViteHtmlPlugin.ts
@@ -21,7 +21,9 @@ export default function ViteHtmlPlugin(api: IApi): Plugin {
             try {
               // 处理通用html
               const viteScripts: IOpts['scripts'] = [
-                api.appData.hasSrcDir ? '/src/.umi/umi.ts' : '/.umi/umi.ts',
+                `${api.appData.hasSrcDir ? '/src' : ''}/.${
+                  api.service.frameworkName
+                }/umi.ts`,
               ];
 
               const markupArgs = (await getMarkupArgs({ api })) as any;