kevisual/code-center
1
0
Fork 0
Code Issues 1 Pull Requests Actions Packages Projects Releases Wiki Activity

fix: 修改错误

Browse Source
This commit is contained in:
xion 2024-10-02 14:13:00 +08:00
parent 8436b0462a
commit 1f81d3400c
1 changed files with 31 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

45
src/routes/chat-history/chat-io.ts
View File

@ -54,29 +54,46 @@ const onMessage = async ({ data, end, ws }) => {
return;
}
const { uid, id, key } = client.data;
const { inputs, message: sendMessage } = data;
const {
inputs,
message: sendMessage,
data: {},
} = data;
let root = data.root || false;
let chatSession = await ChatSession.findByPk(id);
const config = await getConfigByKey(key);
const { prompt, aiAgent, chatPrompt } = config;
let userQuestion = sendMessage;
if (!chatSession) {
chatSession = await ChatSession.create({ key, id, data: {}, uid, chatPromptId: chatPrompt.id });
chatSession = await ChatSession.create({ key, id, data: data, uid, chatPromptId: chatPrompt.id });
root = true;
} else {
// 更新session context的值
const newData = JSON.parse(data);
if (newData !== '{}' && JSON.stringify(chatSession.data) !== JSON.stringify(data)) {
await chatSession.update({ data: data });
}
if (root) {
const chatHistory = await ChatHistory.findAll({ where: { chatId: id }, logging: false });
chatHistory.forEach((item) => {
end({ code: 200, data: item, message: 'success', type: 'messages' });
});
// return;
}
root = false;
const chatHistory = await ChatHistory.findAll({ where: { chatId: id }, logging: false });
chatHistory.forEach((item) => {
end({ code: 200, data: item, message: 'success', type: 'messages' });
});
return;
}
const template = await getTemplate({ data: prompt.presetData.data, inputs });
if (!template) {
end({ code: 404, data: {}, message: 'template not found' });
return;
if (!userQuestion) {
if (!prompt?.presetData) {
end({ code: 404, data: {}, message: 'prompt not set, need presetData' });
return;
}
const template = await getTemplate({ data: prompt.presetData.data, inputs });
if (!template) {
end({ code: 404, data: {}, message: 'template not found' });
return;
}
userQuestion = template;
}
const userQuestion = template || sendMessage;
// 保存到数据库
const roleUser = await ChatHistory.create({
data: {
@ -90,7 +107,7 @@ const onMessage = async ({ data, end, ws }) => {
role: 'user',
});
end({ code: 200, data: roleUser, message: 'success', type: 'messages' });
const result = await aiAgent.sendHumanMessage(template, { thread_id: id });
const result = await aiAgent.sendHumanMessage(userQuestion, { thread_id: id });
const lastMessage = result.messages[result.messages.length - 1];
const message = result.messages[result.messages.length - 1].content;
// 根据key找到对应的prompt