mirror of
https://github.com/danny-avila/LibreChat.git
synced 2025-12-17 08:50:15 +01:00
* feat(mongoDb): utitlize lean queries and index createdAt timestamps for cosmosDB support * fix: remove unnecessary lean() method from deleteMany calls * fix: remove unnecessary lean() method from deleteMany calls * fix: remove lean() from queries that need hydration * chore(migrateDb.js): remove unused migration script fix(Preset.js): return lean documents when retrieving presets refactor(index.js): remove migration script from server initialization refactor(convos.js): remove toObject() when sending conversation object refactor(presets.js): remove toObject() when sending presets object
107 lines
2 KiB
JavaScript
107 lines
2 KiB
JavaScript
const mongoose = require('mongoose');
|
|
const mongoMeili = require('../plugins/mongoMeili');
|
|
const messageSchema = mongoose.Schema(
|
|
{
|
|
messageId: {
|
|
type: String,
|
|
unique: true,
|
|
required: true,
|
|
index: true,
|
|
meiliIndex: true,
|
|
},
|
|
conversationId: {
|
|
type: String,
|
|
required: true,
|
|
meiliIndex: true,
|
|
},
|
|
model: {
|
|
type: String,
|
|
},
|
|
conversationSignature: {
|
|
type: String,
|
|
// required: true
|
|
},
|
|
clientId: {
|
|
type: String,
|
|
},
|
|
invocationId: {
|
|
type: String,
|
|
},
|
|
parentMessageId: {
|
|
type: String,
|
|
// required: true
|
|
},
|
|
tokenCount: {
|
|
type: Number,
|
|
},
|
|
refinedTokenCount: {
|
|
type: Number,
|
|
},
|
|
sender: {
|
|
type: String,
|
|
required: true,
|
|
meiliIndex: true,
|
|
},
|
|
text: {
|
|
type: String,
|
|
required: true,
|
|
meiliIndex: true,
|
|
},
|
|
refinedMessageText: {
|
|
type: String,
|
|
},
|
|
isCreatedByUser: {
|
|
type: Boolean,
|
|
required: true,
|
|
default: false,
|
|
},
|
|
unfinished: {
|
|
type: Boolean,
|
|
default: false,
|
|
},
|
|
cancelled: {
|
|
type: Boolean,
|
|
default: false,
|
|
},
|
|
error: {
|
|
type: Boolean,
|
|
default: false,
|
|
},
|
|
_meiliIndex: {
|
|
type: Boolean,
|
|
required: false,
|
|
select: false,
|
|
default: false,
|
|
},
|
|
plugin: {
|
|
latest: {
|
|
type: String,
|
|
required: false,
|
|
},
|
|
inputs: {
|
|
type: [mongoose.Schema.Types.Mixed],
|
|
required: false,
|
|
},
|
|
outputs: {
|
|
type: String,
|
|
required: false,
|
|
},
|
|
},
|
|
},
|
|
{ timestamps: true },
|
|
);
|
|
|
|
if (process.env.MEILI_HOST && process.env.MEILI_MASTER_KEY) {
|
|
messageSchema.plugin(mongoMeili, {
|
|
host: process.env.MEILI_HOST,
|
|
apiKey: process.env.MEILI_MASTER_KEY,
|
|
indexName: 'messages',
|
|
primaryKey: 'messageId',
|
|
});
|
|
}
|
|
|
|
messageSchema.index({ createdAt: 1 });
|
|
|
|
const Message = mongoose.models.Message || mongoose.model('Message', messageSchema);
|
|
|
|
module.exports = Message;
|