46 lines
930 B
TypeScript
46 lines
930 B
TypeScript
import { Schema, model, models, Model } from 'mongoose';
|
|
import { modelList } from '@/constants/model';
|
|
import { BillSchema as BillType } from '@/types/mongoSchema';
|
|
|
|
const BillSchema = new Schema({
|
|
userId: {
|
|
type: Schema.Types.ObjectId,
|
|
ref: 'user',
|
|
required: true
|
|
},
|
|
type: {
|
|
type: String,
|
|
enum: ['chat', 'splitData', 'return'],
|
|
required: true
|
|
},
|
|
modelName: {
|
|
type: String,
|
|
enum: modelList.map((item) => item.model),
|
|
required: true
|
|
},
|
|
chatId: {
|
|
type: Schema.Types.ObjectId,
|
|
ref: 'chat'
|
|
},
|
|
time: {
|
|
type: Date,
|
|
default: () => new Date()
|
|
},
|
|
textLen: {
|
|
// 提示词+响应的总字数
|
|
type: Number,
|
|
required: true
|
|
},
|
|
tokenLen: {
|
|
// 折算成 token 的数量
|
|
type: Number,
|
|
required: true
|
|
},
|
|
price: {
|
|
type: Number,
|
|
required: true
|
|
}
|
|
});
|
|
|
|
export const Bill: Model<BillType> = models['bill'] || model('bill', BillSchema);
|