From 9667ced39ea1343e044d18552221f2ad41e22c6a Mon Sep 17 00:00:00 2001
From: gerson <1405270578@qq.com>
Date: 星期日, 30 六月 2024 22:16:41 +0800
Subject: [PATCH] sample input
---
src/components/chat/Chat.vue | 341 ++++++++++++++++++++++++++++++++------------------------
1 files changed, 196 insertions(+), 145 deletions(-)
diff --git a/src/components/chat/Chat.vue b/src/components/chat/Chat.vue
index 0d6cdc2..1bd8702 100644
--- a/src/components/chat/Chat.vue
+++ b/src/components/chat/Chat.vue
@@ -1,21 +1,23 @@
<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-3 hover:bg-slate-100 rounded-lg"
- v-for="(item, index) of messageList.filter((v) => v.role !== 'system')"
- :key="index"
- >
- <img class="rounded-full size-10 mr-4" :src="roleImageMap[item.role]" alt="" srcset="" />
+ <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="{ 'bg-[#d8d8ff]': item.role !== RoleEnum.user }"
- class="prose text-sm rounded-[6px] p-4 leading-relaxed max-w-[100ch]"
- v-if="item.content"
- v-html="md.render(item.content)"
- ></div>
+ <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 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>
+
<Loding v-else />
</div>
</div>
@@ -23,168 +25,217 @@
</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 { nextTick, onMounted, ref, watch } from 'vue';
+import { ElMessage } from 'element-plus';
+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, type ChatMessage, roleImageMap } from './types';
+import type { ChatContent } from './model/types';
+import { 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 router from '/@/router';
+import { activeChatRoom, activeRoomId } 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 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;
+};
+
+const questionAi = async (text) => {
+ const res = await QuestionAi({
+ question: text,
+ // FIXME: 鏆傛椂杩欐牱
+ section_a_id: 'undefined',
+ history_group_id: activeRoomId.value,
+ });
+ // 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();
+};
+
+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;
+ }
try {
isTalking.value = true;
- if (messageList.value.length === 2) {
- messageList.value.pop();
- }
+ // 鍙戦�佸綋鍓�
messageList.value.push({ role: RoleEnum.user, content });
+ // 娓呯┖杈撳叆妗�
clearMessageContent();
- messageList.value.push({ role: RoleEnum.assistant, content: '' });
+ // 鍑虹幇鍥炲锛岀疆绌哄嚭鐜扮瓑寰呭姩鐢�
+ messageList.value.push({ role: RoleEnum.assistant, content: null });
- // 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);
- });
-
- const msg = await a;
- appendLastMessageContent(msg);
+ let resMsgContent: ChatContent = null;
+ resMsgContent = await questionAi(content.values);
+ 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);
+const appendLastMessageContent = (content: ChatContent) => {
+ if (messageList.value.at(-1)) {
+ messageList.value.at(-1).content = content;
+ }
+};
+let currentSectionId = '';
+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 {
+ role: RoleEnum.user,
+ content: {
+ type: AnswerType.Text,
+ values: item.question,
+ },
+ } as ChatMessage;
+ });
+ currentSectionId = res?.details?.[0]?.section_a_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, {
+ role: RoleEnum.assistant,
+ content: parseContent(item.answer),
+ });
+ 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();
+watch(
+ messageList,
+ () => {
+ nextTick(() => scrollToBottom());
+ },
+ {
+ deep: true,
}
+);
+
+//#region ====================== 鑱婂ぉ鍐呭鎿嶄綔 ======================
+
+const { toClipboard } = useClipboard();
+
+const copyClick = (content) => {
+ ElMessage.success('澶嶅埗鎴愬姛');
+ toClipboard(content);
};
-
-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()));
+//#endregion
</script>
<style scoped>
--
Gitblit v1.9.3