fixed Qdrant payload for add_documents_to_doc_group

UIUC-Chatbot · Feb 26, 2024 · c635c7e · c635c7e
1 parent 055d960
commit c635c7e
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/ai_ta_backend/vector_database.py b/ai_ta_backend/vector_database.py
@@ -1712,6 +1712,9 @@ def add_documents_to_doc_group(self, course_name: str, docs: MaterialDocument |
             collection_name=os.environ['QDRANT_COLLECTION_NAME'],
             payload={
                 "doc_group": doc.doc_groups,
+                "course_name": course_name,
+                "s3_path": doc.s3_path,
+                "url": doc.url,
             },
             points=models.Filter(must=[
                 models.FieldCondition(