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