fix prompt rag association including

Author: Veikkosuhonen

src/client/components/ChatV2/PromptState.tsx

@@ -103,7 +103,7 @@ export const PromptStateProvider: React.FC<{
       }
       sync()
     }
-  }, [isPromptEditable, activePrompt, setActivePrompt])
+  }, [activePrompt?.id])
 
   // Just the analytics dispatch.
   useEffect(() => {

src/server/db/models/index.ts

@@ -60,6 +60,8 @@ RagFile.belongsTo(RagIndex, { as: 'ragIndex' })
 
 RagIndex.hasMany(RagFile, { as: 'ragFiles' })
 
+RagIndex.hasMany(Prompt, { as: 'prompts' })
+
 RagFile.belongsTo(User, { as: 'user' })
 
 /**

src/server/db/models/prompt.ts

@@ -2,7 +2,7 @@ import { type CreationOptional, DataTypes, type InferAttributes, type InferCreat
 
 import type { CustomMessage } from '../../types'
 import { sequelize } from '../connection'
-import type RagIndex from './ragIndex'
+import RagIndex from './ragIndex'
 
 export const PromptTypeValues = ['CHAT_INSTANCE', 'PERSONAL'] as const
 export type PromptType = (typeof PromptTypeValues)[number]
@@ -58,6 +58,10 @@ Prompt.init(
     ragIndexId: {
       type: DataTypes.INTEGER,
       allowNull: true,
+      references: {
+        model: RagIndex,
+        key: 'id',
+      },
     },
     systemMessage: {
       type: DataTypes.TEXT,

src/server/db/models/ragIndex.ts

@@ -48,6 +48,7 @@ RagIndex.init(
   },
   {
     underscored: true,
+    tableName: 'rag_indices',
     sequelize,
   },
 )

src/server/routes/ai/v3.ts

@@ -119,7 +119,6 @@ router.post('/stream', upload.single('file'), async (r, res) => {
         {
           model: RagIndex,
           as: 'ragIndex',
-          where: courseId ? { courseId } : {},
         },
       ],
     })