Update auditqa/process_chunks.py

auditqa/process_chunks.py

@@ -69,12 +69,13 @@ def load_chunks():
                 # add metadata information 
                 chunks_list = []
                 for doc in doc_processed:
-                    chunks_list.append(Document(page_content=doc['content'], 
+                    chunks_list.append(Document(page_content= doc['content'], 
                              metadata={"source": category,
                                       "subtype":subtype,
                                       "year":file[-4:],
                                       "filename":file,
-                                      "page":doc['metadata']['page']}))
+                                      "page":doc['metadata']['page'],
+                                      "headings":doc['metadata']['headings']}))
 
                 all_documents[category].append(chunks_list)