wujingjing
2024-07-18 46df470bbba226da5224d9adc1c47ecfadf11e1d
src/components/chat/Chat.vue
@@ -1,63 +1,129 @@
<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 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 ref="chatListDom" class="h-full" :style="{ width: chatWidth }">
            <div
               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 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="relative 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="text-red-500 w-full">{{ item.content.msg }}</div>
                           <component v-else :is="answerTypeMapCom[item.content.type]" :data="item.content.values" />
                        </div>
               <div class="flex">
                  <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 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>
                     <!-- <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> -->
                     <Loding v-else class="w-fit" :process="process" />
                  </div>
                  <Loding v-else />
               </div>
            </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-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 { ElMessage } from 'element-plus';
import { computed, nextTick, ref, watch } from 'vue';
import { computed, nextTick, onActivated, onMounted, ref, watch } from 'vue';
import useClipboard from 'vue-clipboard3';
import Loding from './components/Loding.vue';
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 { 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, activeLLMId, activeRoomId, activeSampleId, activeSectionAId } from '/@/stores/chatRoom';
import { activeChatRoom, activeLLMId, activeRoomId, activeSampleId, activeSectionAId, roomConfig } from '/@/stores/chatRoom';
import { v4 as uuidv4 } from 'uuid';
import _ from 'lodash';
import { ErrorCode } from '/@/utils/request';
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[]>([]);
const computedMessageList = computed(() => {
   return messageList.value.filter((v) => v.role !== RoleEnum.system);
});
const showAskMore = computed(() => {
   if (!computedMessageList.value || computedMessageList.value.length === 0) return false;
   const last = computedMessageList.value.at(-1);
   const isShow = last?.role === RoleEnum.assistant && last?.content?.values && last.content?.askMoreList?.length > 0;
   return isShow;
});
const parseContent = (res) => {
   let content: ChatContent = {
      type: AnswerType.Text,
      values: '发生错误!',
      values: '解析失败!',
   };
   switch (res.answer_type) {
      case AnswerType.RecordSet:
         content = {
@@ -85,26 +151,70 @@
            values: res.summary,
         };
         break;
      case AnswerType.Url:
         content = {
            type: AnswerType.Url,
            values: res.url,
         };
         break;
      default:
         content = {
            type: AnswerType.Text,
            values: '发生错误!',
            values: '解析失败!',
         };
         break;
   }
   content.askMoreList = _.orderBy(res.context_history, [(item) => Number(item.radio)], ['desc']);
   content.errCode = res?.err_code;
   content.msg = res?.json_msg;
   return content;
};
//#region ====================== 查询进度 ======================
let processId = '';
const QUERY_PROCESS_INTERVAL = 1000;
const process = ref('');
let processTimer = null;
let finishProcess = true;
const queryProcessApi = async () => {
   const res = await getQuestionProcess({
      process_id: processId,
   }).catch((err) => {
      process.value = err;
   });
   process.value = res.process;
   finishProcess = true;
};
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: activeRoomId.value,
      history_group_id: currentRouteId,
      raw_mode: roomConfig.value?.[currentRouteId]?.isAnswerByLLM ?? false,
   } as any;
   if (currentSampleId) {
@@ -114,23 +224,12 @@
   if (currentLLMId) {
      params.llm_id = currentLLMId;
   }
   const res = await QuestionAi(params);
   // 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)的最大压力值',
   //    },
   // };
   clearQueryProcess();
   queryProcess();
   const res = await QuestionAi(params).finally(() => {
      clearQueryProcess();
   });
   questionRes = res;
   const content = parseContent(res);
   return content;
};
@@ -143,22 +242,22 @@
const scrollToBottom = () => {
   if (!chatListDom.value) return;
   chatListDom.value.lasElementChild?.scrollIntoView();
   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 (!content?.values) return;
   if (messageList.value.length === 0) {
      if (activeSampleId.value) {
         currentSampleId = activeSampleId.value;
      }
@@ -171,17 +270,23 @@
         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({ role: RoleEnum.user, content });
      messageList.value.push(userItem);
      // 清空输入框
      clearMessageContent();
      // 出现回复,置空出现等待动画
      messageList.value.push({ role: RoleEnum.assistant, content: null });
      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({
@@ -198,57 +303,56 @@
   }
};
watch(
   () => activeRoomId.value,
   async (val) => {
      if (!val) {
         router.replace({
            name: 'Home',
         });
         return;
      }
      if (activeChatRoom.value.isInitial) {
         messageContent.value = {
const askMoreClick = (item) => {
   if (!item.question) return;
   sendChatMessage({ type: AnswerType.Text, values: item.question });
};
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: 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;
         currentSampleId = res?.details?.[0]?.sample_id;
         const resList = await Promise.all((res.details ?? []).map((item) => getAnswerById(item.history_id)));
         let i = 0;
            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, {
               role: RoleEnum.assistant,
               content: parseContent(item.answer),
            });
            i++;
         });
      }
   },
   {
      immediate: true,
   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());
   },
   {
@@ -256,13 +360,52 @@
   }
);
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>