From acaf8207447eb243e1dc8bec3aaca447cc4d082f Mon Sep 17 00:00:00 2001
From: wujingjing <gersonwu@qq.com>
Date: 星期三, 17 七月 2024 16:32:58 +0800
Subject: [PATCH] error msg

---
 src/components/chat/Chat.vue |  381 ++++++++++++++++++++++++++++++++++++++++++------------
 1 files changed, 295 insertions(+), 86 deletions(-)

diff --git a/src/components/chat/Chat.vue b/src/components/chat/Chat.vue
index ec0c7c9..897cbdf 100644
--- a/src/components/chat/Chat.vue
+++ b/src/components/chat/Chat.vue
@@ -1,94 +1,100 @@
 <template>
 	<div class="flex flex-col h-full">
 		<div class="h-full flex flex-col items-center overflow-y-auto">
-			<div ref="chatListDom" class="h-full w-[100ch]">
+			<div ref="chatListDom" class="h-full" :style="{ width: chatWidth }">
 				<div
-					class="group flex px-4 py-4 hover:bg-slate-100 rounded-lg"
-					v-for="(item, index) of messageList.filter((v) => v.role !== 'system')"
+					class="group flex px-4 py-6 hover:bg-slate-100 rounded-lg"
+					:class="{ 'flex-row-reverse': item.role === RoleEnum.user }"
+					v-for="(item, index) of computedMessageList"
 					:key="index"
 				>
-					<img class="rounded-full size-12 mr-4" :src="roleImageMap[item.role]" alt="" srcset="" />
+					<img
+						class="rounded-full size-12 mr-4"
+						:class="{ 'mr-4': item.role === RoleEnum.assistant, 'ml-4': item.role === RoleEnum.user }"
+						:src="roleImageMap[item.role]"
+						alt=""
+						srcset=""
+					/>
 
-					<div class="flex">
-						<div class="relative" v-if="item.content?.values">
+					<div class="flex" :class="{ 'w-full': item.role === RoleEnum.assistant }">
+						<div class="relative w-full" v-if="item.content?.values">
 							<div
-								class="text-sm rounded-[6px] p-4 leading-relaxed max-w-[100ch] bg-white"
+								class="text-sm rounded-[6px] p-4 leading-relaxed"
+								:style="{ backgroundColor: item.role === RoleEnum.user ? 'rgb(197 224 255)' : 'white' }"
 							>
-								<component class="max-w-[100ch]" :is="answerTypeMapCom[item.content.type]" :data="item.content.values" />
+								<component :is="answerTypeMapCom[item.content.type]" :data="item.content.values" :error="item.content.error" />
 							</div>
 
-							<!-- <div v-if="item.role === RoleEnum.assistant" class="absolute flex items-center right-0 space-x-2 mr-2 mt-2">
-								<SvgIcon class="cursor-pointer" name="ele-CopyDocument" @click="copyClick(item.content)" />
-								<SvgIcon class="cursor-pointer" name="ywicon icon-dianzan" />
-								<SvgIcon class="cursor-pointer" :size="12" name="ywicon icon-buzan" />
-							</div> -->
+							<div v-if="item.role === RoleEnum.assistant" class="absolute flex items-center right-0 mr-2 mt-2 space-x-2">
+								<div
+									class="flex items-center justify-center size-[15px]"
+									v-if="item.content?.type === AnswerType.Text || item.content?.type === AnswerType.Knowledge"
+								>
+									<i class="p-2 ywicon icon-copy cursor-pointer hover:text-[#0284ff] hover:!text-[18px]" @click="copyClick(item)" />
+								</div>
+								<div class="flex items-center justify-center size-[15px]">
+									<i
+										:class="{ 'text-[#0284ff]': item.state === AnswerState.Like }"
+										class="p-2 ywicon icon-dianzan cursor-pointer hover:text-[#0284ff] font-medium hover:!text-[18px]"
+										@click="likeClick(item)"
+									/>
+								</div>
+								<div class="flex items-center justify-center size-[15px]">
+									<i
+										:class="{ 'text-[#0284ff]': item.state === AnswerState.Unlike }"
+										class="p-2 ywicon icon-buzan cursor-pointer hover:text-[#0284ff] !text-[13px] hover:!text-[15px]"
+										@click="unLikeClick(item)"
+									/>
+								</div>
+							</div>
 						</div>
 
