save course chat save status to db

Author: mluukkai

src/server/db/migrations/20250123_00_saving_to_chat_instances.ts

@@ -0,0 +1,13 @@
+import { DataTypes } from 'sequelize'
+
+import { Migration } from '../connection'
+
+export const up: Migration = async ({ context: queryInterface }) => {
+  await queryInterface.addColumn('chat_instances', 'save_discussions', {
+    type: DataTypes.BOOLEAN,
+  })
+}
+
+export const down: Migration = async ({ context: queryInterface }) => {
+  await queryInterface.removeColumn('chat_instances', 'save_discussions')
+}

src/server/db/models/chatInstance.ts

@@ -34,6 +34,8 @@ class ChatInstance extends Model<
   declare courseUnitRealisationTypeUrn: string | null
 
   declare courseUnits: CreationOptional<CourseUnit[]>
+
+  declare saveDiscussions: boolean
 }
 
 ChatInstance.init(
@@ -93,6 +95,9 @@ ChatInstance.init(
       allowNull: true,
       defaultValue: [],
     },
+    saveDiscussions: {
+      type: DataTypes.BOOLEAN,
+    },
   },
   {
     underscored: true,

src/server/routes/openai.ts

@@ -21,7 +21,7 @@ import getEncoding from '../util/tiktoken'
 import logger from '../util/logger'
 import { inProduction, DEFAULT_TOKEN_LIMIT, FREE_MODEL } from '../../config'
 import { pdfToText } from '../util/pdfToText'
-import { Discussion } from '../db/models'
+import { Discussion, ChatInstance } from '../db/models'
 
 const openaiRouter = express.Router()
 
@@ -162,9 +162,15 @@ openaiRouter.post('/stream', upload.single('file'), async (r, res) => {
     courseId,
   })
 
+  const course = await ChatInstance.findOne({
+    where: { courseId },
+  })
+
+  console.log('-->', course.saveDiscussions)
+
   const consentToSave =
-    ['testUser', 'mluukkai', 'admini2'].includes(user.username) &&
-    courseId === 'sandbox'
+    course.saveDiscussions &&
+    ['testUser', 'mluukkai', 'admini2'].includes(user.username)
 
   // eslint-disable-next-line no-console
   console.log('consentToSave', consentToSave, user.username)