From 495d1510f9a9ebf1f488569f6ca24a6be56bc9f3 Mon Sep 17 00:00:00 2001
From: wujingjing <gersonwu@qq.com>
Date: 星期四, 12 九月 2024 13:59:23 +0800
Subject: [PATCH] Revert "对接接口"
---
src/components/chat/Chat.vue | 227 ++++++++++++++++++++++++++++++++++----------------------
1 files changed, 137 insertions(+), 90 deletions(-)
diff --git a/src/components/chat/Chat.vue b/src/components/chat/Chat.vue
index ec0c7c9..cf387e6 100644
--- a/src/components/chat/Chat.vue
+++ b/src/components/chat/Chat.vue
@@ -10,18 +10,17 @@
<img class="rounded-full size-12 mr-4" :src="roleImageMap[item.role]" alt="" srcset="" />
<div class="flex">
- <div class="relative" v-if="item.content?.values">
+ <div class="relative" v-if="item.content">
<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">
+ :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> -->
+ </div>
</div>
<Loding v-else />
@@ -31,124 +30,172 @@
</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" />
+ <PlayBar :isTalking="isTalking" v-model="messageContent" @sendClick="sendOrSave" />
</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 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 { md } from './libs/markdown';
+import { RoleEnum, roleImageMap, type ChatMessage } from './types';
import PlayBar from '/@/components/chat/components/playBar/PlayBar.vue';
-import router from '/@/router';
import { activeChatRoom } from '/@/stores/chatRoom';
-import { ElMessage } from 'element-plus';
+import router from '/@/router';
+let apiKey = '';
+let isConfig = ref(false);
let isTalking = ref(false);
-let messageContent = ref<ChatContent>({
- type: AnswerType.Text,
- values: '',
-});
+let messageContent = ref('');
const chatListDom = ref<HTMLDivElement>();
-const messageList = ref<ChatMessage[]>([]);
+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();
+ }
if (!activeChatRoom.value) {
router.replace({
name: 'Home',
});
return;
}
- messageContent.value = {
- type: AnswerType.Text,
- values: activeChatRoom.value.title,
- };
- sendChatMessage();
+ messageContent.value = activeChatRoom.value.title;
+ sendOrSave();
});
-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)鐨勬渶澶у帇鍔涘��',
- },
- };
- 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,
- };
- break;
- default:
- break;
- }
- return content;
-};
-
-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;
- }
+const sendChatMessage = async (content: string = messageContent.value) => {
try {
isTalking.value = true;
- // 鍙戦�佸綋鍓�
+ // if (messageList.value.length === 0) {
+ // messageList.value.pop();
+ // }
messageList.value.push({ role: RoleEnum.user, content });
- // 娓呯┖杈撳叆妗�
clearMessageContent();
- // 鍑虹幇鍥炲锛岀疆绌哄嚭鐜扮瓑寰呭姩鐢�
- messageList.value.push({ role: RoleEnum.assistant, content: null });
+ messageList.value.push({ role: RoleEnum.assistant, content: '' });
- let resMsgContent: ChatContent = null;
- resMsgContent = await questionAi(content.values);
- appendLastMessageContent(resMsgContent);
- } catch (error: any) {
- appendLastMessageContent({
- type: AnswerType.Text,
- values: '鍙戠敓閿欒锛�',
+ // 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);
});
+
+ const msg = await a;
+ appendLastMessageContent(msg);
+ } catch (error: any) {
+ appendLastMessageContent(error);
} finally {
isTalking.value = false;
}
};
-const appendLastMessageContent = (content: ChatContent) => {
- 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);
+ 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 clearMessageContent = () =>
- (messageContent.value = {
- type: AnswerType.Text,
- values: '',
- });
+const appendLastMessageContent = (content: string) => (messageList.value[messageList.value.length - 1].content += content);
+
+const sendOrSave = () => {
+ if (!messageContent.value.length) return;
+ if (activeChatRoom.value.isInitial) {
+ activeChatRoom.value.title = messageContent.value;
+ activeChatRoom.value.isInitial = false;
+ }
+ if (isConfig.value) {
+ if (saveAPIKey(messageContent.value.trim())) {
+ switchConfigStatus();
+ }
+ clearMessageContent();
+ } else {
+ sendChatMessage();
+ }
+};
+
+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;
--
Gitblit v1.9.3