mirror of
https://github.com/danny-avila/LibreChat.git
synced 2025-12-17 00:40:14 +01:00
379e470e38
* chore: bump langchain to v0.0.213 from v0.0.186 * fix: handle abort edge cases: - abort message server-side if response experienced error mid-generation - attempt to recover message if aborting resulted in error - if abortKey is not provided, use conversationId if it exists - if headers were already sent, send an Event stream message - issue warning for possible Google censor/filter refactor(streamResponse): for `sendError`, allow passing overrides so that error can include partial generation, improve typing for `sendMessage` * chore(MessageContent): remove eslint warning for unused `i`, rephrase unfinished message text * fix(useSSE): avoid invoking cancelHandler if the abort response was 404 * chore(TMessage): remove unnecessary, unused legacy message property `submitting` * chore(TMessage): remove unnecessary legacy message property `cancelled` * chore(abortMiddleware): remove unused `errorText` property to avoid confusion
117 lines
2.2 KiB
JavaScript
117 lines
2.2 KiB
JavaScript
const mongoose = require('mongoose');
|
|
const mongoMeili = require('~/models/plugins/mongoMeili');
|
|
const messageSchema = mongoose.Schema(
|
|
{
|
|
messageId: {
|
|
type: String,
|
|
unique: true,
|
|
required: true,
|
|
index: true,
|
|
meiliIndex: true,
|
|
},
|
|
conversationId: {
|
|
type: String,
|
|
required: true,
|
|
meiliIndex: true,
|
|
},
|
|
user: {
|
|
type: String,
|
|
index: true,
|
|
default: null,
|
|
},
|
|
model: {
|
|
type: String,
|
|
},
|
|
conversationSignature: {
|
|
type: String,
|
|
// required: true
|
|
},
|
|
clientId: {
|
|
type: String,
|
|
},
|
|
invocationId: {
|
|
type: Number,
|
|
},
|
|
parentMessageId: {
|
|
type: String,
|
|
// required: true
|
|
},
|
|
tokenCount: {
|
|
type: Number,
|
|
},
|
|
summaryTokenCount: {
|
|
type: Number,
|
|
},
|
|
sender: {
|
|
type: String,
|
|
required: true,
|
|
meiliIndex: true,
|
|
},
|
|
text: {
|
|
type: String,
|
|
required: true,
|
|
meiliIndex: true,
|
|
},
|
|
summary: {
|
|
type: String,
|
|
},
|
|
isCreatedByUser: {
|
|
type: Boolean,
|
|
required: true,
|
|
default: false,
|
|
},
|
|
isEdited: {
|
|
type: Boolean,
|
|
default: false,
|
|
},
|
|
unfinished: {
|
|
type: Boolean,
|
|
default: false,
|
|
},
|
|
error: {
|
|
type: Boolean,
|
|
default: false,
|
|
},
|
|
finish_reason: {
|
|
type: String,
|
|
},
|
|
_meiliIndex: {
|
|
type: Boolean,
|
|
required: false,
|
|
select: false,
|
|
default: false,
|
|
},
|
|
files: [{ type: mongoose.Schema.Types.Mixed }],
|
|
plugin: {
|
|
latest: {
|
|
type: String,
|
|
required: false,
|
|
},
|
|
inputs: {
|
|
type: [mongoose.Schema.Types.Mixed],
|
|
required: false,
|
|
},
|
|
outputs: {
|
|
type: String,
|
|
required: false,
|
|
},
|
|
},
|
|
plugins: [{ type: mongoose.Schema.Types.Mixed }],
|
|
},
|
|
{ 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;
|