71 lines
1.3 KiB
TypeScript
71 lines
1.3 KiB
TypeScript
import { connectionMongo, getMongoModel } from '../../../common/mongo';
|
|
const { Schema } = connectionMongo;
|
|
import { TeamSchema as TeamType } from '@fastgpt/global/support/user/team/type.d';
|
|
import { userCollectionName } from '../../user/schema';
|
|
import { TeamCollectionName } from '@fastgpt/global/support/user/team/constant';
|
|
|
|
const TeamSchema = new Schema({
|
|
name: {
|
|
type: String,
|
|
required: true
|
|
},
|
|
ownerId: {
|
|
type: Schema.Types.ObjectId,
|
|
ref: userCollectionName
|
|
},
|
|
avatar: {
|
|
type: String,
|
|
default: '/icon/logo.svg'
|
|
},
|
|
createTime: {
|
|
type: Date,
|
|
default: () => Date.now()
|
|
},
|
|
balance: Number,
|
|
teamDomain: {
|
|
type: String
|
|
},
|
|
limit: {
|
|
lastExportDatasetTime: {
|
|
type: Date
|
|
},
|
|
lastWebsiteSyncTime: {
|
|
type: Date
|
|
}
|
|
},
|
|
lafAccount: {
|
|
token: {
|
|
type: String
|
|
},
|
|
appid: {
|
|
type: String
|
|
},
|
|
pat: {
|
|
type: String
|
|
}
|
|
},
|
|
openaiAccount: {
|
|
type: {
|
|
key: String,
|
|
baseUrl: String
|
|
}
|
|
},
|
|
externalWorkflowVariables: {
|
|
type: Object,
|
|
default: {}
|
|
},
|
|
notificationAccount: {
|
|
type: String,
|
|
required: false
|
|
}
|
|
});
|
|
|
|
try {
|
|
TeamSchema.index({ name: 1 });
|
|
TeamSchema.index({ ownerId: 1 });
|
|
} catch (error) {
|
|
console.log(error);
|
|
}
|
|
|
|
export const MongoTeam = getMongoModel<TeamType>(TeamCollectionName, TeamSchema);
|