ben-ai-models-registry 1.0.7 → 1.0.8

package/models/Dataset.js

@@ -0,0 +1,34 @@
+const mongoose = require('mongoose');
+const { Schema, Types } = mongoose;
+
+const QAPairSchema = new Schema(
+  {
+    question: { type: String },
+    expectedAnswer: { type: String },
+  },
+  { timestamps: true, versionKey: false }
+);
+
+const DatasetSchema = new Schema(
+  {
+    programID: {
+      type: Types.ObjectId,
+    },
+    name: {
+      type: String,
+    },
+    description: {
+      type: String,
+    },
+    qaPairs: [QAPairSchema],
+    runs: [
+      {
+        type: Types.ObjectId,
+        ref: "Run",
+      },
+    ],
+  },
+  { timestamps: true, versionKey: false }
+);
+
+module.exports = mongoose.models.Dataset || mongoose.model("Dataset", DatasetSchema);

package/models/Files.js

@@ -0,0 +1,29 @@
+const mongoose = require("mongoose");
+const { Schema } = mongoose;
+
+const FileSchema = new Schema(
+  {
+    displayName: {
+      type: String,
+      required: [true, "Please provide a display name."],
+    },
+    refName: {
+      type: String,
+    },
+    description: {
+      type: String,
+    },
+    type: {
+      type: String,
+    },
+    url: {
+      type: String,
+    },
+    openAIFileID: {
+      type: String,
+    },
+  },
+  { timestamps: true, versionKey: false }
+);
+
+module.exports = mongoose.models.Files || mongoose.model("Files", FileSchema);

package/models/Program.js

@@ -0,0 +1,120 @@
+const mongoose = require("mongoose");
+const { Schema, Types } = mongoose;
+
+const FaqSchema = new Schema(
+  {
+    question: {
+      type: [String],
+      required: true,
+    },
+    createdAt: {
+      type: Date,
+      default: Date.now,
+    },
+  },
+  { timestamps: true, versionKey: false }
+);
+
+const TestOpenAISchema = new Schema(
+  {
+    testAssistantID: {
+      type: String,
+    },
+    testVectorStoreID: {
+      type: String,
+    },
+    files: [{ type: Schema.Types.ObjectId, ref: "Files" }],
+  },
+  { timestamps: true, versionKey: false }
+);
+
+const ProgramSchema = new Schema(
+  {
+    name: {
+      type: String,
+      required: [true, "Please provide a program name."],
+    },
+    state: {
+      type: String,
+      enum: [
+        "Alabama",
+        "Alaska",
+        "Arizona",
+        "Arkansas",
+        "California",
+        "Colorado",
+        "Connecticut",
+        "Delaware",
+        "District of Columbia",
+        "Florida",
+        "Georgia",
+        "Hawaii",
+        "Idaho",
+        "Illinois",
+        "Indiana",
+        "Iowa",
+        "Kansas",
+        "Kentucky",
+        "Louisiana",
+        "Maine",
+        "Maryland",
+        "Massachusetts",
+        "Michigan",
+        "Minnesota",
+        "Mississippi",
+        "Missouri",
+        "Montana",
+        "Nebraska",
+        "Nevada",
+        "New Hampshire",
+        "New Jersey",
+        "New Mexico",
+        "New York",
+        "North Carolina",
+        "North Dakota",
+        "Ohio",
+        "Oklahoma",
+        "Oregon",
+        "Pennsylvania",
+        "Rhode Island",
+        "South Carolina",
+        "South Dakota",
+        "Tennessee",
+        "Texas",
+        "Utah",
+        "Vermont",
+        "Virginia",
+        "Washington",
+        "West Virginia",
+        "Wisconsin",
+        "Wyoming",
+      ],
+      required: [true, "Please provide the state this program is in."],
+    },
+    description: {
+      type: String,
+    },
+    faq: [FaqSchema],
+    assistantID: {
+      type: String,
+    },
+    vectorStoreID: {
+      type: String,
+    },
+    testOpenAI: [TestOpenAISchema],
+    // Add the embedding collection name into Program every time we create an embedding collection
+    // This embedding is only used for Langchain RAG system, not openAI Assistant
+    embeddingCollection: {
+      type: String,
+    },
+    status: {
+        type: String,
+    },
+    dataset: [{ type: Schema.Types.ObjectId, ref: "Dataset" }],
+    files: [{ type: Schema.Types.ObjectId, ref: "Files" }],
+  },
+  { timestamps: true, versionKey: false }
+);
+
+module.exports =
+  mongoose.models.Program || mongoose.model("Program", ProgramSchema);

package/models/QA.js

@@ -0,0 +1,69 @@
+const mongoose = require('mongoose');
+const { Schema, Types } = mongoose;
+
+const QASchema = new Schema(
+  {
+    programID: {
+      type: Types.ObjectId,
+      ref: "Program",
+    },
+    question: {
+      type: String,
+    },
+    answer: {
+      type: String,
+    },
+    assistantModel: {
+      type: String,
+      enum: [
+        "langchain-assistant",
+        "openAI-assistant"
+      ],
+    },
+    instruction: {
+      type: String,
+    },
+    modelName: {
+      type: String,
+      enum: [
+        "gpt-4o",
+        "gpt-4-turbo-preview",
+        "gpt-4-turbo",
+        "gpt-4-turbo-2024-04-09",
+        "gpt-4-0125-preview",
+        "gpt-4-1106-preview",
+        "gpt-4",
+        "gpt-4-0613",
+        "gpt-4-32k",
+        "gpt-4-32k-0613",
+        "gpt-3.5-turbo-0125",
+        "gpt-3.5-turbo",
+        "gpt-3.5-turbo-1106",
+        "gpt-3.5-turbo-instruct"
+      ],
+    },
+    score: {
+      type: Number,
+      enum: [
+        0,
+        1,
+        2,
+        3,
+        4,
+        5
+      ],
+    },
+    messageId: {
+      type: String,
+    },
+    batchId: {
+      type: String,
+    },
+  },
+  {
+    timestamps: true,
+    versionKey: false,
+  }
+);
+
+module.exports = mongoose.models.QA || mongoose.model("QA", QASchema);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ben-ai-models-registry",
-  "version": "1.0.7",
+  "version": "1.0.8",
   "description": "",
   "main": "index.js",
   "scripts": {