From abb7c5308b1b18ee6548cfde063652152a9476ef Mon Sep 17 00:00:00 2001 From: Maxwell-Lindsey Date: Tue, 27 Feb 2024 12:38:02 -0600 Subject: [PATCH] created enable_doc_group function --- ai_ta_backend/vector_database.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/ai_ta_backend/vector_database.py b/ai_ta_backend/vector_database.py index 4dfb7ae1..d40e7da2 100644 --- 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).