From f15262ef12a04a280c4fd029a037fff4ca54441d Mon Sep 17 00:00:00 2001
From: shenjunjian <shenjunjian5@huawei.com>
Date: Mon, 9 Jun 2025 16:18:06 +0800
Subject: [PATCH] =?UTF-8?q?fix(robot):=20=E4=BC=98=E5=8C=96robot=E7=9A=84?=
 =?UTF-8?q?=E6=B6=88=E6=81=AF=E5=A4=84=E7=90=86?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../views/components-doc/composition/utils.ts | 90 ++++++-------------
 1 file changed, 29 insertions(+), 61 deletions(-)

diff --git a/examples/sites/src/views/components-doc/composition/utils.ts b/examples/sites/src/views/components-doc/composition/utils.ts
index 0844caf87c..4ed5968931 100644
--- a/examples/sites/src/views/components-doc/composition/utils.ts
+++ b/examples/sites/src/views/components-doc/composition/utils.ts
@@ -41,6 +41,25 @@ export async function handleSSEStream(
     )
   }
 
+  let messageIndex = 0
+  function printMessage(data, str: string, endln = false) {
+    handler.onData({
+      id: '',
+      created: data.created_at,
+      choices: [
+        {
+          index: messageIndex++,
+          delta: {
+            role: 'assistant',
+            content: str + (endln ? '\n\n' : '')
+          },
+          finish_reason: null
+        }
+      ],
+      object: '',
+      model: ''
+    })
+  }
   try {
     while (true) {
       if (signal?.aborted) {
@@ -59,8 +78,6 @@ export async function handleSSEStream(
       const lines = buffer.split('\n\n')
       buffer = lines.pop() || ''
 
-      let messageIndex = 0
-
       for (const line of lines) {
         if (line.trim() === '') continue
         if (line.trim() === 'data: [DONE]') {
@@ -75,68 +92,19 @@ export async function handleSSEStream(
 
           const data = JSON.parse(dataMatch[1])
           // console.log('SSE data:', data)
-          if (data?.event === 'workflow_started') {
-            handler.onData({
-              id: data.workflow_run_id,
-              created: data.created_at,
-              choices: [
-                {
-                  index: messageIndex++,
-                  delta: {
-                    role: 'assistant',
-                    content: ' '
-                  },
-                  finish_reason: null
-                }
-              ],
-              object: '',
-              model: ''
-            })
-            message.value.slice(-1)[0].loading = true
-            if (!globalConversation.id) {
-              globalConversation.id = data.conversation_id
-            }
+          if (data?.event === 'node_started') {
+            printMessage(data, `${data.data.title} 节点运行...`, true)
           }
-
-          if (data?.event === 'message' && data?.answer) {
-            handler.onData({
-              id: data.id,
-              created: data.created_at,
-              choices: [
-                {
-                  index: messageIndex++,
-                  delta: {
-                    role: 'assistant',
-                    content: data.answer
-                  },
-                  finish_reason: null
-                }
-              ],
-              object: '',
-              model: ''
-            })
-            message.value.slice(-1)[0].loading = false
+          if (data?.event === 'node_finished') {
+            printMessage(
+              data,
+              `${data.data.title} 节点结束\n\n` +
+                (data.data.node_type === 'answer' ? `${data.data.outputs.answer}` : '')
+            )
           }
-          if (data?.event === 'message_end') {
-            handler.onData({
-              id: data.id,
-              created: data.created_at,
-              choices: [
-                {
-                  index: messageIndex++,
-                  delta: {
-                    role: 'assistant',
-                    content: ''
-                  },
-                  finish_reason: 'stop'
-                }
-              ],
-              object: '',
-              model: ''
-            })
-            message.value.slice(-1)[0].loading = false
+          if (data?.event === 'agent_log' && data.data.status === 'success' && data.data.label.startsWith('CALL')) {
+            printMessage(data, `--${data.data.label}(${JSON.stringify(data.data.data.output.tool_call_input)})`, true)
           }
-          // handler.onData(data)
         } catch (error) {
           console.error('Error parsing SSE message:', error)
         }