feat: AiAssistantModel

apps/app/src/features/openai/server/models/ai-assistant.ts

@@ -0,0 +1,132 @@
+import { type IGrantedGroup, GroupType } from '@growi/core';
+import type mongoose from 'mongoose';
+import { type Model, type Document, Schema } from 'mongoose';
+
+import { getOrCreateModel } from '~/server/util/mongoose-utils';
+
+/*
+*  Objects
+*/
+const AiAssistantType = {
+  KNOWLEDGE: 'knowledge',
+  // EDITOR: 'editor',
+  // LEARNING: 'learning',
+} as const;
+
+const AiAssistantSharingScope = {
+  PUBLIC: 'public',
+  ONLY_ME: 'onlyMe',
+  USER_GROUP: 'userGroup',
+} as const;
+
+const AiAssistantLearningScope = {
+  PUBLIC: 'public',
+  ONLY_ME: 'onlyMe',
+  USER_GROUP: 'userGroup',
+} as const;
+
+
+/*
+*  Interfaces
+*/
+type AiAssistantType = typeof AiAssistantType[keyof typeof AiAssistantType];
+type AiAssistantSharingScope = typeof AiAssistantSharingScope[keyof typeof AiAssistantSharingScope];
+type AiAssistantLearningScope = typeof AiAssistantLearningScope[keyof typeof AiAssistantLearningScope];
+
+interface AiAssistant {
+  name: string;
+  description?: string
+  instruction?: string
+  vectorStoreId: string // VectorStoreId of OpenAI Specify (https://platform.openai.com/docs/api-reference/vector-stores/object)
+  types: AiAssistantType[]
+  grantedGroups: IGrantedGroup[];
+  pages: mongoose.Types.ObjectId[]
+  sharingScope: AiAssistantSharingScope
+  learningScope: AiAssistantLearningScope
+  isDeleted: boolean
+}
+
+interface AiAssistantDocument extends AiAssistant, Document {}
+
+type AiAssistantModel = Model<AiAssistantDocument>
+
+
+/*
+ * Schema Definition
+ */
+const schema = new Schema<AiAssistantDocument>(
+  {
+    name: {
+      type: String,
+      required: true,
+    },
+    description: {
+      type: String,
+    },
+    instruction: {
+      type: String,
+    },
+    vectorStoreId: {
+      type: String,
+      required: true,
+    },
+    types: [{
+      type: String,
+      enum: Object.values(AiAssistantType),
+      required: true,
+    }],
+    pages: [{
+      type: Schema.Types.ObjectId,
+      ref: 'Page',
+      required: true,
+    }],
+    grantedGroups: {
+      type: [{
+        type: {
+          type: String,
+          enum: Object.values(GroupType),
+          required: true,
+          default: 'UserGroup',
+        },
+        item: {
+          type: Schema.Types.ObjectId,
+          refPath: 'grantedGroups.type',
+          required: true,
+          index: true,
+        },
+      }],
+      validate: [function(arr: IGrantedGroup[]): boolean {
+        if (arr == null) return true;
+        const uniqueItemValues = new Set(arr.map(e => e.item));
+        return arr.length === uniqueItemValues.size;
+      }, 'grantedGroups contains non unique item'],
+      default: [],
+      required: true,
+    },
+    sharingScope: {
+      type: String,
+      enum: Object.values(AiAssistantSharingScope),
+      required: true,
+    },
+    learningScope: {
+      type: String,
+      enum: Object.values(AiAssistantLearningScope),
+      required: true,
+    },
+    isDeleted: {
+      type: Boolean,
+      default: false,
+      required: true,
+    },
+  },
+  {
+    timestamps: true,
+  },
+);
+
+schema.methods.markAsDeleted = async function(): Promise<void> {
+  this.isDeleted = true;
+  await this.save();
+};
+
+export default getOrCreateModel<AiAssistantDocument, AiAssistantModel>('AiAssistant', schema);