created enable_doc_group function

UIUC-Chatbot · Feb 27, 2024 · abb7c53 · abb7c53
1 parent d266dfc
commit abb7c53
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/ai_ta_backend/vector_database.py b/ai_ta_backend/vector_database.py
@@ -1692,6 +1692,12 @@ def disable_doc_group(self, course_name: str, doc_group_name: str):
       """
       self.set_enabled_doc_group(course_name, doc_group_name, False)
 
+  def enable_doc_group(self, course_name: str, doc_group_name: str):
+    """
+    Enable a document group.
+    """
+    self.set_enabled_doc_group(course_name, doc_group_name, True)
+
   def add_documents_to_doc_group(self, course_name: str, docs: MaterialDocument | list[MaterialDocument]):
     """
     Add document group name to documents (in both supabase and qdrant).