Make aila required in Aila Rag and pass chatId and userId

oaknational · Aug 28, 2024 · 206f72b · 206f72b
1 parent 7308bd3
commit 206f72b
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/packages/aila/src/features/rag/AilaRag.ts b/packages/aila/src/features/rag/AilaRag.ts
@@ -8,20 +8,23 @@ import { LooseLessonPlan } from "../../protocol/schema";
 import { minifyLessonPlanForRelevantLessons } from "../../utils/lessonPlan/minifyLessonPlanForRelevantLessons";
 
 export class AilaRag {
-  private _aila?: AilaServices;
+  private _aila: AilaServices;
   private _rag: RAG;
   private _prisma: PrismaClientWithAccelerate;
 
   constructor({
     aila,
     prisma,
   }: {
-    aila?: AilaServices;
+    aila: AilaServices;
     prisma?: PrismaClientWithAccelerate;
   }) {
     this._aila = aila;
     this._prisma = prisma ?? globalPrisma;
-    this._rag = new RAG(this._prisma);
+    this._rag = new RAG(this._prisma, {
+      userId: aila.userId,
+      chatId: aila.chatId,
+    });
   }
 
   public async fetchRagContent({