diff --git a/crazy_functions/rag_fns/arxiv_fns/arxiv_splitter.py b/crazy_functions/rag_fns/arxiv_fns/arxiv_splitter.py index f9d28c44..b96f173f 100644 --- a/crazy_functions/rag_fns/arxiv_fns/arxiv_splitter.py +++ b/crazy_functions/rag_fns/arxiv_fns/arxiv_splitter.py @@ -625,7 +625,7 @@ class ArxivSplitter: base_fragment_template = { 'title': doc_structure.title, 'abstract': doc_structure.abstract, - 'section_tree': section_tree, + 'catalogs': section_tree, 'arxiv_id': arxiv_id }