LibreChat/models/Conversation.js
2023-02-11 11:37:20 -05:00

66 lines
1.7 KiB
JavaScript

const mongoose = require('mongoose');
const { getMessages, deleteMessages } = require('./Message');
const convoSchema = mongoose.Schema({
conversationId: {
type: String,
unique: true,
required: true
},
parentMessageId: {
type: String,
required: true
},
title: {
type: String,
default: 'New conversation',
},
messages: [{ type: mongoose.Schema.Types.ObjectId, ref: 'Message' }],
created: {
type: Date,
default: Date.now
}
});
const Conversation =
mongoose.models.Conversation || mongoose.model('Conversation', convoSchema);
module.exports = {
saveConvo: async ({ conversationId, parentMessageId, title }) => {
try {
const messages = await getMessages({ conversationId });
const update = { parentMessageId, messages };
if (title) {
update.title = title;
}
return await Conversation.findOneAndUpdate(
{ conversationId },
{ $set: update },
{ new: true, upsert: true }
).exec();
} catch (error) {
console.log(error);
return { message: 'Error saving conversation' };
}
},
updateConvo: async ({ conversationId, ...update }) => {
try {
return await Conversation.findOneAndUpdate(
{ conversationId },
update,
{ new: true }
).exec();
} catch (error) {
console.log(error);
return { message: 'Error updating conversation' };
}
},
getConvos: async () => await Conversation.find({}).sort({ created: -1 }).exec(),
deleteConvos: async (filter) => {
let deleteCount = await Conversation.deleteMany(filter).exec();
deleteCount.messages = await deleteMessages(filter);
return deleteCount;
}
};