new framwork

client/src/service/models/authCode.ts

@@ -0,0 +1,26 @@
+import { Schema, model, models, Model } from 'mongoose';
+import { AuthCodeSchema as AuthCodeType } from '@/types/mongoSchema';
+
+const AuthCodeSchema = new Schema({
+  username: {
+    type: String,
+    required: true
+  },
+  code: {
+    type: String,
+    required: true,
+    length: 6
+  },
+  type: {
+    type: String,
+    enum: ['register', 'findPassword'],
+    required: true
+  },
+  expiredTime: {
+    type: Number,
+    default: () => Date.now() + 5 * 60 * 1000
+  }
+});
+
+export const AuthCode: Model<AuthCodeType> =
+  models['auth_code'] || model('auth_code', AuthCodeSchema);

client/src/service/models/bill.ts

@@ -0,0 +1,53 @@
+import { Schema, model, models, Model } from 'mongoose';
+import { ChatModelMap, embeddingModel } from '@/constants/model';
+import { BillSchema as BillType } from '@/types/mongoSchema';
+import { BillTypeMap } from '@/constants/user';
+
+const BillSchema = new Schema({
+  userId: {
+    type: Schema.Types.ObjectId,
+    ref: 'user',
+    required: true
+  },
+  type: {
+    type: String,
+    enum: Object.keys(BillTypeMap),
+    required: true
+  },
+  modelName: {
+    type: String,
+    enum: [...Object.keys(ChatModelMap), embeddingModel],
+    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
+  }
+});
+
+try {
+  BillSchema.index({ time: -1 });
+  BillSchema.index({ userId: 1 });
+} catch (error) {
+  console.log(error);
+}
+
+export const Bill: Model<BillType> = models['bill'] || model('bill', BillSchema);

client/src/service/models/chat.ts

@@ -0,0 +1,78 @@
+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: '历史记录'
+  },
+  customTitle: {
+    type: String,
+    default: ''
+  },
+  latestChat: {
+    type: String,
+    default: ''
+  },
+  top: {
+    type: Boolean
+  },
+  content: {
+    type: [
+      {
+        obj: {
+          type: String,
+          required: true,
+          enum: Object.keys(ChatRoleMap)
+        },
+        value: {
+          type: String,
+          required: true
+        },
+        quote: {
+          type: [
+            {
+              id: String,
+              q: String,
+              a: String,
+              source: String
+            }
+          ],
+          default: []
+        },
+        systemPrompt: {
+          type: String,
+          default: ''
+        }
+      }
+    ],
+    default: []
+  }
+});
+
+export const Chat: Model<ChatType> = models['chat'] || model('chat', ChatSchema);

client/src/service/models/collection.ts

@@ -0,0 +1,18 @@
+import { Schema, model, models, Model as MongoModel } from 'mongoose';
+import { CollectionSchema as CollectionType } from '@/types/mongoSchema';
+
+const CollectionSchema = new Schema({
+  userId: {
+    type: Schema.Types.ObjectId,
+    ref: 'user',
+    required: true
+  },
+  modelId: {
+    type: Schema.Types.ObjectId,
+    ref: 'model',
+    required: true
+  }
+});
+
+export const Collection: MongoModel<CollectionType> =
+  models['collection'] || model('collection', CollectionSchema);

client/src/service/models/inform.ts

@@ -0,0 +1,40 @@
+import { Schema, model, models, Model } from 'mongoose';
+import { informSchema } from '@/types/mongoSchema';
+import { InformTypeMap } from '@/constants/user';
+
+const InformSchema = new Schema({
+  userId: {
+    type: Schema.Types.ObjectId,
+    ref: 'user',
+    required: true
+  },
+  time: {
+    type: Date,
+    default: () => new Date()
+  },
+  type: {
+    type: String,
+    enum: Object.keys(InformTypeMap)
+  },
+  title: {
+    type: String,
+    required: true
+  },
+  content: {
+    type: String,
+    required: true
+  },
+  read: {
+    type: Boolean,
+    default: false
+  }
+});
+
+try {
+  InformSchema.index({ time: -1 });
+  InformSchema.index({ userId: 1 });
+} catch (error) {
+  console.log(error);
+}
+
+export const Inform: Model<informSchema> = models['inform'] || model('inform', InformSchema);

client/src/service/models/kb.ts

