fix: html_render supports chatUserProfile function (#4934)

shaohuzhang1 · web-flow · commit 2d6183454806 · 2026-03-24T15:37:12.000+08:00
diff --git a/ui/src/components/ai-chat/index.vue b/ui/src/components/ai-chat/index.vue
@@ -208,6 +208,14 @@ provide('getSelectModelList', (params: any) => {
     return loadSharedApi({ type: 'model', systemType: 'workspace' }).getSelectModelList(params)
   }
 })
+provide('chatUserProfile', () => {
+  if (props.type === 'ai-chat') {
+    if (chatUser.chat_profile?.authentication_type === 'login') {
+      return chatUser.getChatUserProfile()
+    }
+  }
+  return Promise.resolve(null)
+})
 
 const transcribing = ref<boolean>(false)
 defineOptions({ name: 'AiChat' })
@@ -798,16 +806,6 @@ const handleScroll = () => {
     }
   }
 }
-onBeforeMount(() => {
-  window.chatUserProfile = () => {
-    if (props.type === 'ai-chat') {
-      if (chatUser.chat_profile?.authentication_type === 'login') {
-        return chatUser.getChatUserProfile()
-      }
-    }
-    return Promise.resolve(null)
-  }
-})
 
 function parseTransform(transformStr: string) {
   const result = { scale: 1, translateX: 0, translateY: 0, translateZ: 0 }
diff --git a/ui/src/components/markdown/HtmlRander.vue b/ui/src/components/markdown/HtmlRander.vue
@@ -9,8 +9,8 @@
 </template>
 
 <script setup lang="ts">
-import { computed, ref, onMounted, onBeforeUnmount } from 'vue'
-
+import { computed, ref, onMounted, onBeforeUnmount, inject } from 'vue'
+const chatUserProfile = inject('chatUserProfile') as any
 const htmlRef = ref<HTMLIFrameElement>()
 const props = withDefaults(
   defineProps<{
@@ -37,6 +37,40 @@ function createIframeHtml(sourceHtml: string) {
 * { margin: 0; padding: 0; box-sizing: border-box; }
 html, body { margin: 0 !important; padding: 0 !important; overflow: hidden; }
 </style>
+<script>
+const _INSTANCE_ID = '${instanceId}';
+window.chatUserProfile=function chatUserProfile() {
+  return new Promise((resolve, reject) => {
+    const requestId = Date.now() + '_' + Math.random()
+
+    function handler(e) {
+      const data = e.data
+
+      if (
+        data?.type === 'chatUserProfile:response' &&
+        data.requestId === requestId
+      ) {
+        window.removeEventListener('message', handler)
+        resolve(data.data)
+      }
+    }
+    window.addEventListener('message', handler)
+    parent.postMessage(
+      {
+        type: 'chatUserProfile',
+        requestId,
+        instanceId: _INSTANCE_ID
+      },
+      '*'
+    )
+
+    setTimeout(() => {
+      window.removeEventListener('message', handler)
+      reject(new Error('timeout'))
+    }, 10000)
+  })
+}
+<\/script>
 </head>
 <body>
 ${sourceHtml}
@@ -46,7 +80,6 @@ const INSTANCE_ID = '${instanceId}';
 function sendMessage(message) {
   parent.postMessage({ type: 'chatMessage', instanceId: INSTANCE_ID, message }, '*');
 }
-
 let lastSentHeight = 0;
 let timer = null;
 
@@ -88,6 +121,20 @@ function onMessage(e: MessageEvent) {
   if (e.data.type === 'chatMessage') {
     props.sendMessage?.(e.data.message, 'new')
   }
+  if (e.data?.type === 'chatUserProfile') {
+    const iframe = htmlRef.value
+    if (!iframe) return
+    chatUserProfile().then((ok: any) => {
+      iframe.contentWindow?.postMessage(
+        {
+          type: 'chatUserProfile:response',
+          requestId: e.data.requestId,
+          data: ok,
+        },
+        '*',
+      )
+    })
+  }
 }
 
 onMounted(() => {
