diff --git a/packages/data-schemas/src/models/convo.ts b/packages/data-schemas/src/models/convo.ts index 38d7e554ee..9de53c577c 100644 --- a/packages/data-schemas/src/models/convo.ts +++ b/packages/data-schemas/src/models/convo.ts @@ -1,17 +1,6 @@ import mongoose from 'mongoose'; import type * as t from '~/types'; -import mongoMeili from '~/models/plugins/mongoMeili'; import convoSchema from '~/schema/convo'; -if (process.env.MEILI_HOST && process.env.MEILI_MASTER_KEY) { - convoSchema.plugin(mongoMeili, { - host: process.env.MEILI_HOST, - apiKey: process.env.MEILI_MASTER_KEY, - /** Note: Will get created automatically if it doesn't exist already */ - indexName: 'convos', - primaryKey: 'conversationId', - }); -} - export const Conversation = mongoose.models.Conversation || mongoose.model('Conversation', convoSchema); diff --git a/packages/data-schemas/src/models/message.ts b/packages/data-schemas/src/models/message.ts index 46af4a4a28..7cd51dc9c7 100644 --- a/packages/data-schemas/src/models/message.ts +++ b/packages/data-schemas/src/models/message.ts @@ -1,16 +1,6 @@ import mongoose from 'mongoose'; -import mongoMeili from '~/models/plugins/mongoMeili'; import messageSchema from '~/schema/message'; import type * as t from '~/types'; -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', - }); -} - export const Message = mongoose.models.Message || mongoose.model('Message', messageSchema); diff --git a/packages/data-schemas/src/schema/convo.ts b/packages/data-schemas/src/schema/convo.ts index 7a18eeb66b..8b0eb5b5c7 100644 --- a/packages/data-schemas/src/schema/convo.ts +++ b/packages/data-schemas/src/schema/convo.ts @@ -1,4 +1,5 @@ -import mongoose, { Schema } from 'mongoose'; +import { Schema } from 'mongoose'; +import mongoMeili from '~/models/plugins/mongoMeili'; import { conversationPreset } from './defaults'; import { IConversation } from '~/types'; @@ -20,9 +21,9 @@ const convoSchema: Schema = new Schema( type: String, index: true, }, - messages: [{ type: mongoose.Schema.Types.ObjectId, ref: 'Message' }], + messages: [{ type: Schema.Types.ObjectId, ref: 'Message' }], agentOptions: { - type: mongoose.Schema.Types.Mixed, + type: Schema.Types.Mixed, }, ...conversationPreset, agent_id: { @@ -47,4 +48,14 @@ convoSchema.index({ expiredAt: 1 }, { expireAfterSeconds: 0 }); convoSchema.index({ createdAt: 1, updatedAt: 1 }); convoSchema.index({ conversationId: 1, user: 1 }, { unique: 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, + /** Note: Will get created automatically if it doesn't exist already */ + indexName: 'convos', + primaryKey: 'conversationId', + }); +} + export default convoSchema; diff --git a/packages/data-schemas/src/schema/message.ts b/packages/data-schemas/src/schema/message.ts index 82970de6de..8406e486d4 100644 --- a/packages/data-schemas/src/schema/message.ts +++ b/packages/data-schemas/src/schema/message.ts @@ -1,4 +1,5 @@ import mongoose, { Schema, Document } from 'mongoose'; +import mongoMeili from '~/models/plugins/mongoMeili'; // @ts-ignore export interface IMessage extends Document { @@ -182,4 +183,13 @@ messageSchema.index({ expiredAt: 1 }, { expireAfterSeconds: 0 }); messageSchema.index({ createdAt: 1 }); messageSchema.index({ messageId: 1, user: 1 }, { unique: 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', + }); +} + export default messageSchema;