Andreas/LibreChat
1
0
Fork 0
mirror of https://github.com/danny-avila/LibreChat.git synced 2026-01-16 23:45:33 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

adds server and client error handling, clear convos in progress

Browse source
This commit is contained in:
Daniel Avila 2023-02-08 00:02:29 -05:00
parent 511ac948b4
commit 51d93da3f8
9 changed files with 98 additions and 48 deletions
Download patch file Download diff file Expand all files Collapse all files

44
server/index.js
View file

@ -52,6 +52,8 @@ app.post('/ask', async (req, res) => {
const userMessageId = crypto.randomUUID();
let userMessage = { id: userMessageId, sender: 'User', text };
console.log(userMessage, req.body);
res.writeHead(200, {
Connection: 'keep-alive',
'Content-Type': 'text/event-stream',
@ -61,22 +63,23 @@ app.post('/ask', async (req, res) => {
});
// res.write(`event: message\ndata: ${JSON.stringify('')}\n\n`);
let i = 0;
const progressCallback = async (partial) => {
// console.log('partial', partial);
if (i === 0) {
userMessage.parentMessageId = parentMessageId ? parentMessageId : partial.id;
userMessage.conversationId = conversationId ? conversationId : partial.conversationId;
await saveMessage(userMessage);
res.write(`event: message\ndata: ${JSON.stringify({ ...partial, initial: true })}\n\n`);
i++;
}
const data = JSON.stringify({ ...partial, message: true });
res.write(`event: message\ndata: ${data}\n\n`);
};
try {
let i = 0;
const progressCallback = async (partial) => {
// console.log('partial', partial);
if (i === 0) {
userMessage.parentMessageId = parentMessageId ? parentMessageId : partial.id;
userMessage.conversationId = conversationId ? conversationId : partial.conversationId;
await saveMessage(userMessage);
res.write(
`event: message\ndata: ${JSON.stringify({ ...partial, initial: true })}\n\n`
);
i++;
}
const data = JSON.stringify({ ...partial, message: true });
res.write(`event: message\ndata: ${data}\n\n`);
};
let gptResponse = await ask(text, progressCallback, { parentMessageId, conversationId });
if (!!parentMessageId) {
gptResponse = { ...gptResponse, parentMessageId };
@ -84,6 +87,12 @@ app.post('/ask', async (req, res) => {
gptResponse.title = await titleConversation(text, gptResponse.text);
}
if (gptResponse.text.includes('2023')) {
res.status(500).write('event: error\ndata: empty string error?');
res.end();
return;
}
gptResponse.sender = 'GPT';
await saveMessage(gptResponse);
await saveConvo(gptResponse);
@ -91,8 +100,9 @@ app.post('/ask', async (req, res) => {
res.write(`event: message\ndata: ${JSON.stringify(gptResponse)}\n\n`);
res.end();
} catch (error) {
console.error(error);
res.status(500).send(error);
console.log(error);
res.status(500).write('event: error\ndata: ' + error.message);
res.end();
}
});