From c123adf7cb8caa3a92f1143755092371ef4a0658 Mon Sep 17 00:00:00 2001
From: wujingjing <gersonwu@qq.com>
Date: 星期一, 22 七月 2024 13:51:37 +0800
Subject: [PATCH] 整理 chat.vue
---
src/components/chat/Chat.vue | 542 +++++++++++++++++++++++++++++++++++++----------------
1 files changed, 378 insertions(+), 164 deletions(-)
diff --git a/src/components/chat/Chat.vue b/src/components/chat/Chat.vue
index e0dd334..154e00e 100644
--- a/src/components/chat/Chat.vue
+++ b/src/components/chat/Chat.vue
@@ -1,209 +1,423 @@
<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
- class="group flex px-4 py-4 hover:bg-slate-100 rounded-lg"
- v-for="(item, index) of messageList.filter((v) => v.role !== 'system')"
- :key="index"
- >
- <img class="rounded-full size-12 mr-4" :src="roleImageMap[item.role]" alt="" srcset="" />
+ <div class="flex h-full">
+ <div class="flex flex-col h-full flex-auto">
+ <div class="h-full flex flex-col items-center overflow-y-auto">
+ <div ref="chatListDom" class="h-full" :style="{ width: chatWidth }">
+ <div
+ class="group flex px-4 py-6 hover:bg-slate-100 rounded-lg relative"
+ :class="{ 'flex-row-reverse': item.role === RoleEnum.user }"
+ v-for="(item, index) of computedMessageList"
+ :key="index"
+ >
+ <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="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="flex-column w-full">
+ <p class="text-red-500">
+ {{ item.content.msg }}
+ </p>
+ <div class="mt-5 flex items-center" v-if="showFixQuestion(item)">
+ <div class="text-gray-600 flex-0">
+ {{ item.content.origin.err_json.fix_question.title + '锛�' }}
+ </div>
+ <div class="ml-1 space-x-2 inline-flex flex-wrap">
+ <div
+ v-for="fixItem in item.content.origin.err_json.fix_question?.values"
+ :key="fixItem"
+ class="bg-gray-200 p-3 hover:bg-[#c5e0ff] hover:text-[#1c86ff] cursor-pointer rounded-lg"
+ @click="fixQuestionClick(fixItem)"
+ >
+ {{ fixItem.title }}
+ </div>
+ </div>
+ </div>
+ </div>
+ <component v-else :is="answerTypeMapCom[item.content.type]" :data="item.content.values" :originData="item" />
+ </div>
- <div class="flex">
- <div class="relative" 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 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 v-if="item.role === RoleEnum.assistant" class="absolute flex items-center right-0 mr-4 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>
+ <template v-if="item.content.errCode !== ErrorCode.Message">
+ <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>
+ </template>
+
+ <div class="flex items-center justify-center size-[15px] relative">
+ <i
+ class="p-2 ywicon icon-wentifankui cursor-pointer hover:text-[#0284ff] !text-[13px] hover:!text-[15px]"
+ @click="
+ ($event) =>
+ feedbackClick(
+ $event,
+ item,
+ computedMessageList
+ .filter((v) => v.role === RoleEnum.assistant)
+ .findIndex((v) => v.historyId === item.historyId)
+ )
+ "
+ />
+ <FeedbackPanel
+ v-show="feedbackIsShow && currentFeedbackMapItem === item"
+ ref="feedbackPanelRef"
+ v-model:isShow="feedbackIsShow"
+ v-model:content="feedbackContent"
+ :chatItem="currentFeedbackMapItem"
+ :position="feedbackPosition"
+ />
+ </div>
+ </div>
+ </div>
+
+ <Loding v-else class="w-fit" :process="process" />
</div>
</div>
-
- <Loding v-else />
+ </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-[rgb(247,248,250)] flex justify-center">
+ <PlayBar
+ v-model:voicePageIsShow="voicePageIsShow"
+ :isTalking="isTalking"
+ :isHome="false"
+ v-model="messageContent.values"
+ @sendClick="sendClick"
+ :style="{ width: chatWidth }"
+ ></PlayBar>
+ </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>
+ <CustomDrawer v-model:isShow="drawerIsShow" @updateChatInput="updateChatInput" />
</div>
</template>
<script setup lang="ts">
-import cryptoJS from 'crypto-js';
import { ElMessage } from 'element-plus';
-import { nextTick, onMounted, ref, watch } from 'vue';
-import useClipboard from 'vue-clipboard3';
+import _ from 'lodash';
+import { v4 as uuidv4 } from 'uuid';
+import { computed, onMounted, ref } from 'vue';
+import FeedbackPanel from './components/FeedbackPanel.vue';
import Loding from './components/Loding.vue';
-import { md } from './libs/markdown';
-import { RoleEnum, roleImageMap, type ChatMessage } from './types';
+import { useAssistantContentOpt } from './hooks/useAssistantContentOpt';
+import { useQueryProcess } from './hooks/useQueryProcess';
+import { useScrollToBottom } from './hooks/useScrollToBottom';
+import type { ChatContent } from './model/types';
+import { AnswerState, AnswerType, RoleEnum, answerTypeMapCom, roleImageMap, type ChatMessage } from './model/types';
+import { GetHistoryAnswer, QueryHistoryDetail, QuestionAi } from '/@/api/ai/chat';
import PlayBar from '/@/components/chat/components/playBar/PlayBar.vue';
+import CustomDrawer from '/@/components/drawer/CustomDrawer.vue';
+import router from '/@/router';
+import { activeChatRoom, activeLLMId, activeSampleId, activeSectionAId, getRoomConfig, roomConfig } from '/@/stores/chatRoom';
+import { ErrorCode } from '/@/utils/request';
-let apiKey = '';
-let isConfig = ref(false);
+const chatWidth = '75%';
+const voicePageIsShow = ref(false);
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. 闂茶亰锛氬鏋滀綘鎰熷埌瀵傚癁鎴栨棤鑱婏紝鎴戜滑鍙互鑱婁竴浜涙湁瓒ｇ殑璇濋锛屼互鍑忚交浣犵殑鍘嬪姏銆�
-
- 璇峰憡璇夋垜浣犻渶瑕佸摢鏂归潰鐨勫府鍔╋紝鎴戜細鏍规嵁浣犵殑闇�姹傜粰浣犳彁渚涚浉搴旂殑淇℃伅鍜屽缓璁�俙,
- },
- {
- 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);
});
-const sendChatMessage = async (content: string = messageContent.value) => {
+const parseContent = (res) => {
+ if (!res) return null;
+ let content: ChatContent = {
+ type: AnswerType.Text,
+ values: '瑙ｆ瀽澶辫触锛�',
+ };
+
+ 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;
+ case AnswerType.Map:
+ content = {
+ type: AnswerType.Map,
+ values: res.values,
+ };
+ break;
+ default:
+ content = {
+ type: AnswerType.Text,
+ values: '瑙ｆ瀽澶辫触锛�',
+ };
+ break;
+ }
+ content.askMoreList = _.orderBy(res.context_history, [(item) => Number(item.radio)], ['desc']);
+ content.errCode = res?.err_code;
+ content.msg = res?.json_msg;
+ content.origin = res;
+ return content;
+};
+
+const { clearQueryProcess, process, processId, queryProcess } = useQueryProcess();
+
+let questionRes = null;
+const questionAi = async (text) => {
+ if (!currentSectionId) {
+ ElMessage.warning('鍙戦�佸け璐ワ紝鏈‘瀹氬簲鐢ㄥ満鏅紒');
+ }
+ processId.value = uuidv4();
+ const params = {
+ process_id: processId.value,
+ question: text,
+ // FIXME: 鏆傛椂杩欐牱
+ section_a_id: currentSectionId,
+ history_group_id: currentRouteId,
+ raw_mode: roomConfig.value?.[currentRouteId]?.isAnswerByLLM ?? false,
+ next_chat: isNextChat.value,
+ } 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: '',
+ });
+
+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, cb?: any) => {
+ if (!content?.values) return;
+ const isNewChat = messageList.value.length === 0;
+ if (isNewChat) {
+ if (activeSampleId.value) {
+ currentSampleId = activeSampleId.value;
+ }
+
+ if (activeLLMId.value) {
+ currentLLMId = activeLLMId.value;
+ }
+
+ if (activeSectionAId.value) {
+ currentSectionId = activeSectionAId.value;
+ }
+ }
+ let resMsgContent: ChatContent = null;
+
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('浣犲ソ锛屾垜鏄疉I璇█妯″瀷 ');
- }, 500);
- });
+ // 鍑虹幇鍥炲锛岀疆绌哄嚭鐜扮瓑寰呭姩鐢�
+ messageList.value.push(assistantItem);
- const msg = await a;
- appendLastMessageContent(msg);
+ resMsgContent = await questionAi(content.values);
+ if (isNewChat) {
+ const firstResCb = getRoomConfig(currentRouteId, 'firstResCb');
+ firstResCb?.(resMsgContent);
+ } else {
+ cb?.(resMsgContent);
+ }
+ 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 sendClick = (cb) => {
+ sendChatMessage(messageContent.value, cb);
+};
+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);
+onMounted(async () => {
+ const res = await QueryHistoryDetail({
+ history_group_id: currentRouteId,
+ });
-const sendOrSave = () => {
- if (!messageContent.value.length) return;
- if (isConfig.value) {
- if (saveAPIKey(messageContent.value.trim())) {
- switchConfigStatus();
- }
- clearMessageContent();
- } else {
+ 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,
+ item.answer === null
+ ? null
+ : {
+ 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();
}
-};
+});
-const clickConfig = () => {
- if (!isConfig.value) {
- messageContent.value = getAPIKey();
- } else {
- clearMessageContent();
- }
- switchConfigStatus();
-};
+const { forbidScroll } = useScrollToBottom({
+ chatListDom: chatListDom,
+ displayMessageList: computedMessageList,
+});
-const getSecretKey = () => 'lianginx';
+const {
+ copyClick,
+ likeClick,
+ unLikeClick,
+ feedbackPosition,
+ feedbackIsShow,
+ feedbackContent,
+ feedbackPanelRef,
+ currentFeedbackMapItem,
+ feedbackClick,
+ askMoreClick,
+ fixQuestionClick,
+ isNextChat,
+ showFixQuestion,
+ showAskMore,
+} = useAssistantContentOpt({
+ forbidScroll,
+ sendChatMessage,
+ displayMessageList: computedMessageList,
+});
-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;
-};
+//#region ====================== 渚ц竟鏍廳rawer ======================
+const drawerIsShow = ref(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 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()));
-
-//#region ====================== 鑱婂ぉ鍐呭鎿嶄綔 ======================
-
-const { toClipboard } = useClipboard();
-
-const copyClick = (content) => {
- ElMessage.success('澶嶅埗鎴愬姛');
- toClipboard(content);
+const updateChatInput = (content) => {
+ messageContent.value.values = content;
};
//#endregion
</script>
--
Gitblit v1.9.3