Andreas/LibreChat
1
0
Fork 0
mirror of https://github.com/danny-avila/LibreChat.git synced 2025-12-16 16:30:15 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 6

Update Message.js (#191)

Browse source 
Fixed Error handling, Code duplication and Naming conventions. Contact me for more information at: DavidTheDev#0166

Co-authored-by: Danny Avila <110412045+danny-avila@users.noreply.github.com>
This commit is contained in:
David 2023-05-09 23:51:39 +02:00 • committed by GitHub
parent 5dd9c11326
commit 4b94af0429
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 37 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

56
api/models/Message.js
View file

@ -1,7 +1,9 @@
const Message = require('./schema/messageSchema');
module.exports = {
Message,
saveMessage: async ({
async saveMessage({
messageId,
newMessageId,
conversationId,
@ -12,7 +14,7 @@ module.exports = {
error,
unfinished,
cancelled
}) => {
}) {
try {
// may also need to update the conversation here
await Message.findOneAndUpdate(
@ -30,39 +32,55 @@ module.exports = {
},
{ upsert: true, new: true }
);
return { messageId, conversationId, parentMessageId, sender, text, isCreatedByUser };
} catch (error) {
console.error(error);
return { message: 'Error saving message' };
return {
messageId,
conversationId,
parentMessageId,
sender,
text,
isCreatedByUser
};
} catch (err) {
console.error(`Error saving message: ${err}`);
throw new Error('Failed to save message.');
}
},
deleteMessagesSince: async ({ messageId, conversationId }) => {
async deleteMessagesSince({ messageId, conversationId }) {
try {
const message = await Message.findOne({ messageId }).exec();
if (message)
if (message) {
return await Message.find({ conversationId })
.deleteMany({ createdAt: { $gt: message.createdAt } })
.exec();
} catch (error) {
console.error(error);
return { message: 'Error deleting messages' };
}
} catch (err) {
console.error(`Error deleting messages: ${err}`);
throw new Error('Failed to delete messages.');
}
},
getMessages: async (filter) => {
async getMessages(filter) {
try {
return await Message.find(filter).sort({ createdAt: 1 }).exec();
} catch (error) {
console.error(error);
return { message: 'Error getting messages' };
} catch (err) {
console.error(`Error getting messages: ${err}`);
throw new Error('Failed to get messages.');
}
},
deleteMessages: async (filter) => {
async deleteMessages(filter) {
try {
return await Message.deleteMany(filter).exec();
} catch (error) {
console.error(error);
return { message: 'Error deleting messages' };
} catch (err) {
console.error(`Error deleting messages: ${err}`);
throw new Error('Failed to delete messages.');
}
}
};