Merge branch 'kl/scrum-41-ai-timetable-generate' of https://github.com/UTSC-CSCC01-Software-Engineering-I/term-group-project-c01w25-project-course-matrix into kl/scrum-41-ai-timetable-generate

Author: kevin-lann

course-matrix/backend/src/controllers/aiController.ts

@@ -43,7 +43,7 @@ const pinecone = new Pinecone({
 });
 
 const index: Index<RecordMetadata> = pinecone.Index(
-  process.env.PINECONE_INDEX_NAME!
+  process.env.PINECONE_INDEX_NAME!,
 );
 
 console.log("Connected to OpenAI API");
@@ -100,7 +100,7 @@ function analyzeQuery(query: string): {
       "prerequisites",
       "corequisites",
       "departments",
-      "programs"
+      "programs",
     );
   }
 
@@ -118,7 +118,7 @@ async function searchSelectedNamespaces(
   query: string,
   k: number,
   namespaces: string[],
-  filters?: Object
+  filters?: Object,
 ): Promise<Document[]> {
   let allResults: Document[] = [];
 
@@ -140,7 +140,7 @@ async function searchSelectedNamespaces(
       const results = await namespaceStore.similaritySearch(
         query,
         Math.max(k, namespaceToMinResults.get(namespace)),
-        namespace === "courses_v3" ? filters : undefined
+        namespace === "courses_v3" ? filters : undefined,
       );
       console.log(`Found ${results.length} results in namespace: ${namespace}`);
       allResults = [...allResults, ...results];
@@ -161,7 +161,7 @@ async function searchSelectedNamespaces(
 // Reformulate user query to make more concise query to database, taking into consideration context
 async function reformulateQuery(
   latestQuery: string,
-  conversationHistory: any[]
+  conversationHistory: any[],
 ): Promise<string> {
   try {
     const openai2 = new OpenAI({
@@ -261,7 +261,7 @@ function includeFilters(query: string) {
       if (keywords.some((keyword) => lowerQuery.includes(keyword))) {
         relaventBreadthRequirements.push(convertBreadthRequirement(namespace));
       }
-    }
+    },
   );
 
   Object.entries(YEAR_LEVEL_KEYWORDS).forEach(([namespace, keywords]) => {
@@ -412,7 +412,7 @@ export const chat = asyncHandler(async (req: Request, res: Response) => {
       // Use GPT-4o to reformulate the query based on conversation history
       const reformulatedQuery = await reformulateQuery(
         latestMessage,
-        conversationHistory.slice(-CHATBOT_MEMORY_THRESHOLD) // last K messages
+        conversationHistory.slice(-CHATBOT_MEMORY_THRESHOLD), // last K messages
       );
       console.log(">>>> Original query:", latestMessage);
       console.log(">>>> Reformulated query:", reformulatedQuery);
@@ -426,8 +426,8 @@ export const chat = asyncHandler(async (req: Request, res: Response) => {
       if (requiresSearch) {
         console.log(
           `Query requires knowledge retrieval, searching namespaces: ${relevantNamespaces.join(
-            ", "
-          )}`
+            ", ",
+          )}`,
         );
 
         const filters = includeFilters(reformulatedQuery);
@@ -438,7 +438,7 @@ export const chat = asyncHandler(async (req: Request, res: Response) => {
           reformulatedQuery,
           3,
           relevantNamespaces,
-          Object.keys(filters).length === 0 ? undefined : filters
+          Object.keys(filters).length === 0 ? undefined : filters,
         );
         // console.log("Search Results: ", searchResults);
 
@@ -448,7 +448,7 @@ export const chat = asyncHandler(async (req: Request, res: Response) => {
         }
       } else {
         console.log(
-          "Query does not require knowledge retrieval, skipping search"
+          "Query does not require knowledge retrieval, skipping search",
         );
       }
 
@@ -513,15 +513,15 @@ export const testSimilaritySearch = asyncHandler(
     if (requiresSearch) {
       console.log(
         `Query requires knowledge retrieval, searching namespaces: ${relevantNamespaces.join(
-          ", "
-        )}`
+          ", ",
+        )}`,
       );
 
       // Search only the relevant namespaces
       const searchResults = await searchSelectedNamespaces(
         message,
         3,
-        relevantNamespaces
+        relevantNamespaces,
       );
       console.log("Search Results: ", searchResults);
 
@@ -531,11 +531,11 @@ export const testSimilaritySearch = asyncHandler(
       }
     } else {
       console.log(
-        "Query does not require knowledge retrieval, skipping search"
+        "Query does not require knowledge retrieval, skipping search",
       );
     }
 
     console.log("CONTEXT: ", context);
     res.status(200).send(context);
-  }
+  },
 );