up

2024-11-17 17:36:01 +00:00
parent cbef9a908c
commit f8b60870e9
3 changed files with 72 additions and 40 deletions
--- a/crazy_functions/rag_fns/arxiv_fns/arxiv_fragment.py
+++ b/crazy_functions/rag_fns/arxiv_fns/arxiv_fragment.py
@@ -14,6 +14,7 @@ class ArxivFragment:
    section: str  # 所属章节
    is_appendix: bool  # 是否是附录
    importance: float = 1.0  # 重要性得分
+    arxiv_id: str = ""  # 添加 arxiv_id 属性

    @staticmethod
    def merge_segments(seg1: 'ArxivFragment', seg2: 'ArxivFragment') -> 'ArxivFragment':