Merge remote-tracking branch 'origin/v2' into v2

Author: liqiang-fit2cloud

ui/src/components/ai-chat/index.vue

@@ -195,6 +195,7 @@ import bus from '@/bus'
 import { throttle } from 'lodash-es'
 import { copyClick } from '@/utils/clipboard'
 import { loadSharedApi } from '@/utils/dynamics-api/shared-api'
+import { getWrite } from '@/utils/chat'
 
 provide('upload', (file: any, loading?: Ref<boolean>) => {
   return props.type === 'debug-ai-chat'
@@ -560,86 +561,7 @@ function getChartOpenId(chat?: any, problem?: string, re_chat?: boolean, other_p
   })
 }
 
-/**
- * 获取一个递归函数,处理流式数据
- * @param chat    每一条对话记录
- * @param reader  流数据
- * @param stream  是否是流式数据
- */
-const getWrite = (chat: any, reader: any, stream: boolean) => {
-  let tempResult = ''
-
-  const write_stream = async () => {
-    try {
-      while (true) {
-        const { done, value } = await reader.read()
-
-        if (done) {
-          ChatManagement.close(chat.id)
-          return
-        }
-
-        const decoder = new TextDecoder('utf-8')
-        let str = decoder.decode(value, { stream: true })
-
-        tempResult += str
-        const split = tempResult.match(/data:.*?}\n\n/g)
-        if (split) {
-          str = split.join('')
-          tempResult = tempResult.replace(str, '')
-
-          // 批量处理所有 chunk
-          for (const item of split) {
-            const chunk = JSON.parse(item.replace('data:', ''))
-            chat.chat_id = chunk.chat_id
-            chat.record_id = chunk.chat_record_id
-
-            if (!chunk.is_end) {
-              ChatManagement.appendChunk(chat.id, chunk)
-            }
-
-            if (chunk.is_end) {
-              return Promise.resolve()
-            }
-          }
-        }
-        // 如果没有匹配到完整chunk，继续读取下一块
-      }
-    } catch (e) {
-      return Promise.reject(e)
-    }
-  }
-
-  const write_json = async () => {
-    try {
-      while (true) {
-        const { done, value } = await reader.read()
-
-        if (done) {
-          const result_block = JSON.parse(tempResult)
-          if (result_block.code === 500) {
-            return Promise.reject(result_block.message)
-          } else {
-            if (result_block.content) {
-              ChatManagement.append(chat.id, result_block.content)
-            }
-          }
-          ChatManagement.close(chat.id)
-          return
-        }
 
-        if (value) {
-          const decoder = new TextDecoder('utf-8')
-          tempResult += decoder.decode(value)
-        }
-      }
-    } catch (e) {
-      return Promise.reject(e)
-    }
-  }
-
-  return stream ? write_stream : write_json
-}
 const errorWrite = (chat: any, message?: string) => {
   ChatManagement.addChatRecord(chat, 50, loading)
   ChatManagement.write(chat.id)

ui/src/locales/lang/en-US/views/tool.ts

@@ -10,6 +10,7 @@ export default {
   toolWorkflow: {
     creatToolWorkflow: 'Create Workflow',
     toActiveTip: 'Unable to enable. Please publish the workflow first.',
+    debugResult: 'Debug results',
   },
   dataSource: {
     title: 'Data Source',

ui/src/locales/lang/zh-CN/views/tool.ts

@@ -9,6 +9,7 @@ export default {
   toolWorkflow: {
     creatToolWorkflow: '创建工作流',
     toActiveTip: '无法启用，请先发布工作流。',
+    debugResult:'调试结果'
   },
   dataSource: {
     title: '数据源',

ui/src/locales/lang/zh-Hant/views/tool.ts

@@ -9,6 +9,7 @@ export default {
   toolWorkflow: {
     creatToolWorkflow: '創建工作流',
     toActiveTip: '無法啓用，請先發布工作流。',
+    debugResult: '調試結果',
   },
   dataSource: {
     title: '數據源',

ui/src/styles/app.scss

@@ -255,6 +255,10 @@ h5 {
   padding-bottom: 0;
 }
 
+.pr-8 {
+  padding-right: 8px;
+}
+
 .float-right {
   float: right;
 }

ui/src/utils/chat.ts

@@ -0,0 +1,81 @@
+import { ChatManagement, type chatType } from '@/api/type/application'
+/**
+ * 获取一个递归函数,处理流式数据
+ * @param chat    每一条对话记录
+ * @param reader  流数据
+ * @param stream  是否是流式数据
+ */
+export const getWrite = (chat: any, reader: any, stream: boolean) => {
+  let tempResult = ''
+
+  const write_stream = async () => {
+    try {
+      while (true) {
+        const { done, value } = await reader.read()
+
+        if (done) {
+          ChatManagement.close(chat.id)
+          return
+        }
+
+        const decoder = new TextDecoder('utf-8')
+        let str = decoder.decode(value, { stream: true })
+
+        tempResult += str
+        const split = tempResult.match(/data:.*?}\n\n/g)
+        if (split) {
+          str = split.join('')
+          tempResult = tempResult.replace(str, '')
+
+          // 批量处理所有 chunk
+          for (const item of split) {
+            const chunk = JSON.parse(item.replace('data:', ''))
+            chat.chat_id = chunk.chat_id
+            chat.record_id = chunk.chat_record_id
+
+            if (!chunk.is_end) {
+              ChatManagement.appendChunk(chat.id, chunk)
+            }
+
+            if (chunk.is_end) {
+              return Promise.resolve()
+            }
+          }
+        }
+        // 如果没有匹配到完整chunk，继续读取下一块
+      }
+    } catch (e) {
+      return Promise.reject(e)
+    }
+  }
+
+  const write_json = async () => {
+    try {
+      while (true) {
+        const { done, value } = await reader.read()
+
+        if (done) {
+          const result_block = JSON.parse(tempResult)
+          if (result_block.code === 500) {
+            return Promise.reject(result_block.message)
+          } else {
+            if (result_block.content) {
+              ChatManagement.append(chat.id, result_block.content)
+            }
+          }
+          ChatManagement.close(chat.id)
+          return
+        }
+
+        if (value) {
+          const decoder = new TextDecoder('utf-8')
+          tempResult += decoder.decode(value)
+        }
+      }
+    } catch (e) {
+      return Promise.reject(e)
+    }
+  }
+
+  return stream ? write_stream : write_json
+}