From f07405d8787e8bbd43453c81ec4f599dc34153b9 Mon Sep 17 00:00:00 2001 From: wujingjing <gersonwu@qq.com> Date: 星期三, 02 四月 2025 16:44:34 +0800 Subject: [PATCH] Merge branch 'master' into huishui_iframe --- src/components/chat/Chat.vue | 130 +++++++++++++++++++++++++++++++++--------- 1 files changed, 101 insertions(+), 29 deletions(-) diff --git a/src/components/chat/Chat.vue b/src/components/chat/Chat.vue index a1a2058..771c7ab 100644 --- a/src/components/chat/Chat.vue +++ b/src/components/chat/Chat.vue @@ -48,7 +48,6 @@ <script setup lang="ts"> import type { CancelTokenSource } from 'axios'; import axios from 'axios'; -import { orderBy } from 'lodash-es'; import moment from 'moment'; import { computed, nextTick, onActivated, onMounted, ref } from 'vue'; import { loadAmisSource } from '../amis/load'; @@ -60,7 +59,7 @@ import CustomDrawer from '/@/components/drawer/CustomDrawer.vue'; import { Logger } from '/@/model/logger/Logger'; import { triggerRef } from 'vue'; -import { ElLoadingService, ElMessage } from 'element-plus'; +import { ElMessage } from 'element-plus'; import ChatContainer from './components/ChatContainer.vue'; import ShareLinkDlg from './components/shareLink/index.vue'; import router from '/@/router'; @@ -76,16 +75,28 @@ } from '/@/stores/chatRoom'; import emitter from '/@/utils/mitt'; import { useCompRef } from '/@/utils/types'; -import { toMyFixed } from '/@/utils/util'; +import { toFormData, toMyFixed } from '/@/utils/util'; import { useLoadData } from './hooks/useLoadData'; import { useSyncMsg } from './hooks/useSyncMsg'; import { getCurrentPosition } from '/@/utils/brower'; +import { deepClone } from '/@/utils/other'; +import { ParentRegister } from '/@/stores/global'; const containerRef = useCompRef(ChatContainer); const chatListDom = computed(() => containerRef.value?.chatListDom); + const scrollToBottom = () => { containerRef.value?.scrollToBottom(); }; -const { loadReplyData, parseContent, parseExtraContent, convertProcessItem, convertProcessToStep, formatShowTimeYear } = useLoadData(); +const { + loadReplyData, + parseContent, + parseExtraContent, + convertProcessItem, + convertProcessToStep, + convertAttach, + formatShowTimeYear, + getStepGroupList, +} = useLoadData(); const voicePageIsShow = ref(false); let isTalking = ref(false); const chatWidth = computed(() => containerRef.value?.chatWidth); @@ -114,6 +125,10 @@ isTalking.value = !isTalking.value; }); }; + +const enableCallback = ref(false); + +let streamOutputIsStart = false; let position: Position = null; const questionAi = async (text) => { let judgeParams = null; @@ -131,7 +146,15 @@ raw_mode: roomConfig.value?.[currentRouteId]?.isAnswerByLLM ?? false, ...judgeParams, } as any; + const tableList = attachList.value.filter((item) => item.type === 'table').map((item) => item.model); + if (tableList?.length > 0) { + params.tables = JSON.stringify(tableList); + } + const metricList = attachList.value.filter((item) => item.type === 'metric').map((item) => item.model); + if (metricList?.length > 0) { + params.metrics = JSON.stringify(metricList); + } // if (!position) { // const loadingInstance = ElLoadingService({ // text: '鑾峰彇浣嶇疆涓�...', @@ -156,7 +179,12 @@ params.sample_id = currentSampleId; currentSampleId = ''; } - + const formDataParams = toFormData(params); + const fileList = attachList.value.filter((item) => item.type === 'file').map((item) => item.model); + for (const item of fileList) { + formDataParams.append('files', item.file); + } + // clearAttach(); let lastTimestamp = new Date().getTime(); questionRes = {}; let lastIsResult = false; @@ -176,10 +204,9 @@ return isEmpty; }; questionStreamByPost( - params, + formDataParams, (chunkRes) => { Logger.info('chunk response锛歕n\n' + JSON.stringify(chunkRes)); - if (chunkRes.mode === 'result') { lastIsResult = true; const res = chunkRes.value; @@ -206,6 +233,26 @@ triggerRefresh(); return; // chunkRes.value = '鍑嗗鏁版嵁鍒嗘瀽'; + } + + if (chunkRes.mode === 'main_frame') { + const jsonObj = JSON.parse(chunkRes.value); + if (!enableCallback.value) { + return; + } + + ParentRegister.notify?.({ + type: 'main_frame', + value: jsonObj, + }); + return; + } + + if (chunkRes.mode === 'create_work_order') { + const lastMsg = computedMessageList.value.at(-1); + lastMsg.modeContent = chunkRes; + triggerRefresh(); + return; } if (chunkRes.mode === 'summary') { @@ -257,10 +304,12 @@ if (chunkRes.mode === 'conclusion') { const lastReport = computedMessageList.value.at(-1)?.content?.values?.at(-1); + if (lastReport) { lastReport.conclusion = chunkRes.value; - chunkRes.value = '鍒嗘瀽缁撴潫'; } + chunkRes.value = '鍒嗘瀽缁撴潫'; + } const getLastGroup = () => { const lastGroup = computedMessageList.value.at(-1).stepGroup[0]; @@ -307,29 +356,44 @@ }); lastIsResult = false; } - const lastGroup = computedMessageList.value.at(-1).stepGroup[0]; + const lastGroup = getLastGroup(); const stepList = lastGroup?.value ?? []; const currentTimeStamp = new Date().getTime(); - const ms = toMyFixed(currentTimeStamp - lastTimestamp, 2) + ' ms'; if (chunkRes.mode === 'finish') { + const ms = toMyFixed(currentTimeStamp - lastTimestamp, 2) + ' ms'; stepList.at(-1).ms = ms; isTalking.value = false; + streamOutputIsStart = false; return; } - if (stepList?.length >= 1) { + if (stepList?.length >= 1 && !streamOutputIsStart) { + const ms = toMyFixed(currentTimeStamp - lastTimestamp, 2) + ' ms'; + stepList.at(-1).ms = ms; + } + + if (!streamOutputIsStart) { + lastTimestamp = currentTimeStamp; + } + if (!streamOutputIsStart) { + const stepItem = convertProcessItem(chunkRes); + stepList.push(stepItem); } else { - const stepGroup = computedMessageList.value.at(-1).stepGroup; - if (stepGroup.length > 1) { - const lastStepList = stepGroup.at(-2).value; - lastStepList.at(-1).ms = ms; + const lastItem = stepList.at(-1); + if (lastItem) { + lastItem.title += chunkRes.value ?? ''; } } - lastTimestamp = currentTimeStamp; - const stepItem = convertProcessItem(chunkRes); - stepList.push(stepItem); + if (chunkRes.mode === 'begin_stream') { + streamOutputIsStart = true; + lastTimestamp = currentTimeStamp; + } + if (chunkRes.mode === 'end_stream') { + streamOutputIsStart = false; + } + // 寮哄埗瑙﹀彂鏇存柊 scrollToBottom(); @@ -354,12 +418,14 @@ const content = parseContent(questionRes, true); return content; }; - -const clearMessageContent = () => - (messageContent.value = { +const playBarRef = useCompRef(PlayBar); +const attachList = computed(() => playBarRef.value?.attachList ?? []); +const clearMessageContent = () => { + messageContent.value = { type: AnswerType.Text, values: '', - }); + }; +}; let currentSampleId = ''; @@ -369,7 +435,7 @@ lastAxiosSource?.cancel(); isTalking.value = false; chatListLoading.value = false; - + streamOutputIsStart = false; computedMessageList.value.at(-1).isStopMsg = true; }; @@ -386,7 +452,7 @@ const addChatItem = (content: ChatContent) => { isTalking.value = true; - const userItem: ChatMessage = { role: RoleEnum.user, content, isChecked: false } as any; + const userItem: ChatMessage = { role: RoleEnum.user, content, isChecked: false, attachList: deepClone(attachList.value) } as any; const assistantItem: ChatMessage = { role: RoleEnum.assistant, content: { @@ -399,6 +465,7 @@ isShow: true, }, ], + isStopMsg: false, isChecked: false, } as any; @@ -408,6 +475,13 @@ messageList.value.push(assistantItem); scrollToBottom(); return [userItem, assistantItem]; +}; + +/** + * 娓呴櫎闄勪欢 + */ +const clearAttach = () => { + playBarRef.value?.clearAttach(); }; const sendChatMessage = async (content: ChatContent = messageContent.value) => { @@ -512,6 +586,7 @@ type: AnswerType.Text, values: msgValue.question, }, + attachList: convertAttach(msgValue), isChecked: false, }; @@ -519,10 +594,8 @@ historyId: msgValue.history_id, role: RoleEnum.assistant, content: parseContent(msgValue), - stepGroup: (msgValue?.reports ?? []).map((item) => ({ - value: convertProcessToStep(item?.exec_process), - isShow: false, - })), + stepGroup: getStepGroupList(msgValue?.reports ?? []), + isStopMsg: false, conclusion: msgValue.conclusion ?? [], @@ -571,7 +644,6 @@ messageContent.value.values = content; }; //#endregion -const playBarRef = useCompRef(PlayBar); //鐢ㄦ埛闂璁剧疆涓哄父鐢ㄨ const setCommonPhraseClick = (item) => { playBarRef.value.addPhrase(item); -- Gitblit v1.9.3