@@ -0,0 +1,28 @@
+import { Schema, model, models, Model } from 'mongoose';
+import { kbSchema as SchemaType } from '@/types/mongoSchema';
+
+const kbSchema = new Schema({
+  userId: {
+    type: Schema.Types.ObjectId,
+    ref: 'user',
+    required: true
+  },
+  updateTime: {
+    type: Date,
+    default: () => new Date()
+  },
+  avatar: {
+    type: String,
+    default: '/icon/logo.png'
+  },
+  name: {
+    type: String,
+    required: true
+  },
+  tags: {
+    type: [String],
+    default: []
+  }
+});
+
+export const KB: Model<SchemaType> = models['kb'] || model('kb', kbSchema);

client/src/service/models/model.ts

@@ -0,0 +1,92 @@
+import { Schema, model, models, Model as MongoModel } from 'mongoose';
+import { ModelSchema as ModelType } from '@/types/mongoSchema';
+import {
+  ModelVectorSearchModeMap,
+  appVectorSearchModeEnum,
+  ChatModelMap,
+  OpenAiChatEnum
+} from '@/constants/model';
+
+const ModelSchema = new Schema({
+  userId: {
+    type: Schema.Types.ObjectId,
+    ref: 'user',
+    required: true
+  },
+  name: {
+    type: String,
+    required: true
+  },
+  avatar: {
+    type: String,
+    default: '/icon/logo.png'
+  },
+  status: {
+    type: String,
+    required: true,
+    enum: ['waiting', 'running', 'training', 'closed']
+  },
+  updateTime: {
+    type: Date,
+    default: () => new Date()
+  },
+  chat: {
+    relatedKbs: {
+      type: [Schema.Types.ObjectId],
+      ref: 'kb',
+      default: []
+    },
+    searchMode: {
+      // knowledge base search mode
+      type: String,
+      enum: Object.keys(ModelVectorSearchModeMap),
+      default: appVectorSearchModeEnum.hightSimilarity
+    },
+    systemPrompt: {
+      // 系统提示词
+      type: String,
+      default: ''
+    },
+    temperature: {
+      type: Number,
+      min: 0,
+      max: 10,
+      default: 0
+    },
+    chatModel: {
+      // 聊天时使用的模型
+      type: String,
+      enum: Object.keys(ChatModelMap),
+      default: OpenAiChatEnum.GPT35
+    }
+  },
+  share: {
+    isShare: {
+      type: Boolean,
+      default: false
+    },
+    isShareDetail: {
+      // share model detail info. false: just show name and intro
+      type: Boolean,
+      default: false
+    },
+    intro: {
+      type: String,
+      default: '',
+      maxlength: 150
+    },
+    collection: {
+      type: Number,
+      default: 0
+    }
+  }
+});
+
+try {
+  ModelSchema.index({ updateTime: -1 });
+  ModelSchema.index({ 'share.collection': -1 });
+} catch (error) {
+  console.log(error);
+}
+
+export const Model: MongoModel<ModelType> = models['model'] || model('model', ModelSchema);

client/src/service/models/openapi.ts

@@ -0,0 +1,23 @@
+import { Schema, model, models, Model } from 'mongoose';
+import { OpenApiSchema } from '@/types/mongoSchema';
+
+const OpenApiSchema = new Schema({
+  userId: {
+    type: Schema.Types.ObjectId,
+    ref: 'user',
+    required: true
+  },
+  apiKey: {
+    type: String,
+    required: true
+  },
+  createTime: {
+    type: Date,
+    default: () => new Date()
+  },
+  lastUsedTime: {
+    type: Date
+  }
+});
+
+export const OpenApi: Model<OpenApiSchema> = models['openapi'] || model('openapi', OpenApiSchema);

client/src/service/models/pay.ts

@@ -0,0 +1,29 @@
+import { Schema, model, models, Model } from 'mongoose';
+import { PaySchema as PayType } from '@/types/mongoSchema';
+const PaySchema = new Schema({
+  userId: {
+    type: Schema.Types.ObjectId,
+    ref: 'user',
+    required: true
+  },
+  createTime: {
+    type: Date,
+    default: () => new Date()
+  },
+  price: {
+    type: Number,
+    required: true
+  },
+  orderId: {
+    type: String,
+    required: true
+  },
+  status: {
+    // 支付的状态
+    type: String,
+    default: 'NOTPAY',
+    enum: ['SUCCESS', 'REFUND', 'NOTPAY', 'CLOSED']
+  }
+});
+
+export const Pay: Model<PayType> = models['pay'] || model('pay', PaySchema);

client/src/service/models/promotionRecord.ts

