From 44025036ca9baea036a629c78273e9df48ba8e2e Mon Sep 17 00:00:00 2001
From: yangyin <1850366751@qq.com>
Date: 星期五, 05 七月 2024 14:04:39 +0800
Subject: [PATCH] fix: 修改应用场景
---
src/components/chat/Chat.vue | 415 +++++++++++++++++++++++++++++++++++++---------------------
1 files changed, 265 insertions(+), 150 deletions(-)
diff --git a/src/components/chat/Chat.vue b/src/components/chat/Chat.vue
index e0dd334..d09d425 100644
--- a/src/components/chat/Chat.vue
+++ b/src/components/chat/Chat.vue
@@ -1,25 +1,37 @@
<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"
- >
+ <div ref="chatListDom" class="h-full w-[100ch]">
+ <div class="group flex px-4 py-4 hover:bg-slate-100 rounded-lg" v-for="(item, index) of computedMessageList" :key="index">
<img class="rounded-full size-12 mr-4" :src="roleImageMap[item.role]" alt="" srcset="" />
<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 class="relative" v-if="item.content?.values">
+ <div class="text-sm rounded-[6px] p-4 leading-relaxed max-w-[100ch] bg-white">
+ <component class="max-w-[100ch]" :is="answerTypeMapCom[item.content.type]" :data="item.content.values" />
+ </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">
+ <i
+ class="p-2 ywicon icon-copy cursor-pointer hover:text-[#0284ff] hover:!text-[18px]"
+ @click="copyClick(item.content.values)"
+ />
+ </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>
@@ -30,172 +42,248 @@
</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" />
+ <PlayBar :isTalking="isTalking" v-model="messageContent.values" @sendClick="sendChatMessage" />
</div>
</div>
</template>
<script setup lang="ts">
-import cryptoJS from 'crypto-js';
import { ElMessage } from 'element-plus';
-import { nextTick, onMounted, ref, watch } from 'vue';
+import { computed, nextTick, ref, watch } from 'vue';
import useClipboard from 'vue-clipboard3';
import Loding from './components/Loding.vue';
-import { md } from './libs/markdown';
-import { RoleEnum, roleImageMap, type ChatMessage } from './types';
+import type { ChatContent } from './model/types';
+import { AnswerType, RoleEnum, answerTypeMapCom, roleImageMap, type ChatMessage, AnswerState } from './model/types';
+import { GetHistoryAnswer, QueryHistoryDetail, QuestionAi, SetHistoryAnswerState } from '/@/api/ai/chat';
import PlayBar from '/@/components/chat/components/playBar/PlayBar.vue';
+import router from '/@/router';
+import { activeChatRoom, activeLLMId, activeRoomId, activeSampleId, activeSectionAId } from '/@/stores/chatRoom';
-let apiKey = '';
-let isConfig = ref(false);
let isTalking = ref(false);
-let messageContent = ref('');
+let messageContent = ref<ChatContent>({
+ type: AnswerType.Text,
+ values: '',
+});
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.role !== RoleEnum.system);
});
-const sendChatMessage = async (content: string = messageContent.value) => {
+const parseContent = (res) => {
+ 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;
+ default:
+ content = {
+ type: AnswerType.Text,
+ values: '鍙戠敓閿欒锛�',
+ };
+ break;
+ }
+ return content;
+};
+
+let questionRes = null;
+const questionAi = async (text) => {
+ if (!currentSectionId) {
+ ElMessage.warning('鍙戦�佸け璐ワ紝鏈‘瀹氬簲鐢ㄥ満鏅紒');
+ }
+
+ const params = {
+ question: text,
+ // FIXME: 鏆傛椂杩欐牱
+ section_a_id: currentSectionId,
+ history_group_id: activeRoomId.value,
+ } as any;
+
+ if (currentSampleId) {
+ params.sample_id = currentSampleId;
+ }
+
+ if (currentLLMId) {
+ params.llm_id = currentLLMId;
+ }
+
+ const res = await QuestionAi(params);
+ questionRes = res;
+ // 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 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 (!messageContent.value?.values) return;
+ if (activeChatRoom.value.isInitial) {
+ activeChatRoom.value.title = messageContent.value.values;
+ activeChatRoom.value.isInitial = false;
+ 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('浣犲ソ锛屾垜鏄疉I璇█妯″瀷 ');
- }, 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 appendLastMessageContent = (content: ChatContent) => {
+ if (messageList.value.at(-1)) {
+ messageList.value.at(-1).content = content;
+ }
+};
-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);
+watch(
+ () => activeRoomId.value,
+ async (val) => {
+ if (!val) {
+ router.replace({
+ name: 'Home',
+ });
return;
}
+ if (activeChatRoom.value.isInitial) {
+ messageContent.value = {
+ type: AnswerType.Text,
+ values: activeChatRoom.value.title,
+ };
+ sendChatMessage();
+ } else {
+ const res = await QueryHistoryDetail({
+ history_group_id: activeRoomId.value,
+ });
+ 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;
- 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);
+ 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++;
+ });
}
+ },
+ {
+ immediate: true,
}
-};
+);
-const appendLastMessageContent = (content: string) => (messageList.value[messageList.value.length - 1].content += content);
-
-const sendOrSave = () => {
- if (!messageContent.value.length) return;
- if (isConfig.value) {
- if (saveAPIKey(messageContent.value.trim())) {
- switchConfigStatus();
- }
- clearMessageContent();
- } else {
- sendChatMessage();
+let forbidScroll = false;
+watch(
+ messageList,
+ () => {
+ if (forbidScroll) return;
+ nextTick(() => scrollToBottom());
+ },
+ {
+ deep: true,
}
-};
-
-const clickConfig = () => {
- if (!isConfig.value) {
- messageContent.value = getAPIKey();
- } else {
- clearMessageContent();
- }
- switchConfigStatus();
-};
-
-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 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 ====================== 鑱婂ぉ鍐呭鎿嶄綔 ======================
@@ -205,6 +293,33 @@
ElMessage.success('澶嶅埗鎴愬姛');
toClipboard(content);
};
+
+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