up

crazy_functions/rag_fns/arxiv_fns/arxiv_splitter.py

@@ -65,9 +65,9 @@ def save_fragments_to_file(fragments: List[SectionFragment], output_dir: str = "
                 f.write(f"\n### Abstract\n{fragments[0].abstract}\n")
 
         # Write section tree if available
-        if fragments and fragments[0].section_tree:
+        if fragments and fragments[0].catalogs:
             f.write("\n## Section Tree\n")
-            f.write(fragments[0].section_tree)
+            f.write(fragments[0].catalogs)
 
         # Generate table of contents
         f.write("\n## Table of Contents\n")
@@ -689,7 +689,7 @@ class ArxivSplitter:
         try:
             return all([
                 fragment.title.strip(),
-                fragment.section_tree.strip(),
+                fragment.catalogs.strip(),
                 fragment.current_section.strip(),
                 fragment.content.strip() or fragment.bibliography.strip()
             ])

crazy_functions/rag_fns/arxiv_fns/section_fragment.py

@@ -5,7 +5,7 @@ class SectionFragment:
     """Arxiv论文片段数据类"""
     title: str  # 文件路径
     abstract: str  # 论文摘要
-    section_tree: str # 文章各章节的目录结构
+    catalogs: str # 文章各章节的目录结构
     arxiv_id: str = ""  # 添加 arxiv_id 属性
     current_section: str = "Introduction" # 当前片段所属的section或者subsection或者孙subsubsection名字
     content: str = '' #当前片段的内容
@@ -15,3 +15,4 @@ class SectionFragment:
 
 
 
+