57 lines
1.1 KiB
TypeScript
57 lines
1.1 KiB
TypeScript
import { Schema, model, models, Model } from 'mongoose';
|
|
import { ChatSchema as ChatType } from '@/types/mongoSchema';
|
|
import { ChatRoleMap } from '@/constants/chat';
|
|
|
|
const ChatSchema = new Schema({
|
|
userId: {
|
|
type: Schema.Types.ObjectId,
|
|
ref: 'user',
|
|
required: true
|
|
},
|
|
modelId: {
|
|
type: Schema.Types.ObjectId,
|
|
ref: 'model',
|
|
required: true
|
|
},
|
|
expiredTime: {
|
|
// 过期时间
|
|
type: Number,
|
|
default: () => new Date()
|
|
},
|
|
loadAmount: {
|
|
// 剩余加载次数
|
|
type: Number,
|
|
default: -1
|
|
},
|
|
updateTime: {
|
|
type: Date,
|
|
default: () => new Date()
|
|
},
|
|
title: {
|
|
type: String,
|
|
default: '历史记录'
|
|
},
|
|
content: {
|
|
type: [
|
|
{
|
|
obj: {
|
|
type: String,
|
|
required: true,
|
|
enum: Object.keys(ChatRoleMap)
|
|
},
|
|
value: {
|
|
type: String,
|
|
required: true
|
|
},
|
|
systemPrompt: {
|
|
type: String,
|
|
default: ''
|
|
}
|
|
}
|
|
],
|
|
default: []
|
|
}
|
|
});
|
|
|
|
export const Chat: Model<ChatType> = models['chat'] || model('chat', ChatSchema);
|