yt-chat-components 1.5.2 → 1.5.3

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "yt-chat-components",
-  "version": "1.5.2",
+  "version": "1.5.3",
   "main": "build/static/js/bundle.min.js",
   "module": "build/static/js/bundle.min.js",
   "types": "build/static/js/index.d.ts",

package/src/YtChatView/chatWidget/chatWindow/index.tsx

@@ -189,7 +189,7 @@ export default function ChatWindow({
     if (delayMessageList.length > 0) {
       const data = delayMessageList.shift();
       const {r_id, form_config} = data;
-      updateMessageItem({chunk:"\n```form\n" + JSON.stringify(form_config) + "\n```\n"})
+      updateMessageItem({updateSrc: "7", chunk:"\n```form\n" + JSON.stringify(form_config) + "\n```\n"})
 
       // ?????? done
       // addMessage({
@@ -206,8 +206,8 @@ export default function ChatWindow({
     }
   }
   
-  const addMessageItem = ({chunk, id, name, icon = null, status = null, isThinkChunk = false, loadingMessage = null}) => {
-    // console.log("--- addMessageItem", chunk, status, loadingMessage)
+  const addMessageItem = ({addSrc = null, chunk, id, name, icon = null, status = null, isThinkChunk = false, loadingMessage = null}) => {
+    // console.log("--- addMessageItem", addSrc, chunk, status, loadingMessage)
     setNowAIContentList((prevState) => {
       const content = {}
       if(isThinkChunk){
@@ -232,8 +232,8 @@ export default function ChatWindow({
     })
   }
 
-  const updateMessageItem = ({chunk, status = null, isThinkChunk = false, loadingMessage = null}) => {
-    // console.log("--- updateMessageItem", chunk, status, loadingMessage)
+  const updateMessageItem = ({updateSrc = null, chunk, status = null, isThinkChunk = false, loadingMessage = null}) => {
+    // console.log("--- updateMessageItem", updateSrc, chunk, status, loadingMessage)
     // chunk 和 status 都为空，则不更新
     if(isEmpty(chunk) && isEmpty(status)){
       return
@@ -286,12 +286,12 @@ export default function ChatWindow({
         // 如果刚才在调用工具，则追加信息
         if (content_id === 'WAIT'){
           content_id = id
-          updateMessageItem({chunk, loadingMessage: loading_message || ''})
+          updateMessageItem({updateSrc: "1", chunk, loadingMessage: loading_message || ''})
         }
         // 新建信息
         else{
           content_id = id
-          addMessageItem({chunk, id, name, icon, loadingMessage: loading_message || ''})
+          addMessageItem({addSrc: "1", chunk, id, name, icon, loadingMessage: loading_message || ''})
         }
       }
       // 输出主体没有变化
@@ -303,18 +303,18 @@ export default function ChatWindow({
         }
 
         // id切换表示一句话说完了， 新增信息
-        // if (content_id !== id){
-        //   content_id = id
-        //
-        //   // 如果 event_latest 是 token，此刻变成了 t_full_token，表示之前都在输出占位符，所以更新信息
-        //   if (event_latest === 'token' && event === 't_full_token'){
-        //     updateMessageItem({chunk, loadingMessage: loading_message || ''})
-        //   } else {
-        //     addMessageItem({chunk, id, name, icon, loadingMessage:loading_message})
-        //   }
-        // }else{
-          updateMessageItem({chunk, loadingMessage: loading_message || ''})
-        // }
+        if (content_id !== id){
+          content_id = id
+
+          // 如果 event_latest 是 token，此刻变成了 t_full_token，表示之前都在输出占位符，所以更新信息
+          if (event_latest === 'token' && event === 't_full_token'){
+            updateMessageItem({updateSrc: "2", chunk, loadingMessage: loading_message || ''})
+          } else {
+            addMessageItem({addSrc: "2", chunk, id, name, icon, loadingMessage:loading_message})
+          }
+        }else{
+          updateMessageItem({updateSrc: "3", chunk, loadingMessage: loading_message || ''})
+        }
       }
       
       if (lastMessage.current) {
@@ -322,16 +322,16 @@ export default function ChatWindow({
       }
     }
     else if (event == 't_token') {
-      let { chunk, id, r_id, ns, name, loading_message } = data
+      let { chunk, id, r_id, ns, name, icon, loading_message } = data
 
       // ns切换表示切换了智能体
       if (content_ns !== ns){
         content_ns = ns
         // 新建信息
         content_id = id
-        addMessageItem({chunk, id, name, icon, status:null, isThinkChunk:true, loadingMessage: loading_message || ''})
+        addMessageItem({addSrc: "3", chunk, id, name, icon, status:null, isThinkChunk:true, loadingMessage: loading_message || ''})
       }else {
-        updateMessageItem({chunk, status:null, isThinkChunk:true, loadingMessage: loading_message || ''})
+        updateMessageItem({updateSrc: "4", chunk, status:null, isThinkChunk:true, loadingMessage: loading_message || ''})
       }
 
       if (lastMessage.current) {
@@ -340,23 +340,23 @@ export default function ChatWindow({
     }
     else if (event == 'status') {
       // 更新状态
-      const {r_id, id, ns, name, status, loading_message} = data
+      const {r_id, id, ns, name, icon, status, loading_message} = data
 
       // ns变化表示切换了智能体，表示智能体一上来就调用工具
       if (content_ns !== ns){
         content_ns = ns
         // 表示智能体连续调用工具
         if (content_id === 'WAIT') {
-          updateMessageItem({chunk:"", status, loadingMessage: loading_message || ''})
+          updateMessageItem({updateSrc: "5", chunk:"", status, loadingMessage: loading_message || ''})
         }else{
           // 这个时候还没有信息的id，所以 content_id 给特殊值
           content_id = "WAIT"
-          addMessageItem({chunk:"", id: "WAIT", name, icon, status})
+          addMessageItem({addSrc: "4", chunk:"", id: "WAIT", name, icon, status})
         }
       }
       // 当前智能体在调用工具
       else{
-        updateMessageItem({chunk:"", status, loadingMessage: loading_message || ''})
+        updateMessageItem({updateSrc: "6", chunk:"", status, loadingMessage: loading_message || ''})
       }
 
       setAiStatus(data.status)