wujingjing
2024-07-17 acaf8207447eb243e1dc8bec3aaca447cc4d082f
src/components/chat/Chat.vue
@@ -1,52 +1,93 @@
<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 mr-4"
                  :class="{ 'mr-4': item.role === RoleEnum.assistant, 'ml-4': item.role === RoleEnum.user }"
                  :src="roleImageMap[item.role]"
                  alt=""
                  srcset=""
               />
               <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 class="flex" :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' }"
                     >
                        <component :is="answerTypeMapCom[item.content.type]" :data="item.content.values" :error="item.content.error" />
                     </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 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>
                  <Loding v-else />
                  <Loding v-else :process="process" />
               </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';
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(() => {
@@ -83,6 +124,13 @@
         content = {
            type: AnswerType.Summary,
            values: res.summary,
            error: res.json_msg,
         };
         break;
      case AnswerType.Url:
         content = {
            type: AnswerType.Url,
            values: res.url,
         };
         break;
      default:
@@ -94,17 +142,52 @@
   }
   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 +197,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;
};
@@ -149,16 +221,16 @@
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 (messageList.value.length === 0) {
      if (activeSampleId.value) {
         currentSampleId = activeSampleId.value;
      }
@@ -171,17 +243,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 +276,51 @@
   }
};
watch(
   () => activeRoomId.value,
   async (val) => {
      if (!val) {
         router.replace({
            name: 'Home',
         });
         return;
      }
      if (activeChatRoom.value.isInitial) {
         messageContent.value = {
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 +328,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>