<template>
|
<ChatContainer
|
:loading="chatListLoading"
|
:more-is-loading="moreIsLoading"
|
:is-share-page="isSharePage"
|
:chat-width="chatWidth"
|
ref="containerRef"
|
>
|
<!-- 消息列表 -->
|
<template #message-list>
|
<MessageList
|
v-if="computedMessageList?.length > 0"
|
:msgList="computedMessageList"
|
:isTalking="isTalking"
|
@shareClick="shareClick"
|
@setCommonQuestionClick="setCommonPhraseClick"
|
@sendChatMessage="sendChatMessage"
|
@askMoreClick="askMoreClick"
|
/>
|
<el-empty v-else-if="isSharePage && !chatListLoading" :image-size="200">
|
<template #description>
|
<span class="text-[15px]">分享的对话不存在或已失效</span>
|
</template>
|
</el-empty>
|
</template>
|
|
<!-- 输入区域 -->
|
<template #input-area>
|
<PlayBar
|
ref="playBarRef"
|
v-model:voicePageIsShow="voicePageIsShow"
|
:isTalking="isTalking"
|
:isHome="false"
|
v-model="messageContent.values"
|
@sendClick="sendClick"
|
@showUpChatClick="showUpChatClick"
|
@stopGenClick="stopGenClick"
|
@showDownChatClick="showDownChatClick"
|
:style="{ width: chatWidth }"
|
/>
|
</template>
|
|
<!-- 抽屉 -->
|
<template #drawer>
|
<CustomDrawer v-model:isShow="drawerIsShow" @updateChatInput="updateChatInput" />
|
<ShareLinkDlg ref="shareLinkDlgRef" />
|
</template>
|
</ChatContainer>
|
</template>
|
|
<script setup lang="ts">
|
import type { CancelTokenSource } from 'axios';
|
import axios from 'axios';
|
import { orderBy } from 'lodash-es';
|
import moment from 'moment';
|
import { computed, onActivated, onMounted, ref } from 'vue';
|
import { loadAmisSource } from '../amis/load';
|
import { convertProcessItem, convertProcessToStep, formatShowTimeYear, useScrollLoad } from './hooks/useScrollLoad';
|
import type { ChatContent } from './model/types';
|
import { AnswerState, AnswerType, RoleEnum, type ChatMessage } from './model/types';
|
import { getShareChatJsonByPost, questionStreamByPost } from '/@/api/ai/chat';
|
import PlayBar from '/@/components/chat/components/playBar/PlayBar.vue';
|
import CustomDrawer from '/@/components/drawer/CustomDrawer.vue';
|
import { Logger } from '/@/model/logger/Logger';
|
|
import { ElMessage } from 'element-plus';
|
import ChatContainer from './components/ChatContainer.vue';
|
import ShareLinkDlg from './components/shareLink/index.vue';
|
import router from '/@/router';
|
import MessageList from './messageList/index.vue';
|
import {
|
activeChatRoom,
|
activeGroupType,
|
activeLLMId,
|
activeRoomId,
|
activeSampleId,
|
isSharePage,
|
roomConfig,
|
} from '/@/stores/chatRoom';
|
import emitter from '/@/utils/mitt';
|
import { useCompRef } from '/@/utils/types';
|
import { toMyFixed } from '/@/utils/util';
|
const containerRef = useCompRef(ChatContainer);
|
const chatListDom = computed(() => containerRef.value?.chatListDom);
|
const chatWidth = '75%';
|
const voicePageIsShow = ref(false);
|
let isTalking = ref(false);
|
|
let messageContent = ref<ChatContent>({
|
type: AnswerType.Text,
|
values: '',
|
});
|
const currentRoute = router.currentRoute;
|
const currentRouteId = currentRoute.value.query.id as string;
|
activeRoomId.value = currentRouteId;
|
const messageList = ref<ChatMessage[]>([]);
|
const computedMessageList = computed(() => {
|
return messageList.value.filter((v) => !!v);
|
});
|
|
const parseExtraContent = (res) => {
|
if (!res) return {};
|
const askMoreList = orderBy(res.context_history, [(item) => Number(item.radio)], ['desc']);
|
const errCode = res?.err_code;
|
const errMsg = res?.json_msg;
|
const origin = res;
|
|
return {
|
askMoreList,
|
errCode,
|
errMsg,
|
origin,
|
};
|
};
|
|
const parseContent = (res, reportIsShow = false, extraContent?) => {
|
if (!res) return null;
|
let content: ChatContent = {
|
type: AnswerType.Text,
|
values: '解析失败！',
|
};
|
if (res.type) {
|
res.answer_type = res.type;
|
}
|
const curExtraContent = parseExtraContent(res);
|
|
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.Script:
|
content = {
|
type: AnswerType.Script,
|
values: res,
|
};
|
break;
|
|
case AnswerType.Knowledge:
|
content = {
|
type: AnswerType.Knowledge,
|
values: res.knowledge,
|
};
|
|
break;
|
case AnswerType.Report:
|
content = {
|
type: AnswerType.Report,
|
values: (res?.reports ?? []).map((item) => ({
|
content: parseContent(item, reportIsShow, { origin: item, conclusion: item.conclusion ?? [] }),
|
})),
|
};
|
break;
|
|
case AnswerType.Summary:
|
content = {
|
type: AnswerType.Summary,
|
values: res.summary?.map((item) => {
|
item.reportIsShow = reportIsShow;
|
return item;
|
}),
|
};
|
break;
|
case AnswerType.Url:
|
content = {
|
type: AnswerType.Url,
|
values: res.url,
|
};
|
break;
|
case AnswerType.Map:
|
content = {
|
type: AnswerType.Map,
|
values: res.values,
|
};
|
break;
|
default:
|
content = {
|
type: AnswerType.Text,
|
values: '解析失败！',
|
};
|
break;
|
}
|
if (!extraContent) {
|
content = {
|
...content,
|
...curExtraContent,
|
};
|
} else {
|
content = {
|
...content,
|
...extraContent,
|
};
|
}
|
|
return content;
|
};
|
|
let questionRes = null;
|
let position = null;
|
const preQuestion = ref(null);
|
|
let lastAxiosSource: CancelTokenSource = null;
|
const questionAi = async (text) => {
|
let judgeParams = null;
|
if (!preQuestion.value) {
|
judgeParams = {};
|
} else {
|
judgeParams = {
|
prev_question: preQuestion.value,
|
};
|
}
|
|
const params = {
|
question: text,
|
history_group_id: currentRouteId,
|
raw_mode: roomConfig.value?.[currentRouteId]?.isAnswerByLLM ?? false,
|
...judgeParams,
|
} as any;
|
|
if (position) {
|
const longitude = position.coords.longitude;
|
const latitude = position.coords.latitude;
|
params.cur_pos = [longitude, latitude].join(',');
|
}
|
|
if (activeGroupType.value) {
|
params.group_type = activeGroupType.value;
|
}
|
|
if (currentSampleId) {
|
params.sample_id = currentSampleId;
|
currentSampleId = '';
|
}
|
|
let lastTimestamp = new Date().getTime();
|
questionRes = {};
|
let lastIsResult = false;
|
const resultP = new Promise((resolve, reject) => {
|
const currentSource = axios.CancelToken.source();
|
lastAxiosSource = currentSource;
|
|
const getResReport = () => {
|
const resReport = {
|
answer_type: AnswerType.Report,
|
reports: [],
|
};
|
return resReport;
|
};
|
const checkReportEmpty = () => {
|
const isEmpty = !questionRes?.reports || questionRes?.reports?.length === 0;
|
|
return isEmpty;
|
};
|
questionStreamByPost(
|
params,
|
(chunkRes) => {
|
Logger.info('chunk response：\n\n' + JSON.stringify(chunkRes));
|
|
if (chunkRes.mode === 'result') {
|
lastIsResult = true;
|
const res = chunkRes.value;
|
|
if (checkReportEmpty()) {
|
const resReport = getResReport();
|
resReport.reports.push(res);
|
questionRes = resReport;
|
resolve(resReport);
|
} else {
|
const lastMsg = computedMessageList.value.at(-1);
|
|
// 已经解析过一次 reports
|
lastMsg.content.values.push({
|
content: parseContent(res, true, {
|
origin: res,
|
}),
|
});
|
}
|
return;
|
// chunkRes.value = '准备数据分析';
|
}
|
|
if (chunkRes.mode === 'summary') {
|
const lastMsg = computedMessageList.value.at(-1);
|
const extraContent = parseExtraContent(chunkRes.value);
|
const isReportEmpty = checkReportEmpty();
|
// 没有经过 result 报告还没初始化
|
if (isReportEmpty) {
|
const resReport = getResReport();
|
questionRes = resReport;
|
}
|
// 此对话已经加入到对话列表
|
if (lastMsg.content?.values && extraContent) {
|
for (const key in extraContent) {
|
if (Object.prototype.hasOwnProperty.call(extraContent, key)) {
|
const value = extraContent[key];
|
if (!lastMsg.content[key] || (Array.isArray(lastMsg.content[key]) && lastMsg.content[key].length === 0)) {
|
lastMsg.content[key] = value;
|
}
|
}
|
}
|
|
lastMsg.historyId = chunkRes.value.history_id;
|
const userMsg = computedMessageList.value.at(-2);
|
userMsg.historyId = chunkRes.value.history_id;
|
userMsg.content.values = chunkRes.value.question;
|
}
|
|
if (Object.keys(questionRes).length === 0) {
|
questionRes = chunkRes.value;
|
}
|
|
// 此对话还未加入到对话列表
|
if (!lastMsg.content?.values && questionRes) {
|
questionRes = {
|
...questionRes,
|
...chunkRes.value,
|
};
|
}
|
|
if (isReportEmpty) {
|
resolve(questionRes);
|
}
|
// computedMessageList.value[computedMessageList.value.length - 1] = finalMsg;
|
scrollToBottom();
|
// chunkRes.value = '你可以继续问我';
|
return;
|
}
|
|
if (chunkRes.mode === 'conclusion') {
|
const lastReport = computedMessageList.value.at(-1)?.content?.values?.at(-1);
|
if (lastReport) {
|
lastReport.conclusion = chunkRes.value;
|
chunkRes.value = '分析结束';
|
}
|
}
|
|
if (chunkRes.mode === 'question') {
|
const lastGroup = computedMessageList.value.at(-1).stepGroup.at(-1);
|
const stepList = lastGroup?.value ?? [];
|
const lastStepItem = stepList.at(-1);
|
if (!lastStepItem.subStep) {
|
lastStepItem.subStep = [];
|
}
|
lastStepItem.subStep.push({
|
type: chunkRes.value.type,
|
data: chunkRes.value,
|
});
|
scrollToBottom();
|
return;
|
}
|
// 暂时不考虑多个 report情况
|
|
// if (lastIsResult && chunkRes.mode !== 'finish') {
|
// // 开始增加新的 stepGroup
|
// computedMessageList.value.at(-1).stepGroup.push({
|
// value: [],
|
// isShow: true,
|
// });
|
// lastIsResult = false;
|
// }
|
const lastGroup = computedMessageList.value.at(-1).stepGroup.at(-1);
|
const stepList = lastGroup?.value ?? [];
|
const currentTimeStamp = new Date().getTime();
|
const ms = toMyFixed(currentTimeStamp - lastTimestamp, 2) + ' ms';
|
if (chunkRes.mode === 'finish') {
|
stepList.at(-1).ms = ms;
|
isTalking.value = false;
|
|
return;
|
}
|
|
if (stepList?.length >= 1) {
|
stepList.at(-1).ms = ms;
|
} else {
|
const stepGroup = computedMessageList.value.at(-1).stepGroup;
|
if (stepGroup.length > 1) {
|
const lastStepList = stepGroup.at(-2).value;
|
lastStepList.at(-1).ms = ms;
|
}
|
}
|
lastTimestamp = currentTimeStamp;
|
const stepItem = convertProcessItem(chunkRes);
|
|
stepList.push(stepItem);
|
// 强制触发更新
|
|
scrollToBottom();
|
},
|
{
|
cancelToken: currentSource.token,
|
}
|
)
|
.catch((err) => {
|
throw err;
|
})
|
.finally(() => {
|
isTalking.value = false;
|
// 收起所有 stepGroup
|
computedMessageList.value.at(-1).stepGroup.forEach((item) => {
|
item.isShow = false;
|
});
|
});
|
});
|
|
await resultP;
|
const content = parseContent(questionRes, true);
|
return content;
|
};
|
|
const clearMessageContent = () =>
|
(messageContent.value = {
|
type: AnswerType.Text,
|
values: '',
|
});
|
|
let currentSampleId = '';
|
|
let currentLLMId = null;
|
|
const stopGenClick = () => {
|
lastAxiosSource?.cancel();
|
isTalking.value = false;
|
chatListLoading.value = false;
|
|
computedMessageList.value.at(-1).isStopMsg = true;
|
};
|
|
const checkCanSend = (content: ChatContent = messageContent.value) => {
|
if (!content?.values) {
|
return false;
|
}
|
if (isTalking.value || chatListLoading.value) {
|
ElMessage.warning('ai 正在回复中，请稍后尝试提问');
|
return false;
|
}
|
return true;
|
};
|
|
const addChatItem = (content: ChatContent) => {
|
isTalking.value = true;
|
const userItem: ChatMessage = { role: RoleEnum.user, content, isChecked: false } as any;
|
const assistantItem: ChatMessage = {
|
role: RoleEnum.assistant,
|
content: {
|
type: AnswerType.Report,
|
},
|
state: AnswerState.Null,
|
stepGroup: [
|
{
|
value: [],
|
isShow: true,
|
},
|
],
|
isStopMsg: false,
|
isChecked: false,
|
} as any;
|
messageList.value.push(userItem);
|
clearMessageContent();
|
|
messageList.value.push(assistantItem);
|
scrollToBottom();
|
return [userItem, assistantItem];
|
};
|
|
const sendChatMessage = async (content: ChatContent = messageContent.value) => {
|
if (!checkCanSend(content)) {
|
return;
|
}
|
const isNewChat = messageList.value.length === 0;
|
if (isNewChat) {
|
if (activeSampleId.value) {
|
currentSampleId = activeSampleId.value;
|
}
|
|
if (activeLLMId.value) {
|
currentLLMId = activeLLMId.value;
|
}
|
}
|
let resMsgContent: ChatContent = null;
|
|
try {
|
const [userItem, assistantItem] = addChatItem(content);
|
|
resMsgContent = await questionAi(content.values);
|
|
updateLoadIndex();
|
|
userItem.historyId = questionRes?.history_id;
|
userItem.content.values = questionRes?.question ?? userItem.content.values;
|
assistantItem.historyId = questionRes?.history_id;
|
const currentTime = formatShowTimeYear(moment().format('YYYY-MM-DD HH:mm:ss'));
|
assistantItem.createTime = currentTime;
|
assistantItem.content = resMsgContent;
|
setTimeout(() => {
|
// 收到回复，继续滚
|
scrollToBottom();
|
}, 300);
|
} catch (error: any) {}
|
};
|
|
const sendClick = () => {
|
sendChatMessage(messageContent.value);
|
};
|
|
const { loadRangeData, onChatListScroll, moreIsLoading, updateLoadIndex } = useScrollLoad({
|
container: chatListDom,
|
historyGroupId: currentRouteId,
|
messageList,
|
parseAnswerContent: parseContent,
|
});
|
|
const chatListLoading = ref(true);
|
|
onActivated(() => {
|
emitter.emit('updateHeaderTitle', activeChatRoom.value?.title ?? '');
|
});
|
|
const initNewChat = () => {
|
messageContent.value = {
|
type: AnswerType.Text,
|
values: activeChatRoom.value?.title,
|
};
|
sendChatMessage();
|
};
|
const scrollToBottom = () => {
|
containerRef.value?.scrollToBottom();
|
};
|
|
const initHistoryChat = () => {
|
// 初始状态滚一下
|
scrollToBottom();
|
|
setTimeout(() => {
|
chatListDom.value.addEventListener('scroll', onChatListScroll);
|
}, 300);
|
};
|
|
/**
|
* 加载分享数据
|
*/
|
const loadShareData = async () => {
|
const res = await getShareChatJsonByPost({
|
share_id: router.currentRoute.value.query.id as string,
|
});
|
|
const msgValue = res?.values;
|
if (!msgValue) {
|
messageList.value = [];
|
return;
|
}
|
const userMsg: ChatMessage = {
|
historyId: msgValue.history_id,
|
role: RoleEnum.user,
|
content: {
|
type: AnswerType.Text,
|
values: msgValue.question,
|
},
|
isChecked: false,
|
};
|
|
const assistantMsg: ChatMessage = {
|
historyId: msgValue.history_id,
|
role: RoleEnum.assistant,
|
content: parseContent(msgValue),
|
stepGroup: (msgValue?.reports ?? []).map((item) => ({
|
value: convertProcessToStep(item?.exec_process),
|
isShow: false,
|
})),
|
isStopMsg: false,
|
|
conclusion: msgValue.conclusion ?? [],
|
isChecked: false,
|
};
|
messageList.value = [userMsg, assistantMsg];
|
};
|
|
onMounted(async () => {
|
messageList.value = [];
|
chatListLoading.value = true;
|
if (isSharePage.value) {
|
await loadShareData().finally(() => {
|
chatListLoading.value = false;
|
});
|
} else {
|
await loadRangeData().finally(() => {
|
chatListLoading.value = false;
|
});
|
}
|
setTimeout(() => {
|
emitter.emit('updateHeaderTitle', activeChatRoom.value?.title ?? '');
|
}, 300);
|
|
if (messageList.value.length === 0) {
|
initNewChat();
|
} else {
|
if (!isSharePage.value) {
|
setTimeout(() => {
|
initHistoryChat();
|
}, 300);
|
}
|
}
|
loadAmisSource();
|
});
|
|
//#region ====================== 光标输入上下箭头显示历史消息 ======================
|
const currentIndex = ref(null);
|
const history_data = computed(() => {
|
return computedMessageList.value.filter((item) => item.role === RoleEnum.user);
|
});
|
//显示上一条消息
|
const showUpChatClick = () => {
|
if (computedMessageList.value.length === 0) return;
|
if (currentIndex.value == 0) {
|
messageContent.value.values = history_data.value[currentIndex.value].content.values;
|
return;
|
} else {
|
currentIndex.value = (currentIndex.value + history_data.value.length - 1) % history_data.value.length;
|
}
|
messageContent.value.values = history_data.value[currentIndex.value].content.values;
|
};
|
//显示下一条消息
|
const showDownChatClick = () => {
|
if (computedMessageList.value.length === 0) return;
|
if (currentIndex.value == history_data.value.length - 1) {
|
messageContent.value.values = history_data.value[currentIndex.value].content.values;
|
return;
|
}
|
if (currentIndex.value === null) {
|
currentIndex.value = 0;
|
} else {
|
currentIndex.value = (currentIndex.value + 1) % history_data.value.length;
|
}
|
messageContent.value.values = history_data.value[currentIndex.value].content.values;
|
};
|
//#endregion
|
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;
|
const result = isShow && !isSharePage.value;
|
return result;
|
});
|
const askMoreClick = (item) => {
|
if (!item.question) return;
|
sendChatMessage({ type: AnswerType.Text, values: item.question });
|
};
|
|
//#region ====================== 侧边栏drawer ======================
|
const drawerIsShow = ref(false);
|
|
const updateChatInput = (content) => {
|
messageContent.value.values = content;
|
};
|
//#endregion
|
const playBarRef = useCompRef(PlayBar);
|
//用户问题设置为常用语
|
const setCommonPhraseClick = (item) => {
|
playBarRef.value.addPhrase(item);
|
};
|
|
//#region ====================== 分享 ======================
|
|
const shareLinkDlgRef = useCompRef(ShareLinkDlg);
|
|
const shareClick = async (item: ChatMessage) => {
|
shareLinkDlgRef.value.openShare(item);
|
};
|
//#endregion
|
</script>
|
|
<style scoped lang="scss">
|
pre {
|
font-family: -apple-system, 'Noto Sans', 'Helvetica Neue', Helvetica, 'Nimbus Sans L', Arial, 'Liberation Sans', 'PingFang SC',
|
'Hiragino Sans GB', 'Noto Sans CJK SC', 'Source Han Sans SC', 'Source Han Sans CN', 'Microsoft YaHei', 'Wenquanyi Micro Hei',
|
'WenQuanYi Zen Hei', 'ST Heiti', SimHei, 'WenQuanYi Zen Hei Sharp', sans-serif;
|
}
|
|
.more-loading {
|
:deep(.el-loading-spinner) {
|
--loading-size: 35px;
|
margin-top: 0;
|
.circular {
|
width: var(--loading-size);
|
height: var(--loading-size);
|
}
|
}
|
}
|
|
:deep(.el-step__icon.is-text) {
|
--radius-size: 24px;
|
width: var(--radius-size);
|
height: var((--radius-size));
|
}
|
|
:deep(.el-step__icon-inner) {
|
font-size: 16px !important;
|
}
|
:deep(.el-step__description) {
|
min-height: 20px;
|
}
|
|
:deep(.el-step:last-of-type .el-step__description) {
|
// display: none;
|
}
|
</style>
|
