251 lines
8.4 KiB
TypeScript
251 lines
8.4 KiB
TypeScript
|
import { message } from '@/components/bs-ui/toast/use-toast';
|
|||
|
|
import { generateUUID } from '@/components/bs-ui/utils'
|
|||
|
|
import { MessageDB, getChatHistory } from '@/controllers/API'
|
|||
|
|
import { ChatMessageType } from '@/types/chat'
|
|||
|
|
import { cloneDeep } from 'lodash'
|
|||
|
|
import { create } from 'zustand'
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 会话消息管理
|
|||
|
|
*/
|
|||
|
|
|
|||
|
|
type State = {
|
|||
|
|
running: boolean,
|
|||
|
|
/**
|
|||
|
|
* 会话 ID
|
|||
|
|
* 变更会触发 ws建立，解锁滚动
|
|||
|
|
*/
|
|||
|
|
chatId: string,
|
|||
|
|
/** 没有更多历史纪录 */
|
|||
|
|
historyEnd: boolean,
|
|||
|
|
messages: ChatMessageType[]
|
|||
|
|
/**
|
|||
|
|
* 控制引导问题的显示状态
|
|||
|
|
*/
|
|||
|
|
showGuideQuestion: boolean
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type Actions = {
|
|||
|
|
loadHistoryMsg: (flowid: string, chatId: string) => Promise<void>;
|
|||
|
|
loadMoreHistoryMsg: (flowid: string) => Promise<void>;
|
|||
|
|
destory: () => void;
|
|||
|
|
createSendMsg: (inputs: any, inputKey?: string) => void;
|
|||
|
|
createWsMsg: (data: any) => void;
|
|||
|
|
updateCurrentMessage: (wsdata: any, cover: boolean) => void;
|
|||
|
|
changeChatId: (chatId: string) => void;
|
|||
|
|
startNewRound: () => void;
|
|||
|
|
insetSeparator: (text: string) => void;
|
|||
|
|
insetSystemMsg: (text: string) => void;
|
|||
|
|
insetBsMsg: (text: string) => void;
|
|||
|
|
setShowGuideQuestion: (text: boolean) => void;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
const handleHistoryMsg = (data: any[]): ChatMessageType[] => {
|
|||
|
|
const correctedJsonString = (str: string) => str
|
|||
|
|
// .replace(/\\([\s\S])|(`)/g, '\\\\$1$2') // 转义反斜线和反引号
|
|||
|
|
.replace(/\n/g, '\\n') // 转义换行符
|
|||
|
|
.replace(/\r/g, '\\r') // 转义回车符
|
|||
|
|
.replace(/\t/g, '\\t') // 转义制表符
|
|||
|
|
.replace(/'/g, '"'); // 将单引号替换为双引号
|
|||
|
|
return data.map(item => {
|
|||
|
|
// let count = 0
|
|||
|
|
let { message, files, is_bot, intermediate_steps, ...other } = item
|
|||
|
|
try {
|
|||
|
|
message = message && message[0] === '{' ? JSON.parse(message) : message || ''
|
|||
|
|
} catch (e) {
|
|||
|
|
// 未考虑的情况暂不处理
|
|||
|
|
console.error('消息 to JSON error :>> ', e);
|
|||
|
|
}
|
|||
|
|
return {
|
|||
|
|
...other,
|
|||
|
|
chatKey: typeof message === 'string' ? undefined : Object.keys(message)[0],
|
|||
|
|
end: true,
|
|||
|
|
files: files ? JSON.parse(files) : [],
|
|||
|
|
isSend: !is_bot,
|
|||
|
|
message,
|
|||
|
|
thought: intermediate_steps,
|
|||
|
|
noAccess: true
|
|||
|
|
}
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
let currentChatId = ''
|
|||
|
|
const runLogsTypes = ['tool', 'flow', 'knowledge']
|
|||
|
|
export const useMessageStore = create<State & Actions>((set, get) => ({
|
|||
|
|
running: false,
|
|||
|
|
chatId: '',
|
|||
|
|
messages: [],
|
|||
|
|
historyEnd: false,
|
|||
|
|
showGuideQuestion: false,
|
|||
|
|
setShowGuideQuestion(bln: boolean) {
|
|||
|
|
set({ showGuideQuestion: bln })
|
|||
|
|
},
|
|||
|
|
async loadHistoryMsg(flowid, chatId) {
|
|||
|
|
const res = await getChatHistory(flowid, chatId, 30, 0)
|
|||
|
|
const msgs = handleHistoryMsg(res)
|
|||
|
|
currentChatId = chatId
|
|||
|
|
set({ historyEnd: false, messages: msgs.reverse() })
|
|||
|
|
},
|
|||
|
|
async loadMoreHistoryMsg(flowid) {
|
|||
|
|
if (get().running) return // 会话进行中禁止加载more历史
|
|||
|
|
if (get().historyEnd) return // 没有更多历史纪录
|
|||
|
|
const chatId = get().chatId
|
|||
|
|
const prevMsgs = get().messages
|
|||
|
|
// 最后一条消息id不存在，忽略 loadmore
|
|||
|
|
if (!prevMsgs[0]?.id) return
|
|||
|
|
const res = await getChatHistory(flowid, chatId, 10, prevMsgs[0]?.id || 0)
|
|||
|
|
// 过滤非同一会话消息
|
|||
|
|
if (res[0]?.chat_id !== currentChatId) {
|
|||
|
|
return console.warn('loadMoreHistoryMsg chatId not match, ignore')
|
|||
|
|
}
|
|||
|
|
const msgs = handleHistoryMsg(res)
|
|||
|
|
if (msgs.length) {
|
|||
|
|
set({ messages: [...msgs.reverse(), ...prevMsgs] })
|
|||
|
|
} else {
|
|||
|
|
set({ historyEnd: true })
|
|||
|
|
}
|
|||
|
|
},
|
|||
|
|
destory() {
|
|||
|
|
set({ chatId: '', messages: [] })
|
|||
|
|
},
|
|||
|
|
createSendMsg(inputs, inputKey) {
|
|||
|
|
console.log('change createSendMsg', inputs, inputKey);
|
|||
|
|
|
|||
|
|
set((state) => ({
|
|||
|
|
messages:
|
|||
|
|
[...state.messages, {
|
|||
|
|
isSend: true,
|
|||
|
|
message: inputs,
|
|||
|
|
chatKey: inputKey,
|
|||
|
|
thought: '',
|
|||
|
|
category: '',
|
|||
|
|
files: [],
|
|||
|
|
end: false,
|
|||
|
|
user_name: ""
|
|||
|
|
}]
|
|||
|
|
}))
|
|||
|
|
},
|
|||
|
|
// start
|
|||
|
|
createWsMsg(data) {
|
|||
|
|
console.log('change createWsMsg');
|
|||
|
|
set((state) => {
|
|||
|
|
let newChat = cloneDeep(state.messages);
|
|||
|
|
newChat.push({
|
|||
|
|
isSend: false,
|
|||
|
|
message: runLogsTypes.includes(data.category) ? JSON.parse(data.message) : '',
|
|||
|
|
chatKey: '',
|
|||
|
|
thought: data.intermediate_steps || '',
|
|||
|
|
category: data.category || '',
|
|||
|
|
files: [],
|
|||
|
|
end: false,
|
|||
|
|
user_name: '',
|
|||
|
|
extra: data.extra
|
|||
|
|
})
|
|||
|
|
return { messages: newChat }
|
|||
|
|
})
|
|||
|
|
},
|
|||
|
|
// stream end
|
|||
|
|
updateCurrentMessage(wsdata, cover = false) {
|
|||
|
|
// console.log( wsdata.chat_id, get().chatId);
|
|||
|
|
// if (wsdata.end) {
|
|||
|
|
// debugger
|
|||
|
|
// }
|
|||
|
|
console.log('change updateCurrentMessage');
|
|||
|
|
const messages = get().messages
|
|||
|
|
const isRunLog = runLogsTypes.includes(wsdata.category);
|
|||
|
|
// run log类型存在嵌套情况，使用 extra 匹配 currentMessage; 否则取最近
|
|||
|
|
const currentMessageIndex = isRunLog ?
|
|||
|
|
messages.findLastIndex((msg) => msg.extra === wsdata.extra)
|
|||
|
|
: messages.findLastIndex((msg) => !runLogsTypes.includes(msg.category))
|
|||
|
|
const currentMessage = messages[currentMessageIndex]
|
|||
|
|
|
|||
|
|
const newCurrentMessage = {
|
|||
|
|
...currentMessage,
|
|||
|
|
...wsdata,
|
|||
|
|
id: isRunLog ? wsdata.extra : wsdata.messageId, // 每条消息必唯一
|
|||
|
|
message: isRunLog ? JSON.parse(wsdata.message) : currentMessage.message + wsdata.message,
|
|||
|
|
thought: currentMessage.thought + (wsdata.thought ? `${wsdata.thought}\n` : ''),
|
|||
|
|
files: wsdata.files || null,
|
|||
|
|
category: wsdata.category || '',
|
|||
|
|
source: wsdata.source
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
messages[currentMessageIndex] = newCurrentMessage
|
|||
|
|
// 会话特殊处理，兼容后端的缺陷
|
|||
|
|
if (!isRunLog) {
|
|||
|
|
// start - end 之间没有内容删除load
|
|||
|
|
if (newCurrentMessage.end && !(newCurrentMessage.files.length || newCurrentMessage.thought || newCurrentMessage.message)) {
|
|||
|
|
messages.pop()
|
|||
|
|
}
|
|||
|
|
// 无 messageid 删除
|
|||
|
|
// if (newCurrentMessage.end && !newCurrentMessage.id) {
|
|||
|
|
// messages.pop()
|
|||
|
|
// }
|
|||
|
|
// 删除重复消息
|
|||
|
|
const prevMessage = messages[currentMessageIndex - 1];
|
|||
|
|
if ((prevMessage
|
|||
|
|
&& prevMessage.message === newCurrentMessage.message
|
|||
|
|
&& prevMessage.thought === newCurrentMessage.thought)
|
|||
|
|
|| cover) {
|
|||
|
|
const removedMsg = messages.pop()
|
|||
|
|
// 使用最后一条的信息作为准确信息
|
|||
|
|
Object.keys(prevMessage).forEach((key) => {
|
|||
|
|
prevMessage[key] = removedMsg[key]
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
set((state) => ({ messages: [...messages] }))
|
|||
|
|
},
|
|||
|
|
changeChatId(chatId) {
|
|||
|
|
set((state) => ({ chatId }))
|
|||
|
|
},
|
|||
|
|
startNewRound() {
|
|||
|
|
get().insetSeparator('配置已更新')
|
|||
|
|
set((state) => ({ showGuideQuestion: true }))
|
|||
|
|
},
|
|||
|
|
insetSeparator(text) {
|
|||
|
|
set((state) => ({
|
|||
|
|
messages: [...state.messages, {
|
|||
|
|
...bsMsgItem,
|
|||
|
|
id: Math.random() * 1000000,
|
|||
|
|
category: 'divider',
|
|||
|
|
message: text,
|
|||
|
|
}]
|
|||
|
|
}))
|
|||
|
|
},
|
|||
|
|
insetSystemMsg(text) {
|
|||
|
|
set((state) => ({
|
|||
|
|
messages: [...state.messages, {
|
|||
|
|
...bsMsgItem,
|
|||
|
|
id: Math.random() * 1000000,
|
|||
|
|
category: 'guide',
|
|||
|
|
thought: text,
|
|||
|
|
}]
|
|||
|
|
}))
|
|||
|
|
},
|
|||
|
|
insetBsMsg(text) {
|
|||
|
|
set((state) => ({
|
|||
|
|
messages: [...state.messages, {
|
|||
|
|
...bsMsgItem,
|
|||
|
|
id: 0,
|
|||
|
|
category: 'guide',
|
|||
|
|
thought: '',
|
|||
|
|
message: text
|
|||
|
|
}]
|
|||
|
|
}))
|
|||
|
|
}
|
|||
|
|
}))
|
|||
|
|
|
|||
|
|
|
|||
|
|
const bsMsgItem = {
|
|||
|
|
id: Math.random() * 1000000,
|
|||
|
|
isSend: false,
|
|||
|
|
message: '',
|
|||
|
|
chatKey: '',
|
|||
|
|
thought: '',
|
|||
|
|
category: '',
|
|||
|
|
files: [],
|
|||
|
|
end: true,
|
|||
|
|
user_name: ''
|
|||
|
|
}
|