fix headings bug (from DS4SD/docling#501)

Co-authored-by: Ben Rood <[email protected]> Signed-off-by: Panos Vagenas <[email protected]>
DS4SD · Dec 5, 2024 · a9f08e9 · a9f08e9
1 parent 0e4a754
commit a9f08e9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/docling_core/transforms/chunker/token_aware_chunker.py b/docling_core/transforms/chunker/token_aware_chunker.py
@@ -234,8 +234,8 @@ def _merge_chunks_with_matching_metadata(self, chunks: list[DocChunk]):
                 else:
                     new_meta = DocMeta(
                         doc_items=window_items,
-                        headings=headings_and_captions[0],
-                        captions=headings_and_captions[1],
+                        headings=current_headings_and_captions[0],
+                        captions=current_headings_and_captions[1],
                     )
                     new_chunk = DocChunk(
                         text=window_text,