From 46df470bbba226da5224d9adc1c47ecfadf11e1d Mon Sep 17 00:00:00 2001
From: wujingjing <gersonwu@qq.com>
Date: 星期四, 18 七月 2024 14:21:01 +0800
Subject: [PATCH] chart resize

---
 src/components/chat/Chat.vue |  508 +++++++++++++++++++++++++++++++++++++++----------------
 1 files changed, 359 insertions(+), 149 deletions(-)

diff --git a/src/components/chat/Chat.vue b/src/components/chat/Chat.vue
index 364d0b3..7088f2e 100644
--- a/src/components/chat/Chat.vue
+++ b/src/components/chat/Chat.vue
@@ -1,203 +1,413 @@
 <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">
+			<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 flex-0"
+						:class="{ 'mr-4': item.role === RoleEnum.assistant, 'ml-4': item.role === RoleEnum.user }"
+						:src="roleImageMap[item.role]"
+						alt=""
+						srcset=""
+					/>
+					<div class="flex-auto flex" :class="{'justify-end':item.role===RoleEnum.user}">
+						<div class="inline-flex flex-col" :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"
+									:style="{ backgroundColor: item.role === RoleEnum.user ? 'rgb(197 224 255)' : 'white' }"
+								>
+									<div v-if="item.content.errCode === ErrorCode.Message" class="text-red-500 w-full">{{ item.content.msg }}</div>
+									<component v-else :is="answerTypeMapCom[item.content.type]" :data="item.content.values" />
+								</div>
 
-					<div class="flex">
-						<div v-if="item.content">
-							<div
-								:class="{ 'bg-[#d8d8ff]': item.role === RoleEnum.assistant, 'bg-white': item.role === RoleEnum.user }"
-								class="prose text-sm rounded-[6px] p-4 leading-relaxed max-w-[100ch]"
-								v-html="md.render(item.content)"
-							></div>
-							<div class="">
-								<SvgIcon name="ele-CopyDocument"/>
-								<SvgIcon name="ele-Check"/>
-								<SvgIcon name="ywicon icon-dianzan"/>
-								<SvgIcon name="ywicon icon-buzan"/>
-
-
+								<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>
-						</div>
 
-						<Loding v-else />
+							<Loding v-else class="w-fit" :process="process" />
+						</div>
+					</div>
+				</div>
+				<div v-if="showAskMore" class="ml-4 mt-5 text-sm">
+					<div class="text-gray-600 mb-5">浣犲彲浠ョ户缁棶鎴戯細</div>
+					<div class="space-y-2 inline-flex flex-col">
+						<div
+							v-for="item in computedMessageList.at(-1).content.askMoreList"
+							:key="item.history_id"
+							class="bg-white p-3 hover:bg-[#c5e0ff] hover:text-[#1c86ff] cursor-pointer rounded-lg"
+							@click="askMoreClick(item)"
+						>
+							{{ item.question }}
+						</div>
 					</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" @sendClick="sendOrSave" />
+		<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 cryptoJS from 'crypto-js';
-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 { md } from './libs/markdown';
-import { RoleEnum, type ChatMessage, roleImageMap } from './types';
+import type { ChatContent } from './model/types';
+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, activeLLMId, activeRoomId, activeSampleId, activeSectionAId, roomConfig } from '/@/stores/chatRoom';
+import { v4 as uuidv4 } from 'uuid';
+import _ from 'lodash';
+import { ErrorCode } from '/@/utils/request';
 
-let apiKey = '';
-let isConfig = ref(false);
+const chatWidth = '75%';
+
 let isTalking = ref(false);
