modified function to create convo map from scratch

UIUC-Chatbot · Mar 27, 2024 · ea36017 · ea36017
1 parent fc69d4c
commit ea36017
Show file tree

Hide file tree

Showing 2 changed files with 374 additions and 663 deletions.
diff --git a/ai_ta_backend/database/sql.py b/ai_ta_backend/database/sql.py
@@ -75,7 +75,7 @@ def getAllFromTableForDownloadType(self, course_name: str, download_type: str, f
 
   def getAllConversationsBetweenIds(self, course_name: str, first_id: int, last_id: int, limit: int = 50):
     if last_id == 0:
-      return self.supabase_client.table("llm-convo-monitor").select("*").eq("course_name", course_name).gte(
+      return self.supabase_client.table("llm-convo-monitor").select("*").eq("course_name", course_name).gt(
           'id', first_id).order('id', desc=False).limit(limit).execute()
     else:
       return self.supabase_client.table("llm-convo-monitor").select("*").eq("course_name", course_name).gte(
@@ -100,3 +100,7 @@ def getDocMapFromProjects(self, course_name: str):
 
   def getConvoMapFromProjects(self, course_name: str):
     return self.supabase_client.table("projects").select("convo_map_id").eq("course_name", course_name).execute()
+
+  def updateProjects(self, course_name: str, data: dict):
+    return self.supabase_client.table("projects").update(data).eq("course_name", course_name).execute()
+