diff --git a/crazy_functions/Arxiv_论文对话.py b/crazy_functions/Arxiv_论文对话.py index 771caae5..f8f7f304 100644 --- a/crazy_functions/Arxiv_论文对话.py +++ b/crazy_functions/Arxiv_论文对话.py @@ -176,7 +176,7 @@ class ArxivRagWorker: f"Paper Title: {first_fragment.title}\n" f"ArXiv ID: {first_fragment.arxiv_id}\n" f"Abstract: {first_fragment.abstract}\n" - f"Section Tree:{first_fragment.section_tree}\n" + f"Table of contents:{first_fragment.catalogs}\n" f"Type: OVERVIEW" ) } diff --git a/crazy_functions/rag_fns/arxiv_fns/arxiv_splitter.py b/crazy_functions/rag_fns/arxiv_fns/arxiv_splitter.py index 9ddb9733..f9d28c44 100644 --- a/crazy_functions/rag_fns/arxiv_fns/arxiv_splitter.py +++ b/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() ]) diff --git a/crazy_functions/rag_fns/arxiv_fns/section_fragment.py b/crazy_functions/rag_fns/arxiv_fns/section_fragment.py index 21f79e6a..f04cdd55 100644 --- a/crazy_functions/rag_fns/arxiv_fns/section_fragment.py +++ b/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: +