myapps

client/src/service/models/app.ts

@@ -0,0 +1,108 @@
+import { Schema, model, models, Model } from 'mongoose';
+import { AppSchema as AppType } from '@/types/mongoSchema';
+import { ChatModelMap, OpenAiChatEnum } from '@/constants/model';
+
+const AppSchema = new Schema({
+  userId: {
+    type: Schema.Types.ObjectId,
+    ref: 'user',
+    required: true
+  },
+  name: {
+    type: String,
+    required: true
+  },
+  avatar: {
+    type: String,
+    default: '/icon/logo.png'
+  },
+  intro: {
+    type: String,
+    default: ''
+  },
+  updateTime: {
+    type: Date,
+    default: () => new Date()
+  },
+  chat: {
+    relatedKbs: {
+      type: [Schema.Types.ObjectId],
+      ref: 'kb',
+      default: []
+    },
+    searchSimilarity: {
+      type: Number,
+      default: 0.8
+    },
+    searchLimit: {
+      type: Number,
+      default: 5
+    },
+    searchEmptyText: {
+      type: String,
+      default: ''
+    },
+    systemPrompt: {
+      type: String,
+      default: ''
+    },
+    limitPrompt: {
+      type: String,
+      default: ''
+    },
+    maxToken: {
+      type: Number,
+      default: 4000,
+      min: 100
+    },
+    temperature: {
+      type: Number,
+      min: 0,
+      max: 10,
+      default: 0
+    },
+    chatModel: {
+      // 聊天时使用的模型
+      type: String,
+      enum: Object.keys(ChatModelMap),
+      default: OpenAiChatEnum.GPT3516k
+    }
+  },
+  share: {
+    topNum: {
+      type: Number,
+      default: 0
+    },
+    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
+    }
+  },
+  modules: {
+    type: Array,
+    default: []
+  }
+});
+
+try {
+  AppSchema.index({ updateTime: -1 });
+  AppSchema.index({ 'share.collection': -1 });
+} catch (error) {
+  console.log(error);
+}
+
+export const App: Model<AppType> = models['model'] || model('model', AppSchema);