ben-ai-models-registry 1.0.43 → 1.0.44

package/index.js

@@ -12,19 +12,21 @@ const EventLog = require("./models/EventLog");
 const FileStore = require("./models/FileStore");
 const State = require("./models/State");
 const StatePermission = require("./models/StatePermission");
+const UserChatHistory = require("./models/UserChatHistory");
 
 module.exports = {
-	JobStatus,
-	Run,
-	Files,
-	Dataset,
-	State,
-	Program,
-	QA,
-	ChatMessages,
-	LanguageObject,
-	User,
-	StatePermission,
-	EventLog,
+  JobStatus,
+  Run,
+  Files,
+  Dataset,
+  State,
+  Program,
+  QA,
+  ChatMessages,
+  LanguageObject,
+  User,
+  StatePermission,
+  EventLog,
   FileStore,
+  UserChatHistory,
 };

package/models/UserChatHistory.js

@@ -0,0 +1,45 @@
+const mongoose = require("mongoose");
+const { Schema } = mongoose;
+
+const UserChatSchema = new Schema(
+  {
+    chatId: {
+      type: Schema.Types.ObjectId,
+      ref: "ChatMessage",
+    },
+    title: { type: String, maxlength: 50, default: "" },
+    updatedOn: {
+      type: Date,
+    },
+  },
+  {
+    _id: false,
+    timestamps: true,
+  }
+);
+
+const ProgramBucketSchema = new Schema(
+  {
+    userChats: { type: [UserChatSchema], default: [] },
+  },
+  { _id: false }
+);
+
+const UserChatHistorySchema = new Schema(
+  {
+    userId: { type: Schema.Types.ObjectId, ref: "User", required: true },
+    state: { type: String, required: true },
+    programs: {
+      type: Map,
+      of: ProgramBucketSchema,
+      default: () => ({}),
+    },
+  },
+  { timestamps: true, versionKey: false }
+);
+
+UserChatHistorySchema.index({ userId: 1, state: 1 }, { unique: true });
+
+module.exports =
+  mongoose.models.UserChatHistory ||
+  mongoose.model("UserChatHistory", UserChatHistorySchema);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ben-ai-models-registry",
-  "version": "1.0.43",
+  "version": "1.0.44",
   "description": "",
   "files": [
     "/models"