Andreas/LibreChat
1
0
Fork 0
mirror of https://github.com/danny-avila/LibreChat.git synced 2026-02-01 07:11:50 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: save error message into database.

Browse source
This commit is contained in:
Wentao Lyu 2023-03-14 03:38:47 +08:00
parent d9e5464b3b
commit 0fa19bb6ad
6 changed files with 37 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

11
api/models/Message.js
View file

@ -36,20 +36,25 @@ const messageSchema = mongoose.Schema({
type: Boolean,
required: true,
default: false
}
},
error: {
type: Boolean,
default: false
},
}, { timestamps: true });
const Message = mongoose.models.Message || mongoose.model('Message', messageSchema);
module.exports = {
saveMessage: async ({ messageId, conversationId, parentMessageId, sender, text, isCreatedByUser=false }) => {
saveMessage: async ({ messageId, conversationId, parentMessageId, sender, text, isCreatedByUser=false, error }) => {
try {
await Message.findOneAndUpdate({ messageId }, {
conversationId,
parentMessageId,
sender,
text,
isCreatedByUser
isCreatedByUser,
error
}, { upsert: true, new: true });
return { messageId, conversationId, parentMessageId, sender, text, isCreatedByUser };
} catch (error) {

8
api/server/routes/ask.js
View file

@ -131,6 +131,10 @@ router.post('/', async (req, res) => {
gptResponse.text.toLowerCase().includes('no response') ||
gptResponse.text.toLowerCase().includes('no answer')
) {
await saveMessage({
messageId: crypto.randomUUID(), sender: model,
conversationId, parentMessageId: userMessageId,
error: true, text: 'Prompt empty or too short'});
return handleError(res, 'Prompt empty or too short');
}
@ -165,6 +169,10 @@ router.post('/', async (req, res) => {
} catch (error) {
console.log(error);
// await deleteMessages({ messageId: userMessageId });
await saveMessage({
messageId: crypto.randomUUID(), sender: model,
conversationId, parentMessageId: userMessageId,
error: true, text: error.message});
handleError(res, error.message);
}
});

4
api/server/routes/askBing.js
View file

@ -107,6 +107,10 @@ router.post('/', async (req, res) => {
} catch (error) {
console.log(error);
// await deleteMessages({ messageId: userMessageId });
await saveMessage({
messageId: crypto.randomUUID(), sender: model,
conversationId, parentMessageId: userMessageId,
error: true, text: error.message});
handleError(res, error.message);
}
});

4
api/server/routes/askSydney.js
View file

@ -117,6 +117,10 @@ router.post('/', async (req, res) => {
} catch (error) {
console.log(error);
// await deleteMessages({ messageId: userMessageId });
await saveMessage({
messageId: crypto.randomUUID(), sender: model,
conversationId, parentMessageId: userMessageId,
error: true, text: error.message});
handleError(res, error.message);
}
});