mirror of
https://github.com/danny-avila/LibreChat.git
synced 2025-12-17 08:50:15 +01:00
6358383001
* feat: add global teardown to remove test data and add registration/log-out to auth flow * refactor(models/Conversation): index user field and add JSDoc to deleteConvos * refactor: add user index to message schema and ensure user is saved to each Message * refactor: add user to each saveMessage call * fix: handle case where title is null in zod schema * feat(e2e): ensure messages are deleted on cleanUp * fix: set last convo for all endpoints on conversation update * fix: enable registration for CI env
69 lines
1.6 KiB
JavaScript
69 lines
1.6 KiB
JavaScript
const mongoose = require('mongoose');
|
|
const mongoMeili = require('../plugins/mongoMeili');
|
|
const { conversationPreset } = require('./defaults');
|
|
const convoSchema = mongoose.Schema(
|
|
{
|
|
conversationId: {
|
|
type: String,
|
|
unique: true,
|
|
required: true,
|
|
index: true,
|
|
meiliIndex: true,
|
|
},
|
|
title: {
|
|
type: String,
|
|
default: 'New Chat',
|
|
meiliIndex: true,
|
|
},
|
|
user: {
|
|
type: String,
|
|
index: true,
|
|
default: null,
|
|
},
|
|
messages: [{ type: mongoose.Schema.Types.ObjectId, ref: 'Message' }],
|
|
// google only
|
|
examples: [{ type: mongoose.Schema.Types.Mixed }],
|
|
agentOptions: {
|
|
type: mongoose.Schema.Types.Mixed,
|
|
default: null,
|
|
},
|
|
...conversationPreset,
|
|
// for bingAI only
|
|
bingConversationId: {
|
|
type: String,
|
|
default: null,
|
|
},
|
|
jailbreakConversationId: {
|
|
type: String,
|
|
default: null,
|
|
},
|
|
conversationSignature: {
|
|
type: String,
|
|
default: null,
|
|
},
|
|
clientId: {
|
|
type: String,
|
|
default: null,
|
|
},
|
|
invocationId: {
|
|
type: Number,
|
|
default: 1,
|
|
},
|
|
},
|
|
{ timestamps: true },
|
|
);
|
|
|
|
if (process.env.MEILI_HOST && process.env.MEILI_MASTER_KEY) {
|
|
convoSchema.plugin(mongoMeili, {
|
|
host: process.env.MEILI_HOST,
|
|
apiKey: process.env.MEILI_MASTER_KEY,
|
|
indexName: 'convos', // Will get created automatically if it doesn't exist already
|
|
primaryKey: 'conversationId',
|
|
});
|
|
}
|
|
|
|
convoSchema.index({ createdAt: 1 });
|
|
|
|
const Conversation = mongoose.models.Conversation || mongoose.model('Conversation', convoSchema);
|
|
|
|
module.exports = Conversation;
|