From b9d28bd6af15026741099c25a6e72fe947ad3772 Mon Sep 17 00:00:00 2001 From: wujingjing <gersonwu@qq.com> Date: 星期四, 10 四月 2025 17:31:50 +0800 Subject: [PATCH] 修改数字人对话 --- src/components/chat/Chat.vue | 230 ++++++++++++++++++++++++++++++++++++++++++++++++++------- 1 files changed, 201 insertions(+), 29 deletions(-) diff --git a/src/components/chat/Chat.vue b/src/components/chat/Chat.vue index 9569345..8c7e10d 100644 --- a/src/components/chat/Chat.vue +++ b/src/components/chat/Chat.vue @@ -1,5 +1,11 @@ <template> - <ChatContainer :loading="chatListLoading" :more-is-loading="moreIsLoading" :is-share-page="isSharePage" ref="containerRef"> + <ChatContainer + :loading="chatListLoading" + :more-is-loading="moreIsLoading" + :is-share-page="isSharePage" + ref="containerRef" + @autoSendMessage="autoSendMessage" + > <!-- 娑堟伅鍒楄〃 --> <template #message-list> <MessageList @@ -48,22 +54,27 @@ <script setup lang="ts"> import type { CancelTokenSource } from 'axios'; import axios from 'axios'; +import { ElMessage } from 'element-plus'; import moment from 'moment'; +import { v4 as uuidv4 } from 'uuid'; import { computed, nextTick, onActivated, onMounted, ref } from 'vue'; import { loadAmisSource } from '../amis/load'; +import ChatContainer from './components/ChatContainer.vue'; +import { getKnowledgePlainText } from './components/playBar/hook/useDigitalHuman'; +import ShareLinkDlg from './components/shareLink/index.vue'; +import type { SendMsg } from './hooks/types'; +import { useLoadData } from './hooks/useLoadData'; import { useScrollLoad } from './hooks/useScrollLoad'; +import { useSyncMsg } from './hooks/useSyncMsg'; +import MessageList from './messageList/index.vue'; import type { ChatContent } from './model/types'; import { AnswerState, AnswerType, RoleEnum, type ChatMessage } from './model/types'; +import type { QuestionLifecycle } from './types'; import { getShareChatJsonByPost, questionStreamByPost } from '/@/api/ai/chat'; import PlayBar from '/@/components/chat/components/playBar/PlayBar.vue'; import CustomDrawer from '/@/components/drawer/CustomDrawer.vue'; import { Logger } from '/@/model/logger/Logger'; -import { triggerRef } from 'vue'; -import { ElMessage } from 'element-plus'; -import ChatContainer from './components/ChatContainer.vue'; -import ShareLinkDlg from './components/shareLink/index.vue'; import router from '/@/router'; -import MessageList from './messageList/index.vue'; import { activeChatRoom, activeGroupType, @@ -73,15 +84,15 @@ isSharePage, roomConfig, } from '/@/stores/chatRoom'; +import { ParentRegister } from '/@/stores/global'; import emitter from '/@/utils/mitt'; +import { deepClone } from '/@/utils/other'; import { useCompRef } from '/@/utils/types'; 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'; + const containerRef = useCompRef(ChatContainer); const chatListDom = computed(() => containerRef.value?.chatListDom); + const scrollToBottom = () => { containerRef.value?.scrollToBottom(); }; @@ -124,9 +135,12 @@ }); }; +const enableCallback = ref(false); + let streamOutputIsStart = false; let position: Position = null; -const questionAi = async (text) => { + +const questionAi = async (text: string, lifecycleCall?: QuestionLifecycle) => { let judgeParams = null; if (!preQuestion.value) { judgeParams = {}; @@ -206,6 +220,9 @@ if (chunkRes.mode === 'result') { lastIsResult = true; const res = chunkRes.value; + if (chunkRes.value?.answer_type === 'knowledge') { + lifecycleCall?.receiveText?.(getKnowledgePlainText(chunkRes.value)); + } if (checkReportEmpty()) { const resReport = getResReport(); resReport.reports.push(res); @@ -229,6 +246,19 @@ 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') { @@ -292,7 +322,6 @@ lastReport.conclusion = chunkRes.value; } chunkRes.value = '鍒嗘瀽缁撴潫'; - } const getLastGroup = () => { const lastGroup = computedMessageList.value.at(-1).stepGroup[0]; @@ -347,6 +376,7 @@ stepList.at(-1).ms = ms; isTalking.value = false; streamOutputIsStart = false; + lifecycleCall?.finish?.(); return; } @@ -414,12 +444,27 @@ let currentLLMId = null; -const stopGenClick = () => { +const resetTalking = () => { lastAxiosSource?.cancel(); isTalking.value = false; chatListLoading.value = false; streamOutputIsStart = false; +}; + +const stopGenClick = () => { + resetTalking(); + if (isFrontQuestion) { + ParentRegister.notify?.({ + type: 'msg_stop', + }); + } + computedMessageList.value.at(-1).isStopMsg = true; +}; + +const finishFrontQuestion = () => { + resetTalking(); + ParentRegister.updateChildCallObj('sendMsg', null); }; const checkCanSend = (content: ChatContent = messageContent.value) => { @@ -440,6 +485,7 @@ role: RoleEnum.assistant, content: { type: AnswerType.Report, + values: [], }, state: AnswerState.Null, stepGroup: [ @@ -467,10 +513,57 @@ playBarRef.value?.clearAttach(); }; -const sendChatMessage = async (content: ChatContent = messageContent.value) => { +const updateUserInfo = (userItem: ChatMessage, other: { historyId: string; question: string }) => { + userItem.historyId = other.historyId; + const current = moment().format('YYYY-MM-DD HH:mm:ss'); + userItem.createTime = current; + userItem.content.values = other.question ?? userItem.content.values; +}; + +const updateAssistantInfo = ( + assistantItem: ChatMessage, + resMsgContent: ChatContent, + other: { historyId: string; question: string } +) => { + const current = moment().format('YYYY-MM-DD HH:mm:ss'); + + assistantItem.historyId = other.historyId; + const currentTime = formatShowTimeYear(current); + assistantItem.createTime = currentTime; + assistantItem.content = resMsgContent; +}; + +const updateInfo = ( + userItem: ChatMessage, + assistantItem: ChatMessage, + resMsgContent: ChatContent, + other: { + historyId: string; + question: string; + } +) => { + updateUserInfo(userItem, other); + updateAssistantInfo(assistantItem, resMsgContent, other); +}; + +const handleAfterQuestion = ( + userItem: ChatMessage, + assistantItem: ChatMessage, + resMsgContent: ChatContent, + other: { historyId: string; question: string } +) => { + updateLoadIndex(); + updateInfo(userItem, assistantItem, resMsgContent, other); + setTimeout(() => { + // 鏀跺埌鍥炲锛岀户缁粴 + scrollToBottom(); + }, 300); +}; +const sendChatMessage = async (content: ChatContent = messageContent.value, lifecycleCall?: QuestionLifecycle) => { if (!checkCanSend(content)) { return; } + isFrontQuestion = false; const isNewChat = messageList.value.length === 0; if (isNewChat) { if (activeSampleId.value) { @@ -485,28 +578,107 @@ try { const [userItem, assistantItem] = addChatItem(content); - resMsgContent = await questionAi(content.values); - updateLoadIndex(); - - userItem.historyId = questionRes?.history_id; - const current = moment().format('YYYY-MM-DD HH:mm:ss'); - userItem.createTime = current; - userItem.content.values = questionRes?.question ?? userItem.content.values; - assistantItem.historyId = questionRes?.history_id; - const currentTime = formatShowTimeYear(current); - assistantItem.createTime = currentTime; - assistantItem.content = resMsgContent; - setTimeout(() => { - // 鏀跺埌鍥炲锛岀户缁粴 - scrollToBottom(); - }, 300); + resMsgContent = await questionAi(content.values, lifecycleCall); + handleAfterQuestion(userItem, assistantItem, resMsgContent, { + historyId: questionRes?.history_id, + question: questionRes?.question, + }); } catch (error: any) {} }; +let isFrontQuestion = false; +const sendFrontChatMessage = async (content: ChatContent = messageContent.value): Promise<any> => { + isFrontQuestion = true; + const [userItem, assistantItem] = addChatItem(content); + const promise = new Promise((resolve, reject) => { + const receiveMsg = (msg: SendMsg) => { + console.log('receiveMsg', msg); + if (!assistantItem.content?.values) { + assistantItem.content.values = []; + } + switch (msg.type) { + case 'text': + // 寮�濮嬪鍔犳柊鐨� stepGroup锛屽悗缁殑 stepGroup 骞舵病鏈夊疄闄呬綔鐢紝鍙槸涓轰簡鍋氳凯浠g敤锛岃凯浠e嚭缁勪欢锛屽睅灞变唬鐮佸疄鍦ㄥお闅炬敼浜嗭紒锛侊紒 + assistantItem.stepGroup.push({ + value: [], + isShow: true, + }); + + assistantItem.content.values.push({ + content: { + type: 'knowledge', + values: [ + { + answer: msg.value, + }, + ], + }, + }); + break; + + case 'select': + case 'confirm': + case 'input': + // 寮�濮嬪鍔犳柊鐨� stepGroup锛屽悗缁殑 stepGroup 骞舵病鏈夊疄闄呬綔鐢紝鍙槸涓轰簡鍋氳凯浠g敤锛岃凯浠e嚭缁勪欢锛屽睅灞变唬鐮佸疄鍦ㄥお闅炬敼浜嗭紒锛侊紒 + assistantItem.stepGroup.push({ + value: [], + isShow: true, + }); + assistantItem.content.values.push({ + content: { + type: 'content_cb', + values: msg, + }, + }); + break; + case 'info': + if (msg.value === 'finish') { + finishFrontQuestion(); + return resolve({ userItem, assistantItem, resMsgContent: assistantItem.content }); + } + break; + default: + break; + } + scrollToBottom(); + + triggerRefresh(); + }; + ParentRegister.updateChildCallObj('sendMsg', receiveMsg); + }); + return promise; +}; + +const questionSelf = (content: string) => { + const myContent = { type: AnswerType.Text, values: content }; + + if (!checkCanSend(myContent)) { + return false; + } + sendFrontChatMessage(myContent).then(({ userItem, assistantItem, resMsgContent }) => { + handleAfterQuestion(userItem, assistantItem, resMsgContent, { + historyId: uuidv4(), + question: content, + }); + }); + return true; +}; + +const backQuestion = (content: { question: string; data: any }) => { + sendChatMessage({ type: AnswerType.Text, values: content.question }); +}; + +ParentRegister.updateChildCallObj('frontQuestion', questionSelf); +ParentRegister.updateChildCallObj('backQuestion', backQuestion); const sendClick = () => { sendChatMessage(messageContent.value); }; +const autoSendMessage = (question: string, lifecycleCall?: QuestionLifecycle) => { + messageContent.value.values = question; + sendChatMessage(messageContent.value, lifecycleCall); +}; + const { loadRangeData, onChatListScroll, moreIsLoading, updateLoadIndex } = useScrollLoad({ container: chatListDom, historyGroupId: currentRouteId, -- Gitblit v1.9.3