added chunk index to supabase metadata

UIUC-Chatbot · Oct 26, 2023 · 936ee47 · 936ee47
1 parent daa0fa9
commit 936ee47
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/ai_ta_backend/vector_database.py b/ai_ta_backend/vector_database.py
@@ -771,6 +771,7 @@ def split_and_upload(self, texts: List[str], metadatas: List[Dict[str, Any]]):
           "text": context.page_content,
           "pagenumber": context.metadata.get('pagenumber'),
           "timestamp": context.metadata.get('timestamp'),
+          "chunk_index": context.metadata.get('chunk_index'),
           "embedding": embeddings_dict[context.page_content]
       } for context in contexts]