fix: site isCustomElement config

Tencent · Jun 13, 2022 · f7e801f · f7e801f
1 parent d7d0390
commit f7e801f
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/site/vite.config.js b/site/vite.config.js
@@ -31,6 +31,8 @@ const testConfig = {
   },
 };
 
+const isCustomElement = (tag) => tag.startsWith('td-') || tag.startsWith('tdesign-theme');
+
 export default ({ mode }) => {
   return defineConfig({
     base: publicPathMap[mode],
@@ -60,12 +62,12 @@ export default ({ mode }) => {
         ssr: false,
         template: {
           compilerOptions: {
-            isCustomElement: (tag) => tag.startsWith('td-'),
+            isCustomElement,
           },
         },
       }),
       vueJsx({
-        isCustomElement: (tag) => tag.startsWith('td-'),
+        isCustomElement,
       }),
       tDocPlugin(),
       VitePWA(pwaConfig),