added null check and handling for null elements in doc_groups list

UIUC-Chatbot · Feb 27, 2024 · 5d1168a · 5d1168a
1 parent 49f354d
commit 5d1168a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ai_ta_backend/main.py b/ai_ta_backend/main.py
@@ -739,7 +739,7 @@ def appendDocumentToGroup():
     doc_group = data['doc_group']
 
     document_data = data.get('document', {})
-    document_data['doc_groups'] = document_data.get('doc_groups', [])
+    document_data['doc_groups'] = [group for group in document_data.get('doc_groups', []) if group is not None]
 
     try:
         document = MaterialDocument(**document_data)
@@ -766,7 +766,7 @@ def removeDocumentFromGroup():
     doc_group = data['doc_group']
 
     document_data = data.get('document', {})
-    document_data['doc_groups'] = document_data.get('doc_groups', [])
+    document_data['doc_groups'] = [group for group in document_data.get('doc_groups', []) if group is not None]
 
     try:
         document = MaterialDocument(**document_data)