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