-						<Loding v-else />
+						<Loding v-else :process="process" />
 					</div>
 				</div>
 			</div>
 		</div>
 
-		<div class="sticky bottom-0 w-full p-6 pb-8 bg-gray-100 flex justify-center">
-			<PlayBar :isTalking="isTalking" v-model="messageContent.values" @sendClick="sendChatMessage" />
+		<div class="sticky bottom-0 w-full p-6 pb-8 bg-[rgb(247,248,250)] flex justify-center">
+			<PlayBar
+				:isTalking="isTalking"
+				v-model="messageContent.values"
+				@sendClick="sendChatMessage"
+				:style="{ width: chatWidth }"
+			></PlayBar>
 		</div>
 	</div>
 </template>
 
 <script setup lang="ts">
-import { nextTick, onMounted, ref, watch } from 'vue';
+import { ElMessage } from 'element-plus';
+import { computed, nextTick, onActivated, onMounted, ref, watch } from 'vue';
 import useClipboard from 'vue-clipboard3';
 import Loding from './components/Loding.vue';
-import { RecordSet } from './model/Record';
 import type { ChatContent } from './model/types';
-import { AnswerType, RoleEnum, answerTypeMapCom, roleImageMap, type ChatMessage } from './model/types';
-import { QuestionAi } from '/@/api/ai/chat';
+import { AnswerType, RoleEnum, answerTypeMapCom, roleImageMap, type ChatMessage, AnswerState } from './model/types';
+import { GetHistoryAnswer, QueryHistoryDetail, QuestionAi, SetHistoryAnswerState, getQuestionProcess } from '/@/api/ai/chat';
 import PlayBar from '/@/components/chat/components/playBar/PlayBar.vue';
 import router from '/@/router';
-import { activeChatRoom } from '/@/stores/chatRoom';
-import { ElMessage } from 'element-plus';
+import { activeChatRoom, activeLLMId, activeRoomId, activeSampleId, activeSectionAId, roomConfig } from '/@/stores/chatRoom';
+import { v4 as uuidv4 } from 'uuid';
+
+const chatWidth = '75%';
 
 let isTalking = ref(false);
 let messageContent = ref<ChatContent>({
 	type: AnswerType.Text,
 	values: '',
 });
+const currentRoute = router.currentRoute;
+const currentRouteId = currentRoute.value.query.id as string;
 const chatListDom = ref<HTMLDivElement>();
 const messageList = ref<ChatMessage[]>([]);
