Setup function calling

Author: kevin-lann

course-matrix/backend/package-lock.json

@@ -0,0 +1,41 @@
+import { supabase } from "../db/setupDb";
+import { Request } from "express";
+
+export type FunctionNames = "getTimetables"; // Add all possible function names here
+
+type AvailableFunctions = {
+  [K in FunctionNames]: (args: any, req: Request) => Promise<any>;
+};
+
+// Functions used for OpenAI function calling
+export const availableFunctions: AvailableFunctions = {
+  getTimetables: async (args: any, req: Request) => {
+    try {
+      //Retrieve user_id
+      const user_id = (req as any).user.id;
+
+      //Retrieve user timetable item based on user_id
+      let timeTableQuery = supabase
+        .schema("timetable")
+        .from("timetables")
+        .select()
+        .eq("user_id", user_id);
+      const { data: timetableData, error: timetableError } =
+        await timeTableQuery;
+
+      if (timetableError) return { status: 400, error: timetableError.message };
+
+      // If no records were updated due to non-existence timetable or it doesn't belong to the user.
+      if (!timetableData || timetableData.length === 0) {
+        return {
+          status: 404,
+          error: "Timetable not found or you are not authorized to update it",
+        };
+      }
+
+      return { status: 200, data: timetableData };
+    } catch (error) {
+      return { status: 400, error: error };
+    }
+  },
+};

course-matrix/backend/src/constants/availableFunctions.ts

@@ -48,3 +48,5 @@ namespaceToMinResults.set("programs", 5);
 
 // Consider the last X messages in history to influence vector DB query
 export const CHATBOT_MEMORY_THRESHOLD = 3;
+
+export const CHATBOT_TIMETABLE_CMD = "/timetable";