Add tool calls for updateTimetable and deleteTimetable

Author: kevin-lann

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

@@ -1,7 +1,8 @@
 import { supabase } from "../db/setupDb";
 import { Request } from "express";
 
-export type FunctionNames = "getTimetables"; // Add all possible function names here
+// Add all possible function names here
+export type FunctionNames = "getTimetables" | "updateTimetable" | "deleteTimetable"; 
 
 type AvailableFunctions = {
   [K in FunctionNames]: (args: any, req: Request) => Promise<any>;
@@ -40,4 +41,124 @@ export const availableFunctions: AvailableFunctions = {
       return { status: 400, error: error };
     }
   },
+  updateTimetable: async (args: any, req: Request) => {
+    try {
+      const { id, timetable_title, semester  } = args;
+
+      if (!timetable_title && !semester) {
+        return {
+          status: 400,
+          error:
+            "New timetable title or semester is required when updating a timetable",
+        };
+      }
+
+      //Retrieve the authenticated user
+      const user_id = (req as any).user.id;
+
+      //Retrieve users allowed to access the timetable
+      const { data: timetableUserData, error: timetableUserError } =
+        await supabase
+          .schema("timetable")
+          .from("timetables")
+          .select("*")
+          .eq("id", id)
+          .eq("user_id", user_id)
+          .maybeSingle();
+
+      const timetable_user_id = timetableUserData?.user_id;
+
+      if (timetableUserError)
+        return {status: 400, error: timetableUserError.message };
+
+      //Validate timetable validity:
+      if (!timetableUserData || timetableUserData.length === 0) {
+        return {status: 404, error: "Calendar id not found" };
+      }
+
+      //Validate user access
+      if (user_id !== timetable_user_id) {
+        return { status: 401,  error: "Unauthorized access to timetable events" };
+      }
+
+      let updateData: any = {};
+      if (timetable_title) updateData.timetable_title = timetable_title;
+      if (semester) updateData.semester = semester;
+
+      //Update timetable title, for authenticated user only
+      let updateTimetableQuery = supabase
+        .schema("timetable")
+        .from("timetables")
+        .update(updateData)
+        .eq("id", id)
+        .eq("user_id", user_id)
+        .select();
+
+      const { data: timetableData, error: timetableError } =
+        await updateTimetableQuery;
+
+      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: 500,  error: error };
+    }
+  },
+  deleteTimetable: async (args: any, req: Request) => {
+    try {
+      const { id } = args;
+
+      // Retrieve the authenticated user
+      const user_id = (req as any).user.id;
+
+      //Retrieve users allowed to access the timetable
+      const { data: timetableUserData, error: timetableUserError } =
+        await supabase
+          .schema("timetable")
+          .from("timetables")
+          .select("*")
+          .eq("id", id)
+          .eq("user_id", user_id)
+          .maybeSingle();
+      const timetable_user_id = timetableUserData?.user_id;
+
+      if (timetableUserError)
+        return { status: 400, error: timetableUserError.message };
+
+      //Validate timetable validity:
+      if (!timetableUserData || timetableUserData.length === 0) {
+        return { status: 404,  error: "Calendar id not found" };
+      }
+
+      //Validate user access
+      if (user_id !== timetable_user_id) {
+        return  { status: 401, error: "Unauthorized access to timetable events" };
+      }
+
+      // Delete only if the timetable belongs to the authenticated user
+      let deleteTimetableQuery = supabase
+        .schema("timetable")
+        .from("timetables")
+        .delete()
+        .eq("id", id)
+        .eq("user_id", user_id);
+
+      const { error: timetableError } = await deleteTimetableQuery;
+
+      if (timetableError)
+        return { status: 400, error: timetableError.message };
+
+      return { status: 200, data: "Timetable successfully deleted"};
+    } catch (error) {
+      return { status: 500, error: error };
+    }
+  }
 };

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

@@ -50,3 +50,5 @@ namespaceToMinResults.set("programs", 5);
 export const CHATBOT_MEMORY_THRESHOLD = 3;
 
 export const CHATBOT_TIMETABLE_CMD = "/timetable";
+
+export const CHATBOT_TOOL_CALL_MAX_STEPS = 3;

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

@@ -26,6 +26,7 @@ import {
 import {
   CHATBOT_MEMORY_THRESHOLD,
   CHATBOT_TIMETABLE_CMD,
+  CHATBOT_TOOL_CALL_MAX_STEPS,
 } from "../constants/constants";
 import { namespaceToMinResults } from "../constants/constants";
 import OpenAI from "openai";
@@ -254,8 +255,30 @@ export const chat = asyncHandler(async (req: Request, res: Response) => {
               return await availableFunctions.getTimetables(args, req);
             },
           }),
+          updateTimetable: tool({
+            description:
+              "Update a user's timetable by title and/or semester",
+            parameters: z.object({
+              id: z.number().positive(),
+              timetable_title: z.string().optional(),
+              semester: z.enum(["Fall 2025", "Summer 2025", "Winter 2026"]).optional(),
+            }),
+            execute: async (args) => {
+              return await availableFunctions.updateTimetable(args, req);
+            }
+          }),
+          deleteTimetable: tool({
+            description:
+              "Delete a user's timetable",
+            parameters: z.object({
+              id: z.number().positive(),
+            }),
+            execute: async (args) => {
+              return await availableFunctions.deleteTimetable(args, req)
+            } 
+          })
         },
-        maxSteps: 3, // Controls how many back and forths the model can take with user or calling multiple tools
+        maxSteps: CHATBOT_TOOL_CALL_MAX_STEPS, // Controls how many back and forths the model can take with user or calling multiple tools
         experimental_repairToolCall: async ({
           toolCall,
           tools,