-let messageContent = ref('');
+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 decoder = new TextDecoder('utf-8');
-const roleAlias = { user: 'ME', assistant: 'ChatGPT', system: 'System' };
-const messageList = ref<ChatMessage[]>([
-	{
-		role: RoleEnum.assistant,
-		content: `浣犲ソ锛屾垜鏄疉I璇█妯″瀷锛屾垜鍙互鎻愪緵涓�浜涘父鐢ㄦ湇鍔″拰淇℃伅锛屼緥濡傦細
-  
-  1. 缈昏瘧锛氭垜鍙互鎶婁腑鏂囩炕璇戞垚鑻辨枃锛岃嫳鏂囩炕璇戞垚涓枃锛岃繕鏈夊叾浠栦竴浜涜瑷�缈昏瘧锛屾瘮濡傛硶璇�佹棩璇�佽タ鐝墮璇瓑銆�
-  
-  2. 鍜ㄨ鏈嶅姟锛氬鏋滀綘鏈変换浣曢棶棰橀渶瑕佸挩璇紝渚嬪鍋ュ悍銆佹硶寰嬨�佹姇璧勭瓑鏂归潰锛屾垜鍙互灏藉彲鑳戒负浣犳彁渚涘府鍔┿��
-  
-  3. 闂茶亰锛氬鏋滀綘鎰熷埌瀵傚癁鎴栨棤鑱婏紝鎴戜滑鍙互鑱婁竴浜涙湁瓒g殑璇濋锛屼互鍑忚交浣犵殑鍘嬪姏銆�
-  
-  璇峰憡璇夋垜浣犻渶瑕佸摢鏂归潰鐨勫府鍔╋紝鎴戜細鏍规嵁浣犵殑闇�姹傜粰浣犳彁渚涚浉搴旂殑淇℃伅鍜屽缓璁�俙,
-	},
-	{
-		role: RoleEnum.user,
-		content: `浣犲ソ`,
-	},
-]);
-
-onMounted(() => {
-	if (getAPIKey()) {
-		switchConfigStatus();
-	}
-
-	const inputValue = history.state.inputValue;
+const messageList = ref<ChatMessage[]>([]);
+const computedMessageList = computed(() => {
+	return messageList.value.filter((v) => v.role !== RoleEnum.system);
 });
 
-const sendChatMessage = async (content: string = messageContent.value) => {
+const showAskMore = computed(() => {
+	if (!computedMessageList.value || computedMessageList.value.length === 0) return false;
+	const last = computedMessageList.value.at(-1);
+	const isShow = last?.role === RoleEnum.assistant && last?.content?.values && last.content?.askMoreList?.length > 0;
+	return isShow;
+});
+
+const parseContent = (res) => {
+	let content: ChatContent = {
+		type: AnswerType.Text,
+		values: '瑙f瀽澶辫触锛�',
+	};
+
+	switch (res.answer_type) {
+		case AnswerType.RecordSet:
+			content = {
+				type: AnswerType.RecordSet,
+				values: res.values,
+			};
+			break;
+		case AnswerType.Text:
+			content = {
+				type: AnswerType.Text,
+				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,
+			};
+			break;
+		case AnswerType.Url:
+			content = {
+				type: AnswerType.Url,
+				values: res.url,
+			};
+			break;
+		default:
+			content = {
+				type: AnswerType.Text,
+				values: '瑙f瀽澶辫触锛�',
+			};
+			break;
+	}
+	content.askMoreList = _.orderBy(res.context_history, [(item) => Number(item.radio)], ['desc']);
+	content.errCode = res?.err_code;
+	content.msg = res?.json_msg;
+	return content;
+};
+//#region ====================== 鏌ヨ杩涘害 ======================
+let processId = '';
+const QUERY_PROCESS_INTERVAL = 1000;
+const process = ref('');
+let processTimer = null;
+let finishProcess = true;
+
+const queryProcessApi = async () => {
+	const res = await getQuestionProcess({
+		process_id: processId,
+	}).catch((err) => {
+		process.value = err;
+	});
+
+	process.value = res.process;
+	finishProcess = true;
+};
+
+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 = () =>
+	(messageContent.value = {
+		type: AnswerType.Text,
+		values: '',
+	});
+
+const scrollToBottom = () => {
+	if (!chatListDom.value) return;
+	chatListDom.value.lastElementChild?.scrollIntoView();
+};
+let currentSectionId = null;
+let currentSampleId = null;
+
+let currentLLMId = null;
+
+const getAnswerById = async (historyId: string) => {
+	return await GetHistoryAnswer({
+		history_id: historyId,
+	});
+};
+
+const sendChatMessage = async (content: ChatContent = messageContent.value) => {
+	if (!content?.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;
-		if (messageList.value.length === 2) {
-			messageList.value.pop();
-		}
-		messageList.value.push({ role: RoleEnum.user, content });
+		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({ role: RoleEnum.assistant, content: '' });
 
-		// const { body, status } = await chat(messageList.value, getAPIKey());
-		// if (body) {
-		//   const reader = body.getReader();
-		//   await readStream(reader, status);
-		// }
-		const a = new Promise<string>((resolve) => {
-			setTimeout(() => {
-				resolve('浣犲ソ ');
-			}, 500);
-		});
+		// 鍑虹幇鍥炲锛岀疆绌哄嚭鐜扮瓑寰呭姩鐢�
+		messageList.value.push(assistantItem);
 
-		const msg = await a;
-		appendLastMessageContent(msg);
+		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(error);
+		appendLastMessageContent({
+			type: AnswerType.Text,
+			values: '鍙戠敓閿欒锛�',
+		});
 	} finally {
 		isTalking.value = false;
 	}
 };
-
-const readStream = async (reader: ReadableStreamDefaultReader<Uint8Array>, status: number) => {
-	let partialLine = '';
-
-	while (true) {
-		// eslint-disable-next-line no-await-in-loop
-		const { value, done } = await reader.read();
-		if (done) break;
-
-		const decodedText = decoder.decode(value, { stream: true });
-
-		if (status !== 200) {
-			const json = JSON.parse(decodedText); // start with "data: "
-			const content = json.error.message ?? decodedText;
-			appendLastMessageContent(content);
-			return;
-		}
-
-		const chunk = partialLine + decodedText;
-		const newLines = chunk.split(/\r?\n/);
-
-		partialLine = newLines.pop() ?? '';
-
-		for (const line of newLines) {
-			if (line.length === 0) continue; // ignore empty message
-			if (line.startsWith(':')) continue; // ignore sse comment message
-			if (line === 'data: [DONE]') return; //
-
-			const json = JSON.parse(line.substring(6)); // start with "data: "
-			const content = status === 200 ? json.choices[0].delta.content ?? '' : json.error.message;
-			appendLastMessageContent(content);
-		}
+const appendLastMessageContent = (content: ChatContent) => {
+	if (messageList.value.at(-1)) {
+		messageList.value.at(-1).content = content;
 	}
 };
 
-const appendLastMessageContent = (content: string) => (messageList.value[messageList.value.length - 1].content += content);
+const askMoreClick = (item) => {
+	if (!item.question) return;
+	sendChatMessage({ type: AnswerType.Text, values: item.question });
+};
 
-const sendOrSave = () => {
-	if (!messageContent.value.length) return;
-	if (isConfig.value) {
-		if (saveAPIKey(messageContent.value.trim())) {
-			switchConfigStatus();
-		}
-		clearMessageContent();
-	} else {
+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,
+	}
+);
 
-const clickConfig = () => {
-	if (!isConfig.value) {
-		messageContent.value = getAPIKey();
+onActivated(() => {
+	if (forbidScroll) return;
+	nextTick(() => scrollToBottom());
+});
+
+//#region ====================== 鑱婂ぉ鍐呭鎿嶄綔 ======================
+
+const { toClipboard } = useClipboard();
+
+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 {
-		clearMessageContent();
+		text = item.content.values;
 	}
-	switchConfigStatus();
+	ElMessage.success('澶嶅埗鎴愬姛');
+	toClipboard(text);
 };
 
-const getSecretKey = () => 'lianginx';
-
-const saveAPIKey = (apiKey: string) => {
-	if (apiKey.slice(0, 3) !== 'sk-' || apiKey.length !== 51) {
-		alert('API Key 閿欒锛岃妫�鏌ュ悗閲嶆柊杈撳叆锛�');
-		return false;
-	}
-	const aesAPIKey = cryptoJS.AES.encrypt(apiKey, getSecretKey()).toString();
-	localStorage.setItem('apiKey', aesAPIKey);
-	return true;
+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 getAPIKey = () => {
-	if (apiKey) return apiKey;
-	const aesAPIKey = localStorage.getItem('apiKey') ?? '';
-	apiKey = cryptoJS.AES.decrypt(aesAPIKey, getSecretKey()).toString(cryptoJS.enc.Utf8);
-	return apiKey;
+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;
+	});
 };
-
-const switchConfigStatus = () => (isConfig.value = !isConfig.value);
-
-const clearMessageContent = () => (messageContent.value = '');
-
-const scrollToBottom = () => {
-	if (!chatListDom.value) return;
-	chatListDom.value.lastElementChild.scrollIntoView();
-	// scrollTo(0, chatListDom.value.scrollHeight);
-};
-
-watch(messageList.value, () => nextTick(() => scrollToBottom()));
+//#endregion
 </script>
 
 <style scoped>

--
Gitblit v1.9.3