-
-onMounted(() => {
-	if (!activeChatRoom.value) {
-		router.replace({
-			name: 'Home',
-		});
-		return;
-	}
-	messageContent.value = {
-		type: AnswerType.Text,
-		values: activeChatRoom.value.title,
-	};
-	sendChatMessage();
+const computedMessageList = computed(() => {
+	return messageList.value.filter((v) => v.role !== RoleEnum.system);
 });
 
-const questionAi = async (text) => {
-	// const res = await QuestionAi({
-	// 	question: text,
-	// });
-	const res = {
-		json_ok: true,
-		question: '鏄ㄦ棩浜斾竴骞垮満鍘嬪姏',
-		answer_type: 'recordset',
-		values: {
-			names: ['yesterday', 'max_pressure'],
-			values: [
-				['2024-06-28 00:00:00', 24.378],
-				['2024-06-29 00:00:00', 24.276],
-			],
-			type: 'records',
-			title: '鏄ㄦ棩浜斾竴骞垮満(D_GW_04)鐨勬渶澶у帇鍔涘��',
-		},
-	};
+const parseContent = (res) => {
 	let content: ChatContent = {
 		type: AnswerType.Text,
 		values: '鍙戠敓閿欒锛�',
@@ -103,45 +109,102 @@
 		case AnswerType.Text:
 			content = {
 				type: AnswerType.Text,
-				values: res.values,
+				values: res.values ?? res.answer,
+			};
+			break;
+
+		case AnswerType.Knowledge:
+			content = {
+				type: AnswerType.Knowledge,
+				values: res.knowledge,
+			};
+
+			break;
+		case AnswerType.Summary:
+			content = {
+				type: AnswerType.Summary,
+				values: res.summary,
+				error: res.json_msg,
+			};
+			break;
+		case AnswerType.Url:
+			content = {
+				type: AnswerType.Url,
+				values: res.url,
 			};
 			break;
 		default:
+			content = {
+				type: AnswerType.Text,
+				values: '鍙戠敓閿欒锛�',
+			};
 			break;
 	}
 	return content;
 };
+//#region ====================== 鏌ヨ杩涘害 ======================
+let processId = '';
+const QUERY_PROCESS_INTERVAL = 1000;
+const process = ref('');
+let processTimer = null;
+let finishProcess = true;
 
-const sendChatMessage = async (content: ChatContent = messageContent.value) => {
-	if (!messageContent.value?.values) return;
-	if (activeChatRoom.value.isInitial) {
-		activeChatRoom.value.title = messageContent.value.values;
-		activeChatRoom.value.isInitial = false;
-	}
-	try {
-		isTalking.value = true;
-		// 鍙戦�佸綋鍓�
-		messageList.value.push({ role: RoleEnum.user, content });
-		// 娓呯┖杈撳叆妗�
-		clearMessageContent();
-		// 鍑虹幇鍥炲锛岀疆绌哄嚭鐜扮瓑寰呭姩鐢�
-		messageList.value.push({ role: RoleEnum.assistant, content: null });
+const queryProcessApi = async () => {
+	const res = await getQuestionProcess({
+		process_id: processId,
+	}).catch((err) => {
+		process.value = err;
+	});
 
-		let resMsgContent: ChatContent = null;
-		resMsgContent = await questionAi(content.values);
-		appendLastMessageContent(resMsgContent);
-	} catch (error: any) {
-		appendLastMessageContent({
-			type: AnswerType.Text,
-			values: '鍙戠敓閿欒锛�',
-		});
-	} finally {
-		isTalking.value = false;
-	}
+	process.value = res.process;
+	finishProcess = true;
 };
 
-const appendLastMessageContent = (content: ChatContent) => {
-	messageList.value.at(-1).content = content;
+const queryProcess = () => {
+	processTimer = setInterval(() => {
+		if (!finishProcess) return;
+		finishProcess = false;
+		queryProcessApi();
+	}, QUERY_PROCESS_INTERVAL);
+};
+
+const clearQueryProcess = () => {
+	process.value = '';
+	clearInterval(processTimer);
+};
+
+//#endregion
+
+let questionRes = null;
+const questionAi = async (text) => {
+	if (!currentSectionId) {
+		ElMessage.warning('鍙戦�佸け璐ワ紝鏈‘瀹氬簲鐢ㄥ満鏅紒');
+	}
+	processId = uuidv4();
+	const params = {
+		process_id: processId,
+		question: text,
+		// FIXME: 鏆傛椂杩欐牱
+		section_a_id: currentSectionId,
+		history_group_id: currentRouteId,
+		raw_mode: roomConfig.value?.[currentRouteId]?.isAnswerByLLM ?? false,
+	} as any;
+
+	if (currentSampleId) {
+		params.sample_id = currentSampleId;
+	}
+
+	if (currentLLMId) {
+		params.llm_id = currentLLMId;
+	}
+	clearQueryProcess();
+	queryProcess();
+	const res = await QuestionAi(params).finally(() => {
+		clearQueryProcess();
+	});
+	questionRes = res;
+	const content = parseContent(res);
+	return content;
 };
 
 const clearMessageContent = () =>
@@ -152,19 +215,165 @@
 
 const scrollToBottom = () => {
 	if (!chatListDom.value) return;
-	chatListDom.value.lastElementChild.scrollIntoView();
-	// scrollTo(0, chatListDom.value.scrollHeight);
+	chatListDom.value.lastElementChild?.scrollIntoView();
+};
+let currentSectionId = null;
+let currentSampleId = null;
+
+let currentLLMId = null;
+
+const getAnswerById = async (historyId: string) => {
+	return await GetHistoryAnswer({
+		history_id: historyId,
+	});
 };
 
-watch(messageList.value, () => nextTick(() => scrollToBottom()));
+const sendChatMessage = async (content: ChatContent = messageContent.value) => {
+	if (!messageContent.value?.values) return;
+	if (messageList.value.length === 0) {
+		if (activeSampleId.value) {
+			currentSampleId = activeSampleId.value;
+		}
+
+		if (activeLLMId.value) {
+			currentLLMId = activeLLMId.value;
+		}
+
+		if (activeSectionAId.value) {
+			currentSectionId = activeSectionAId.value;
+		}
+	}
+
+	try {
+		isTalking.value = true;
+		const userItem: ChatMessage = { role: RoleEnum.user, content } as any;
+		const assistantItem: ChatMessage = { role: RoleEnum.assistant, content: null, state: AnswerState.Null } as any;
+		// 鍙戦�佸綋鍓�
+		messageList.value.push(userItem);
+		// 娓呯┖杈撳叆妗�
+		clearMessageContent();
+
+		// 鍑虹幇鍥炲锛岀疆绌哄嚭鐜扮瓑寰呭姩鐢�
+		messageList.value.push(assistantItem);
+
+		let resMsgContent: ChatContent = null;
+		resMsgContent = await questionAi(content.values);
+		userItem.historyId = questionRes.history_id;
+		assistantItem.historyId = questionRes.history_id;
+		appendLastMessageContent(resMsgContent);
+	} catch (error: any) {
+		appendLastMessageContent({
+			type: AnswerType.Text,
+			values: '鍙戠敓閿欒锛�',
+		});
+	} finally {
+		isTalking.value = false;
+	}
+};
+const appendLastMessageContent = (content: ChatContent) => {
+	if (messageList.value.at(-1)) {
+		messageList.value.at(-1).content = content;
+	}
+};
+
+onMounted(async () => {
+	const res = await QueryHistoryDetail({
+		history_group_id: currentRouteId,
+	});
+
+	messageList.value = (res.details ?? []).map((item) => {
+		return {
+			historyId: item.history_id,
+			role: RoleEnum.user,
+			content: {
+				type: AnswerType.Text,
+				values: item.question,
+			},
+		} as ChatMessage;
+	});
+	currentSectionId = res?.details?.[0]?.section_a_id;
+	currentSampleId = res?.details?.[0]?.sample_id;
+	const resList = await Promise.all((res.details ?? []).map((item) => getAnswerById(item.history_id)));
+	let i = 0;
+
+	resList.map((item, index) => {
+		const insertIndex = index + 1 + i;
+		messageList.value.splice(insertIndex, 0, {
+			historyId: item.answer.history_id,
+			role: RoleEnum.assistant,
+			content: parseContent(item.answer),
+			state: item.answer_state,
+		});
+		i++;
+	});
+
+	if (messageList.value.length === 0) {
+		messageContent.value = {
+			type: AnswerType.Text,
+			values: activeChatRoom.value.title,
+		};
+
+		sendChatMessage();
+	}
+});
+let forbidScroll = false;
+watch(
+	messageList,
+	() => {
+		if (forbidScroll) return;
+		nextTick(() => scrollToBottom());
+	},
+	{
+		deep: true,
+	}
+);
+
+onActivated(() => {
+	if (forbidScroll) return;
+	nextTick(() => scrollToBottom());
+});
 
 //#region ====================== 鑱婂ぉ鍐呭鎿嶄綔 ======================
 
 const { toClipboard } = useClipboard();
 
-const copyClick = (content) => {
+const copyClick = (item) => {
+	const type = item.content.type;
+	let text = '';
+	if (type === AnswerType.Knowledge) {
+		text = item.content.values?.map((item) => item.answer).join('\n\n') ?? '';
+	} else {
+		text = item.content.values;
+	}
 	ElMessage.success('澶嶅埗鎴愬姛');
-	toClipboard(content);
+	toClipboard(text);
+};
+
+const likeClick = async (item) => {
+	const toSetState = item.state === AnswerState.Like ? AnswerState.Null : AnswerState.Like;
+	const res = await SetHistoryAnswerState({
+		history_id: item.historyId,
+		answer_state: toSetState,
+	});
+	item.state = toSetState;
+	forbidScroll = true;
+	nextTick(() => {
+		forbidScroll = false;
+	});
+};
+
+const unLikeClick = async (item) => {
+	const toSetState = item.state === AnswerState.Unlike ? AnswerState.Null : AnswerState.Unlike;
+	const res = await SetHistoryAnswerState({
+		history_id: item.historyId,
+		answer_state: toSetState,
+	});
+	item.state = toSetState;
+
+	forbidScroll = true;
+	nextTick(() => {
+		forbidScroll = false;
+	});
 };
 //#endregion
 </script>

--
Gitblit v1.9.3