@@ -0,0 +1,31 @@
+import { Schema, model, models, Model } from 'mongoose';
+import { PromotionRecordSchema as PromotionRecordType } from '@/types/mongoSchema';
+
+const PromotionRecordSchema = new Schema({
+  userId: {
+    type: Schema.Types.ObjectId,
+    ref: 'user',
+    required: true
+  },
+  objUId: {
+    type: Schema.Types.ObjectId,
+    ref: 'user',
+    required: false
+  },
+  createTime: {
+    type: Date,
+    default: () => new Date()
+  },
+  type: {
+    type: String,
+    required: true,
+    enum: ['invite', 'shareModel', 'withdraw']
+  },
+  amount: {
+    type: Number,
+    required: true
+  }
+});
+
+export const promotionRecord: Model<PromotionRecordType> =
+  models['promotionRecord'] || model('promotionRecord', PromotionRecordSchema);

client/src/service/models/shareChat.ts

@@ -0,0 +1,38 @@
+import { Schema, model, models, Model } from 'mongoose';
+import { ShareChatSchema as ShareChatSchemaType } from '@/types/mongoSchema';
+import { hashPassword } from '@/service/utils/tools';
+
+const ShareChatSchema = new Schema({
+  userId: {
+    type: Schema.Types.ObjectId,
+    ref: 'user',
+    required: true
+  },
+  modelId: {
+    type: Schema.Types.ObjectId,
+    ref: 'model',
+    required: true
+  },
+  name: {
+    type: String,
+    required: true
+  },
+  password: {
+    type: String,
+    set: (val: string) => hashPassword(val)
+  },
+  tokens: {
+    type: Number,
+    default: 0
+  },
+  maxContext: {
+    type: Number,
+    default: 20
+  },
+  lastTime: {
+    type: Date
+  }
+});
+
+export const ShareChat: Model<ShareChatSchemaType> =
+  models['shareChat'] || model('shareChat', ShareChatSchema);

client/src/service/models/trainingData.ts

@@ -0,0 +1,48 @@
+/* 模型的知识库 */
+import { Schema, model, models, Model as MongoModel } from 'mongoose';
+import { TrainingDataSchema as TrainingDateType } from '@/types/mongoSchema';
+import { TrainingTypeMap } from '@/constants/plugin';
+
+// pgList and vectorList, Only one of them will work
+const TrainingDataSchema = new Schema({
+  userId: {
+    type: Schema.Types.ObjectId,
+    ref: 'user',
+    required: true
+  },
+  kbId: {
+    type: Schema.Types.ObjectId,
+    ref: 'kb',
+    required: true
+  },
+  lockTime: {
+    type: Date,
+    default: () => new Date('2000/1/1')
+  },
+  mode: {
+    type: String,
+    enum: Object.keys(TrainingTypeMap),
+    required: true
+  },
+  prompt: {
+    // 拆分时的提示词
+    type: String,
+    default: ''
+  },
+  q: {
+    // 如果是
+    type: String,
+    default: ''
+  },
+  a: {
+    type: String,
+    default: ''
+  },
+  source: {
+    type: String,
+    default: ''
+  }
+});
+
+export const TrainingData: MongoModel<TrainingDateType> =
+  models['trainingData'] || model('trainingData', TrainingDataSchema);

client/src/service/models/user.ts

@@ -0,0 +1,57 @@
+import { Schema, model, models, Model } from 'mongoose';
+import { hashPassword } from '@/service/utils/tools';
+import { PRICE_SCALE } from '@/constants/common';
+import { UserModelSchema } from '@/types/mongoSchema';
+
+const UserSchema = new Schema({
+  username: {
+    // 可以是手机/邮箱，新的验证都只用手机
+    type: String,
+    required: true,
+    unique: true // 唯一
+  },
+  password: {
+    type: String,
+    required: true,
+    set: (val: string) => hashPassword(val),
+    get: (val: string) => hashPassword(val),
+    select: false
+  },
+  createTime: {
+    type: Date,
+    default: () => new Date()
+  },
+  avatar: {
+    type: String,
+    default: '/icon/human.png'
+  },
+  balance: {
+    // 平台余额，不可提现
+    type: Number,
+    default: 2 * PRICE_SCALE
+  },
+  inviterId: {
+    // 谁邀请注册的
+    type: Schema.Types.ObjectId,
+    ref: 'user'
+  },
+  promotion: {
+    rate: {
+      // 返现比例
+      type: Number,
+      default: 15
+    }
+  },
+  openaiKey: {
+    type: String,
+    default: ''
+  },
+  limit: {
+    exportKbTime: {
+      // Every half hour
+      type: Date
+    }
+  }
+});
+
+export const User: Model<UserModelSchema> = models['user'] || model('user', UserSchema);