fix: fuzzy key stage from correct table (#125)

oaknational · Sep 12, 2024 · f7740ac · f7740ac
1 parent 68b9cd1
commit f7740ac
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/packages/core/src/rag/index.ts b/packages/core/src/rag/index.ts
@@ -616,7 +616,7 @@ Thank you and happy classifying!`;
         this._chatMeta,
       );
       if (categorisation.keyStage) {
-        foundKeyStage = await this.prisma.subject.findFirst({
+        foundKeyStage = await this.prisma.keyStage.findFirst({
           where: {
             slug: categorisation.keyStage,
           },