camel case sql query response

oaknational · Dec 12, 2024 · dba54e4 · dba54e4
1 parent 1fd47b2
commit dba54e4
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 6 deletions.
diff --git a/packages/rag/index.ts b/packages/rag/index.ts
@@ -40,6 +40,10 @@ export async function getRelevantLessonPlans({
   });
   log.info(`Got ${vectorSearchResults.length} search results`);
 
+  if (vectorSearchResults.length === 0) {
+    return [];
+  }
+
   log.info(`Reranking lesson plans`);
   const rerankedResults = await rerankResults({
     cohereClient,

diff --git a/packages/rag/lib/search.ts b/packages/rag/lib/search.ts
@@ -44,12 +44,12 @@ export async function vectorSearch({
   const startAt = new Date();
   const response = await prisma.$queryRaw<RagLessonPlanResult[]>`
           SELECT
-            rag_lesson_plan_id as ragLessonPlanId,
-            oak_lesson_id as oakLessonId,
-            oak_lesson_slug as oakLessonSlug,
-            lesson_plan as lessonPlan,
-            key as matchedKey,
-            value_text as matchedValue,
+            rag_lesson_plan_id as "ragLessonPlanId",
+            oak_lesson_id as "oakLessonId",
+            oak_lesson_slug as "oakLessonSlug",
+            lesson_plan as "lessonPlan",
+            key as "matchedKey",
+            value_text as "matchedValue",
             embedding <-> ${queryEmbedding}::vector as distance
           FROM rag.rag_lesson_plan_parts JOIN rag.rag_lesson_plans ON rag_lesson_plan_id = rag_lesson_plans.id
           WHERE rag_lesson_plans.is_published = true