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