大模型RAG文档切分进阶实战
总体目标与原则
数据清洗的核心目标是:
- 去除无关内容和噪声
-
- HTML/XML标签:div, span, p 等
- Markdown符号:粗体, [链接], #标题 等
- 代码片段:无意义的程序代码块
- 标准化文本格式
-
- 修复断行问题:PDF转换经常把一句话拆成多行
- 统一换行符:Windows(\r\n), Unix(\n), Mac(\r)
- 合并短行:把被错误分割的短行重新合并
- 标准化空格:多个连续空格→单个空格
- 提高切分边界识别的准确性
不同类型数据的清洗步骤
纯文本文档清洗
import re
# text文本处理
def clean_text(text):
# 去除多余空白
text = re.sub(r'\s+', ' ', text)
# 修复断开的单词
text = re.sub(r'(\w+)-\s+(\w+)', r'\1\2', text)
# 标准化引号
text = text.replace('"', '"').replace('"', '"')
return text.strip()
# html文本处理
def clean_html(text):
return re.sub(r'<.*?>', '', text) # 移除HTML标签
# markdown文本处理
def clean_markdown(text):
text = re.sub(r'\*\*(.*?)\*\*', r'\1', text) # 移除粗体标记但保留文字
text = re.sub(r'\[(.*?)\]\(.*?\)', r'\1', text) # 移除链接标记但保留文字
return text
表格数据清洗
# 安装pandas
# pip install pandas
import pandas as pd
def clean_table(df):
if type(df) != pd.DataFrame:
df = pd.DataFrame(df)
# 去除完全空白的行和列
df = df.dropna(how='all').dropna(axis=1, how='all')
# 填充NaN值
df = df.fillna('')
# 删除完全重复的行 (不要使用 inplace=True,它会返回 None)
df = df.drop_duplicates()
# 标准化列名
df.columns = [str(col).strip() for col in df.columns]
return df
图像文档清洗
图像文档(如扫描的PDF)需要OCR预处理:
- 二值化、降噪、倾斜校正
#安装依赖包
# pip install cv2 matplotlib
import cv2
from matplotlib import pyplot as plt
# 显示图像的函数
def show_image(img, title="Image", cmap=None):
plt.figure(figsize=(4, 2))
if cmap:
plt.imshow(img, cmap=cmap)
else:
plt.imshow(img)
plt.title(title)
plt.axis('off')
plt.show()
def clean_image(img_path):
"""
图像降噪 - 去除扫描文档中的噪声点
"""
# 读取图像
img = cv2.imread(img_path)
img_rgb = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
# 转换为灰度图
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
print("原始图像:")
show_image(img_rgb, "Original Image")
# 方法1: 高斯模糊去噪
gaussian_denoised = cv2.GaussianBlur(gray, (5, 5), 0)
# 方法2: 中值滤波去噪 (对椒盐噪声效果好)
median_denoised = cv2.medianBlur(gray, 3)
# 方法3: 双边滤波 (保持边缘)
bilateral_denoised = cv2.bilateralFilter(gray, 9, 75, 75)
print("高斯模糊去噪:")
show_image(gaussian_denoised, "Gaussian Denoised", cmap='gray')
print("中值滤波去噪:")
show_image(median_denoised, "Median Denoised", cmap='gray')
print("双边滤波去噪:")
show_image(bilateral_denoised, "Bilateral Denoised", cmap='gray')
return gray, gaussian_denoised, median_denoised, bilateral_denoised
# 使用示例
gray, gaussian, median, bilateral = clean_image("手写公式.png")

代码块清洗
代码块需要保持格式和缩进:
- 移除行尾空白:每行代码末尾的空格和制表符
- 合并连续空行:多个连续空行合并为一个
- 清理首尾空行:文件开头和结尾的空白行
import re
def clean_code(code_text, remove_comments=False):
"""
增强版代码清洗
"""
if not code_text:
return ""
cleaned_lines = []
in_multiline_comment = False
for line in code_text.split('\n'):
# 移除行尾空白
clean_line = line.rstrip()
# 可选:移除单行注释
if remove_comments:
if not in_multiline_comment:
# 检查是否进入多行注释
if '"""' in clean_line or "'''" in clean_line:
in_multiline_comment = not in_multiline_comment
# 简单处理:直接跳过含有多行注释符号的行
continue
# 移除单行注释(# 后面的内容)
clean_line = re.sub(r'#.*$', '', clean_line)
else:
# 在多行注释中,跳过这行
if '"""' in clean_line or "'''" in clean_line:
in_multiline_comment = False
continue
# 如果行不为空,或者我们保留空行(这里保留一个空行)
if clean_line or (cleaned_lines and not cleaned_lines[-1]):
cleaned_lines.append(clean_line)
# 重新组合并确保首尾没有空行
result = '\n'.join(cleaned_lines).strip()
# 确保以换行符结束(可选)
if result and not result.endswith('\n'):
result += '\n'
return result
混合文档清洗
- 对于企业级RAG项目,你可以根据具体文档类型调整清洗策略
- 图片处理主要是元数据清理,实际图像处理需要专门的库
- 表格处理假设数据已经是结构化的,如果是图片表格需要先OCR识别
def clean_mixed_content(content_type, content):
"""
统一清洗入口
content_type: 'text', 'table', 'image', 'code'
"""
try:
if content_type == 'text':
text = clean_text(content)
text = clean_html(text)
text = clean_markdown(text)
return text
elif content_type == 'table':
return clean_table(content)
elif content_type == 'image':
return clean_image(content)
elif content_type == 'code':
return clean_code(content)
else:
print(f"未知内容类型: {content_type}")
return content
except Exception as e:
print(f"清洗 {content_type} 时出错: {e}")
return content
def batch_clean(documents):
"""
批量清洗文档
documents: 列表,每个元素是 (content_type, content) 元组
"""
results = []
for i, (doc_type, content) in enumerate(documents):
print(f"清洗文档 {i+1}: {doc_type}")
cleaned = clean_mixed_content(doc_type, content)
results.append((doc_type, cleaned))
return results
数据清洗的最佳实践
数据清洗是文档切分的基础工作,良好的清洗能够显著提高后续切分质量和检索效果。
- 保持原始结构:清洗过程中尽量保留文档的原始结构和层次关系
- 最小化信息损失:只去除明确的噪声,避免删除可能有用的内容
- 标准化格式:统一标点符号、引号、连字符等格式元素
- 处理特殊字符:转义或替换可能影响后续处理的特殊字符
- 版本控制:保留原始文档副本,以便需要时回滚
RAG文档切分概述
RAG(Retrieval-Augmented Generation)系统中的文档切分是构建高效检索系统的关键步骤。文档切分,也称为分块(Chunking),是将长文档分割成更小、更易于管理的片段的过程,防止长文档有大部分的噪音数据进入上下文中,这些片段随后被转换为向量并存储在向量数据库中,以便在查询时进行快速检索。
文档切分的重要性
文档切分直接影响RAG系统的性能表现:
- 检索准确性:合理的切分能确保检索到的片段包含完整的相关信息,避免语义割裂
- 上下文完整性:适当的块大小保持足够的上下文信息,使LLM能够生成准确的回答
- 计算效率:合理的块大小平衡了检索精度和计算成本
- 召回率与精确度:切分策略直接影响检索系统的召回率和精确度
切分粒度对RAG效果的影响
粒度是RAG性能的"第一性变量"。不同粒度级别对系统性能有显著影响:
|
粒度级别 |
检索准确性 |
生成质量 |
计算成本 |
典型策略 |
|
细粒度(句子级) |
高 |
低(上下文不足) |
高 |
SentenceWindowNodeParser |
|
中等粒度(段落级) |
中高 |
中高 |
中 |
SentenceSplitter |
|
粗粒度(文档级) |
低(噪声多) |
高(上下文完整) |
低 |
直接使用Document |
- 分块过大: 可能导致检索到的单个块包含过多无关信息(噪音),增加了 LLM 理解上下文的难度,降低了答案的精确性,甚至可能超出 LLM 的上下文窗口限制。
- 分块过小或切分不当: 可能破坏原文的语义连贯性,导致一个完整的知识点被拆散到多个块中。检索时可能只召回了部分信息,使得 LLM 无法获得完整的背景,难以生成全面、准确的答案。
- 未能适应文档结构: 不同的文档类型(如论文、手册、报告、网页)具有不同的结构特点。死板的分块方式可能无法有效利用标题、列表、表格等结构信息,影响信息提取的完整性。
- 一般工程上,层次化切分与句子窗口相结合,通过"粗检索—细生成"的两阶段路径,有效平衡召回率与上下文完整性。
文档切分的基本流程
文档切分通常包括以下基本步骤:
- 文档加载:使用适当的文档加载器加载文档内容(unstructured,Reader等)
- 预处理:根据文档类型进行必要的清洗和格式化
- 切分策略选择:根据文档特点和需求选择合适的切分方法
- 执行切分:应用选定的切分策略将文档分割成块
- 后处理:对切分结果进行必要的调整和优化
切分效果评估指标
评估文档切分效果的常用指标包括:
- 语义完整性:切分后的块是否保持完整的语义信息,能够表达完整的含义
- 上下文连贯性:相邻块之间的内容是否连贯,重叠部分保证思路不中断
- 检索相关性:切分后的块是否能有效支持相关查询
- 生成质量:基于切分块生成的回答质量
- 计算效率:切分和检索过程的计算成本
LlamaIndex核心对象概念
LlamaIndex提供了灵活的文档处理框架,理解其核心概念是有效使用文档切分功能的基础。它们是将大文档首先会读取为document对象,然后拆解为适宜检索的语义片段(节点Node)。选择合适的切分器对于提升 RAG 系统的检索精度和回答质量至关重要。
Document和Node的概念
|
维度 |
Document (文档) |
Node (节点) |
|
概念层级 |
顶层数据容器,代表一个完整的数据源(如一个PDF文件、一个API响应) |
基础数据单元,由Document解析/分块而成,代表其中一段文本 |
|
核心职责 |
数据的统一与标准化:将不同来源、格式的数据封装成统一对象,便于系统处理 |
数据的精细化组织与关联:通过分块、元数据和关系,构建细粒度的数据网络以支持高效检索 |
|
内容与关系 |
包含原始、完整的数据内容;Document之间通常独立 |
包含数据片段;Node之间可通过关系(如父子、先后)构建复杂的图结构 |
|
典型使用场景 |
数据加载与初始化,统一元数据管理(如为整个文档设置来源、作者) |
构建各类索引(向量、关键词等)的基础,实现精确的语义检索,构建复杂的关系知识图谱 |
关于Document
- 数据的起点:在LlamaIndex中,任何外部数据(PDF、数据库、网页等)首先需要通过数据连接器被加载成一个或多个
Document对象。它充当了原始数据的统一接口,随后,通过称为 NodeParser(节点解析器) 的组件,将Document解析为多个Node。 - 元数据承载:
Document级别非常适合存储文档整体的元数据,例如file_name(文件名)、author(作者)、category(分类)等。这些元数据可以被其下的所有Node继承,也可用于高级检索策略。
关于Node
- 索引的基石:
Node是LlamaIndex构建索引的真正原材料。将Document解析成Node的过程(常称为"分块"或"切分")对检索性能至关重要。 - 灵活的关系定义:Node不仅是文本块,每个
Node可以与其他Node建立关系,例如Next(下一个)、Previous(上一个)、Parent(父级)、Child(子级)等。这使得LlamaIndex不仅能处理线性文本,还能构建树状或图状的复杂知识结构,这对于理解长文档的逻辑或回答需要多步推理的问题非常关键。
Document和Node在LlamaIndex中分别扮演着数据容器和语义单元的角色。理解它们的分工与协作,是有效使用LlamaIndex构建高效检索系统的关键。Document负责承载原始数据,而Node则作为构建索引、进行语义检索和生成回答的真正基石。
元数据传播机制
LlamaIndex中的元数据传播遵循继承原则:
- Document级元数据:自动传播到所有由该Document生成的Node
- Node级元数据:可以覆盖或补充Document级元数据
- 关系元数据:存储Node之间的关系信息,如父子、前后关系
from llama_index.core import Document
# 导入SentenceSplitter句子分割器
from llama_index.core.node_parser import SentenceSplitter
# 创建Document并设置元数据
doc = Document(
text="这是一份关于RAG技术的文档...",
metadata={
"file_name": "rag_guide.pdf",
"category": "技术文档",
"author": "AI研究团队",
"created_date": "2023-11-15"
}
)
# 从Document创建Node时,元数据会自动传播
splitter = SentenceSplitter()
nodes = splitter.get_nodes_from_documents([doc])
# 每个node都会继承doc的metadata
nodes[0].metadata
Node结构
|
属性名 |
类型 |
说明 |
|
id_ (node_id) |
str |
节点的唯一标识符,可自动生成或手动指定 |
|
text |
str |
节点包含的文本内容(chunk) |
|
metadata |
Dict[str, Any] |
存储文档的元数据信息(如文件名、页码等) |
|
embedding |
List[float] |
节点的向量嵌入表示 |
|
relationships |
Dict[NodeRelationship, RelatedNodeInfo] |
节点间关系映射 |
|
hash |
str |
内容的哈希值,用于去重和变更检测 |
|
excluded_embed_metadata_keys |
List[str] |
embedding 时排除的元数据键 |
|
excluded_llm_metadata_keys |
List[str] |
LLM 处理时排除的元数据键 |
|
start_char_idx |
Optional[int] |
在原始文档中的起始字符位置 |
|
end_char_idx |
Optional[int] |
在原始文档中的结束字符位置 |
|
text_template |
str |
文本格式化模板 |
|
metadata_template |
str |
元数据格式化模板 |
LlamaIndex中,切分后的基本单元是Node,每个Node包含以下核心属性:
from llama_index.core.schema import TextNode
# Node的基本结构
node = TextNode(
text="这是切分后的文本内容", # 文本内容
metadata={ # 元数据
"file_name": "document.pdf",
"page_number": 1,
"chunk_id": 0
},
id_="node_id_123", # 唯一标识符
embeddings=[] # 文本嵌入向量(可选)
)
关系结构
切分后的Node之间可以建立多种关系:
- 前后关系:表示Node在原文档中的顺序
- 父子关系:表示层次化切分中的层级关系
- 相似关系:表示语义相似的Node
from llama_index.core.schema import NodeRelationship
import json
# Node的基本结构
node0 = TextNode(
text="这是切分后的文本内容0", # 文本内容
metadata={ # 元数据
"file_name": "document.pdf",
"page_number": 1,
"chunk_id": 0
},
id_="node_id_0", # 唯一标识符
)
node1 = TextNode(
text="这是切分后的文本内容1", # 文本内容
metadata={ # 元数据
"file_name": "document.pdf",
"page_number": 1,
"chunk_id": 1
},
id_="node_id_1", # 唯一标识符
)
node2 = TextNode(
text="这是切分后的文本内容2", # 文本内容
metadata={ # 元数据
"file_name": "document.pdf",
"page_number": 1,
"chunk_id": 2
},
id_="node_id_2", # 唯一标识符
)
# 顺序关系
node1.relationships[NodeRelationship.SOURCE] = node0.id_
node1.relationships[NodeRelationship.NEXT] = node2.id_
node2.relationships[NodeRelationship.PREVIOUS] = node1.id_
# 层次化关系
node2.relationships[NodeRelationship.SOURCE] = node0.id_
node1.relationships[NodeRelationship.CHILD] = node2.id_
node2.relationships[NodeRelationship.PARENT] = node1.id_
# 打印节点关系
print(node1.relationships)
print(node2.relationships)
关系类型说明
- SOURCE (值=1)
-
- 含义: 指向源文档(Document)的引用
- 作用: 建立节点与原始文档的追溯链接
- 使用场景: 每个从 Document 分割出的 Node 都会有一个 SOURCE 关系指向原始文档
- PREVIOUS (值=2)
-
- 含义: 指向文档中前一个兄弟节点
- 作用: 构建文档的线性阅读顺序
- 使用场景: 用于保持文档的原始顺序,支持上下文窗口扩展
- 典型应用: Sentence Window Retrieval 技术
- NEXT (值=3)
-
- 含义: 指向文档中下一个兄弟节点
- 作用: 与 PREVIOUS 配合构成双向链表结构
- 使用场景: 支持前后文扩展检索
- PARENT (值=4)
-
- 含义: 指向层级结构中的父节点
- 作用: 建立层级树状结构,支持粗粒度到细粒度的递归组织
- 使用场景: HierarchicalNodeParser 的核心机制
- 层级示例:
-
-
- Level 1: chunk_size=2048 (父节点)
- Level 2: chunk_size=512 (子节点)
- Level 3: chunk_size=128 (孙节点)
-
- CHILD (值=5)
-
- 含义: 指向子节点集合
- 作用: 父节点维护对所有子节点的引用
- 特点: 值可以是 List[RelatedNodeInfo],因为一个父节点可能有多个子节点
文档切分核心原则
有效的文档切分应遵循以下核心原则,以确保切分结果既保持语义完整性又满足检索需求。
语义完整性原则
核心思想:切分应尽量不破坏语义单元的完整性,避免在句子或段落中间进行不合理的分割。
- 句子边界优先:优先在自然语言句子结束处(如句号、问号、感叹号)进行分割
- 段落边界考虑:在段落分隔符处进行分割,保持段落的完整性
- 主题连贯性:切分点应选择在主题转换处,避免将不同主题的内容混合在一个块中
度控制原则
核心思想:控制每个文本块的长度,使其适应模型的上下文限制和检索需求。
- 模型上下文限制:确保块大小不超过模型的输入限制
-
- DeepSeek-R1: 128K tokens
- 检索效率考虑:过大的块会增加检索噪声,过小的块会丢失上下文
-
- 对于一般文档:300-800 tokens
- 技术文档:400-600 tokens
- 对话记录:200-400 tokens
重叠率原则
核心思想:在相邻文本块之间设置适当的重叠区域,避免重要信息在边界处丢失。
- 上下文连续性:重叠区域确保跨边界的语义连续性
- 信息完整性:防止关键信息因切分而被分割到不同块中
- 重叠大小优化:通常设置为块大小的10-20%,根据具体应用场景调整
特殊格式策略原则
核心思想:针对特殊格式的文档(如代码、表格、列表)采用专门的切分策略。
- 代码块完整性:保持函数、类等代码单元的完整性
- 表格结构保持:尽量保持表格的完整性,避免将表格内容分割
- 列表项处理:保持列表项的完整性,避免将单个列表项分割
切分工具选型与实战
LlamaIndex提供了多种切分工具,适应不同的文档类型和应用场景。了解这些工具的特点和适用场景是选择合适切分策略的关键。
|
维度 |
TextSplitter |
NodeParser |
|
核心定位 |
基础的文本分割工具 |
高级的文档解析与节点生成框架 |
|
处理逻辑 |
通常基于固定规则,如长度、标点或字符递归分割 |
除基础分割外,可集成语义分割、代码解析等复杂策略 |
|
输出结果 |
文本块(字符串列表) |
Node对象列表,包含文本、元数据及节点间关系信息 |
|
语义感知 |
通常不具备 |
部分解析器(如SemanticSplitterNodeParser)具备语义感知能力 |
|
性能特点 |
轻量快速,计算开销小 |
功能更强的解析器(如语义分割)可能速度较慢,计算成本高 |
|
适用场景 |
基于语义相似度进行文本切分,保持语义连贯性处理主题转换自然、结构复杂的文档对切分质量要求较高的生产环境 |
构建生产级RAG系统处理复杂文档(如代码、学术论文)需要利用节点间关系(如父节点、子节点)的复杂查询 |
Text-Splitters(文本分割器)类型
Text-Splitters专注于将任意文本字符串拆分成多个片段,按字符/句子/token/自定义分隔规则切分,通常只关心文本长度与重叠上下文,不会理解文件格式。
TokenTextSplitter Token切分器
TokenTextSplitter按照token长度进行切分,适用于需要精确控制token数量的场景,特别是在有严格token限制的嵌入模型或语言模型中使用。
核心特性:
- 基于token而非字符进行切分,更准确地反映模型处理能力
- 支持不同的token计算方法(如tiktoken、huggingface tokenizer)
- 适用于多语言场景,不同语言的token密度不同
from llama_index.core.node_parser import TokenTextSplitter
# 初始化 TokenTextSplitter
token_splitter = TokenTextSplitter(
chunk_size=512, # 每 chunk 目标 token 数(可调)
chunk_overlap=64, # 重叠 token 数(可调)
separator=" " # 分隔符(一般用空格)
)
# 将 documents 转成 nodes(LlamaIndex 内部 node/fragment)
#nodes_from_tokens = token_splitter.get_nodes_from_documents(documents)
#print(nodes_from_tokens[1].get_content())
# 测试文本
test_text = """在检索增强生成(RAG)系统中,文档切分与 Node 转换作为连接原始数据与语言模型的关键预处理环节,直接决定了系统的检索精度、生成质量及整体性能。行业实践数据表明,90% 的 RAG 效果问题源于元数据与分块策略不当,而通过优化分块策略可使检索准确率提升 30 - 50%,语义分块较固定分块的准确率优势可达 27%。这一技术环节的重要性体现在:分块过大易引入冗余噪音,增加语言模型理解负担;分块过小或切分不当则可能破坏语义连贯性,导致完整知识点被拆分;未能适配文档结构的机械分块方式还会忽视标题、列表等结构化信息,影响信息提取完整性。
LlamaIndex 作为连接自定义数据与大语言模型(LLMs)的核心框架,通过将文档(如 PDF、文本文件)分解为包含文本内容、向量嵌入和元数据的 Node 组件,构建了结构化文档管理的技术范式。其核心抽象在于将原始文档转换为语义连贯的 Node 集合,向量存储仅保留 Node 内容的嵌入向量与文本信息,这一机制简化了索引构建流程并提升了检索相关性。文档切分与 Node 转换的质量不仅影响向量检索的效率,更决定了上下文增强(Context Augmentation)这一 RAG 核心能力的实现效果。
本文聚焦文档切分与 Node 转换的技术实践,结合 LlamaIndex 框架的实现机制,系统调研分块策略设计、元数据管理及 Node 组件化等关键技术点。通过分析行业最佳实践与典型案例,旨在为 RAG 系统开发者提供可落地的优化方案,解决分块噪音、语义断裂、结构信息丢失等核心痛点,为构建高性能检索增强生成应用奠定技术基础。"""
# 按照文本来进行切分
nodes_from_tokens = token_splitter.split_text(test_text)
# 打印切分后的文本数量
print( "切分后的文本数量: " + str(len(nodes_from_tokens)))
print("===============================================")
# 打印第一个文本
print("第一个文本:")
print(nodes_from_tokens[0])
print("===============================================")
# 打印第二个文本
print("第二个文本:")
print(nodes_from_tokens[1])
print("===============================================")
SentenceSplitter 句子切分器
SentenceSplitter是一种基于自然语言句子和段落边界进行分割的解析器,类似于LangChain的RecursiveCharacterTextSplitter。它优先在句子结束处或段落分隔符处进行分割,尽量避免在句子中间切断,以保持语义单元的完整性。
核心特性与工作原理:
- 保持句子完整性:首先尝试按句子边界(如中文的"。""!""?"或英文的".""""!")进行分割
- 递归分割策略:如果单个句子长度超过chunk_size,递归地使用更小的分隔符进行分割
- 重叠控制:通过chunk_overlap参数,允许相邻文本块之间有少量重叠的token
- 多语言支持:通过separator参数可以自定义分隔符
关键参数:
|
参数名 |
类型 |
说明 |
默认值 |
|
chunk_size |
int |
每个文本块的目标最大token数 |
1024 |
|
chunk_overlap |
int |
相邻文本块之间重叠的token数 |
200 |
|
separator |
str |
用于分割的主要分隔符 |
" " (空格) |
|
paragraph_separator |
str |
用于识别段落的分隔符 |
"\n\n\n" |
适用场景: SentenceSplitter非常适用于自然语言文本,如新闻文章、博客文章、书籍章节等结构清晰的散文体内容。它是LlamaIndex中最常用且默认的文本分割器之一。
from llama_index.core.node_parser import SentenceSplitter
# 初始化 SentenceSplitter
sentence_splitter = SentenceSplitter(
chunk_size=512, # 这里 chunk_size 表示 token 近似或字符近似,视版本调整
chunk_overlap=64
)
# 将 documents 转成 nodes
# nodes_from_sentences = sentence_splitter.get_nodes_from_documents(documents)
# print(nodes_from_sentences[0].get_content())
# 测试文本
test_text = """在检索增强生成(RAG)系统中,文档切分与 Node 转换作为连接原始数据与语言模型的关键预处理环节,直接决定了系统的检索精度、生成质量及整体性能。行业实践数据表明,90% 的 RAG 效果问题源于元数据与分块策略不当,而通过优化分块策略可使检索准确率提升 30 - 50%,语义分块较固定分块的准确率优势可达 27%。这一技术环节的重要性体现在:分块过大易引入冗余噪音,增加语言模型理解负担;分块过小或切分不当则可能破坏语义连贯性,导致完整知识点被拆分;未能适配文档结构的机械分块方式还会忽视标题、列表等结构化信息,影响信息提取完整性。
LlamaIndex 作为连接自定义数据与大语言模型(LLMs)的核心框架,通过将文档(如 PDF、文本文件)分解为包含文本内容、向量嵌入和元数据的 Node 组件,构建了结构化文档管理的技术范式。其核心抽象在于将原始文档转换为语义连贯的 Node 集合,向量存储仅保留 Node 内容的嵌入向量与文本信息,这一机制简化了索引构建流程并提升了检索相关性。文档切分与 Node 转换的质量不仅影响向量检索的效率,更决定了上下文增强(Context Augmentation)这一 RAG 核心能力的实现效果。
本文聚焦文档切分与 Node 转换的技术实践,结合 LlamaIndex 框架的实现机制,系统调研分块策略设计、元数据管理及 Node 组件化等关键技术点。通过分析行业最佳实践与典型案例,旨在为 RAG 系统开发者提供可落地的优化方案,解决分块噪音、语义断裂、结构信息丢失等核心痛点,为构建高性能检索增强生成应用奠定技术基础。"""
# 按照文本来进行切分
nodes_from_sentences = sentence_splitter.split_text(test_text)
print( "切分后的文本数量: " + str(len(nodes_from_sentences)))
print("===============================================")
print("第一个文本:")
print(nodes_from_sentences[0])
print("===============================================")
print("第二个文本:")
print(nodes_from_sentences[1])
print("===============================================")
递归测试
- 如果分块的大小超过预定义的块大小限制,则将每个块拆分成更小的块。然而,如果块符合块大小限制,则不再进行进一步拆分
- 与固定大小的块不同,这种方法还保持了语言的自然流畅性,并保留了完整的思想
from llama_index.core.node_parser import SentenceSplitter
from llama_index.core.schema import Document
# SentenceSplitter 测试递归切分
# 示例文本:一个长句子和一个短句子
text = "这是一个非常长的句子,它包含了多个逗号分隔的部分,和一些短的文本,因此整个句子的长度会很容易超过我们设定的块大小限制。短期。"
document = Document(text=text)
# 初始化SentenceSplitter,设定块大小非常小以触发递归分割
splitter = SentenceSplitter(chunk_size=10, chunk_overlap=0,separator=",。!?!?.\n¡¿")
nodes = splitter.get_nodes_from_documents([document])
# 打印分割结果
print("=" * 60)
for i, node in enumerate(nodes):
print(f"块 {i+1}: {node.text}")
print("=" * 60)
print("打印Node信息:")
print(nodes[0].__dict__)
CodeSplitter 代码切分器
CodeSplitter专为编程语言源代码设计,利用编程语言的抽象语法树(AST)来理解代码结构,确保将代码按功能单元进行分割。
核心特性:
- 基于抽象语法树(AST)的结构化解析
- 语言特定,需要指定编程语言
- 保持代码块的功能完整性
from llama_index.core.node_parser import CodeSplitter
from llama_index.core.schema import Document
# 示例Python代码:一个简单的函数和类
sample_code = '''
def calculate_fibonacci(n):
"""计算斐波那契数列的第n项"""
if n <= 1:
return n
else:
return calculate_fibonacci(n-1) + calculate_fibonacci(n-2)
class MathOperations:
"""一个简单的数学操作类"""
def __init__(self):
self.version = "1.0"
def factorial(self, n):
"""计算阶乘"""
if n == 0:
return 1
result = 1
for i in range(1, n+1):
result *= i
return result
def is_prime(self, n):
"""判断是否为质数"""
if n < 2:
return False
for i in range(2, int(n**0.5)+1):
if n % i == 0:
return False
return True
'''
# 创建文档对象
# document = Document(text=sample_code)
#
# 初始化CodeSplitter,指定Python语言
code_splitter = CodeSplitter(
language="python", # 指定编程语言
chunk_lines=10, # 每块大约行数
chunk_lines_overlap=2, # 块之间重叠行数
max_chars=600 # 每块最大字符数
)
# 执行切分
# nodes = code_splitter.get_nodes_from_documents([document])
nodes = code_splitter.split_text(sample_code)
print(f"原始代码字符数: {len(sample_code)}")
print(f"切分后的节点数量: {len(nodes)}")
print("\n" + "="*50 + "\n")
# 显示切分结果
for i, node in enumerate(nodes):
print(f"节点 {i+1} (字符数: {len(node)}):")
print("-" * 30)
print(node)
print("\n" + "="*50 + "\n")
File-Based Node Parsers(文件型节点切分器)
File-Based Node Parsers面向文件类型与结构(如Markdown、JSON、PDF、HTML、代码文件等),会根据文件的语义/格式选择专门解析器,把整个文件解析成带元数据的Node,可能保持章节/层级、标题、表格等结构。
MarkdownNodeParser markdown切分器
切分器 MarkdownNodeParser专门用于处理Markdown文件,能够识别Markdown的层级结构(如标题、列表、代码块等),并据此进行切分。
# 文本类型是markdown的
from llama_index.core.node_parser import MarkdownNodeParser
from llama_index.core.readers import SimpleDirectoryReader
markdown_docs = SimpleDirectoryReader(input_files=["扩展调参.md"]).load_data()
# markdown_docs = [Document(text="# 主标题\n\n这是第一段。\n\n## 子标题\n\n这是第二段。")]
# 创建Markdown解析器
parser = MarkdownNodeParser()
# 从Markdown文件创建节点
nodes = parser.get_nodes_from_documents(markdown_docs)
# 显示切分结果
for i, node in enumerate(nodes):
print(f"节点 {i+1} (字符数: {len(node.text)}):")
print("-" * 30)
print(node.text)
print("\n" + "="*50 + "\n")
JSONNodeParser Json切分器
JSONNodeParser用于处理JSON文件,能够根据JSON结构进行切分,保持数据的层次关系。
from llama_index.core.node_parser import JSONNodeParser
json = """
{
"id_": "0a1eee9a-635a-4391-8b74-75bf3c648f0e",
"embedding": null,
"metadata": {
"document_id": "FULadzkWmovlfkxSgLPcE4oWnPf"
},
"excluded_embed_metadata_keys": [],
"excluded_llm_metadata_keys": [],
"mimetype": "text/plain"
}
"""
# 创建JSON解析器
parser = JSONNodeParser()
json_docs = [Document(text=json)]
# 从JSON文件创建节点
nodes = parser.get_nodes_from_documents(json_docs)
# 显示切分结果
for i, node in enumerate(nodes):
print(f"节点 {i+1} (字符数: {len(node.text)}):")
print("-" * 30)
print(node.text)
print("\n" + "="*50 + "\n")
SemanticSplitterNodeParser 语义切分器
SemanticSplitterNodeParser通过嵌入模型计算文本块间的语义相似度,实现自适应断点识别,核心解决固定分块的语义割裂问题。其检索准确率较固定分块提升20%左右,适合对上下文连贯性要求高的场景(如学术论文、长文档理解)。
实现原理
- 句子分割:将文档拆分为独立句子单元
- 嵌入计算:通过嵌入模型(如OpenAIEmbedding、BAAI/bge-m3)生成句子向量,计算成本较高
- 相似度判断:计算相邻句子向量的余弦相似度
- 断点识别:当相似度低于设定阈值(如breakpoint_percentile_threshold=90)时执行切分
- 块生成:合并语义相近的句子为完整分块,适用于对语义连贯性要求高的场景
简单来说,它的工作流程是:先将文本拆分成句子,然后通过滑动窗口计算句子群的综合语义,最后在语义发生显著变化的地方(即相似度低于阈值时) 进行分割,它的分割点是动态的、由语义决定的,因此无法像固定大小的分割器那样,简单地在前一个块的末尾和后一个块的开头插入一段重叠的文本;这种基于语义的分割方式,其设计目标之一就是让每个分割出的文本块在语义上尽可能独立和完整。buffer_size 参数在某种程度上扮演了维持上下文连贯性的角色,因为它确保了在判断是否分割时,已经考虑了当前句子周围一定范围内的语义上下文
# requirements (示例)
# pip install llama-index
from llama_index.core import Document
from llama_index.embeddings.openai import OpenAIEmbedding
import os
from llama_index.core.settings import Settings
from dotenv import load_dotenv
# 加载环境变量
load_dotenv()
# 设置为全局默认Embedding模型
Settings.embed_model = OpenAIEmbedding(
model="text-embedding-3-small",
api_key=os.getenv("OPENAI_API_KEY"),
api_base=os.getenv("OPENAI_BASE_URL", "https://api.openai.com/v1")
)
调优建议
- 尝试不同buffer_size:
-
1对局部句子差异敏感2~3会以更宽的窗口判断相似度(可能得到更长但更连贯的chunk)
- 断点阈值(
breakpoint_percentile_threshold):
-
- 算法不是使用一个固定的相似度值(如0.5)作为阈值,而是采用一种基于数据分布的方法。
- 它将上一步得到的所有相邻相似度值进行排序,并计算该列表的指定百分位数。
- breakpoint_percentile_threshold = 95 意味着:“将阈值设定在比95%的相邻相似度都要低的那个位置”。
- 计算出的阈值 = 相似度列表的第5百分位值(因为100-95=5)。也就是说,只有相似度最低的那5%的相邻对,才会被考虑作为分割点。
- 降低阈值会更容易切断(产生更多小块)
- 提高阈值会合并更多句子
- 举例说明: 假设你的文档被切成10个句子,有9个相邻相似度,按升序排列为: [0.1, 0.2, 0.3, 0.5, 0.7, 0.8, 0.85, 0.9, 0.95]
-
- 如果 breakpoint_percentile_threshold = 95:
-
-
- 阈值 = 第5百分位值 ≈ 列表中最小的5%的值。在这个9个元素的列表中,5%的位置大约是第0.45个元素,通常取第一个或前几个。在实践中,可能会取0.1或0.2作为阈值。
- 最终,只有在相似度为0.1和0.2的地方会被分割。这意味着你会得到数量较少但体积较大的节点。
-
-
- 如果 breakpoint_percentile_threshold = 80:
-
-
- 阈值 = 第20百分位值。在排序列表中,20%的位置大约是第1.8个元素,可以取0.3作为阈值。
- 那么,相似度低于0.3(即0.1和0.2)以及等于0.3的地方都会被分割。这会生成更多、更细的节点。
-
- 中文句子拆分要牢靠:务必先用可靠的拆句器;对复杂文本(引号、括号、列表)需要更细致的预处理。
- 超长chunk的"安全裁剪":在极端结构化文本中可能产生超长chunk(导致embedding报错),可以在
SemanticSplitterNodeParser之后接一个SentenceSplitterNodeParser或TokenTextSplitter作"后备分割"。
SentenceWindowNodeParser 句子窗口切分器
SentenceWindowNodeParser的工作流程核心在于检索单元和上下文窗口的分离:
- 精细索引:在索引构建阶段,它会将文档拆分成单个句子作为基础节点(Node)。这种细粒度拆分有助于向量模型更好地表征句子语义,从而在检索时能更精准地找到相关句子。
- 窗口上下文:每个句子节点都会在元数据(metadata)中存储其周围句子构成的窗口文本。检索时,系统首先找到最相关的句子节点,然后将其替换为对应的上下文窗口,再将这个更大的文本块传递给LLM生成答案。
这种方法有效缓解了RAG系统中"检索精度"与"生成答案所需上下文完整性"之间的矛盾。
|
特性维度 |
具体说明 |
|
核心原理 |
将文档按句子拆分并建立索引,检索时返回匹配句子及其周围句子(滑动窗口)。 |
|
主要优势 |
检索与上下文解耦:检索用小粒度句子提升精度,提供给LLM的是包含更完整上下文的窗口文本。 |
|
关键参数 |
windowsize:控制窗口大小;windowmetadata_key:存储窗口文本的元数据键名。 |
|
典型应用场景 |
处理文档结构清晰、句子间逻辑连贯的文档,如技术文档、学术论文、法律合同等。 |
import re
from typing import List
from llama_index.core import Document # or from llama_index.schema import Document depending on version
from llama_index.core.node_parser import SentenceWindowNodeParser
from llama_index.core.postprocessor import MetadataReplacementPostProcessor
# 1) 中文拆句器(示例)
def split_chinese_sentences(text: str) -> List[str]:
"""
将中文文本按常见中文句末标点拆为句子(保留标点)。
返回去除空串和首尾空白的句子列表。
"""
# 以中文/英文句末标点为分界(正则使用前后肯定断言保留标点)
pieces = re.split(r'(?<=[。!?\?!.])', text)
# 清理空白与空串
sentences = [p.strip() for p in pieces if p and p.strip()]
return sentences
# 2) 创建 SentenceWindowNodeParser
node_parser = SentenceWindowNodeParser(
sentence_splitter=split_chinese_sentences, # 自定义中文拆句器
window_size=1, # 左右各1句作为window
window_metadata_key="window", # metadata 中存储窗口的 key
original_text_metadata_key="original_text", # 可选:保存原始文本
include_metadata=True, # 是否包含 metadata
include_prev_next_rel=True, # 是否包含上一句与下一句的关系
)
# 3) 示例中文文本与节点生成(调试)
# text = (
# "这是第一句。这里是第二句,它比较长,会测试拆分。"
# "第三句来了!第四句?第五句,继续测试。"
# )
text = "本报告中的信息均来源于我们认为可靠的已公开资料,本公司对这些信息的真实性、准确性及完整性不作任何保证。 本报告中的信息、意见等均仅供客户参考,该等信息、意见并未考虑到获取本报告人员的具体投资目的、财务状况以 及特定需求,在任何时候均不构成对任何人的个人推荐。客户应当对本报告中的信息和意见进行独立评估,并应同时 思量各自的投资目的、财务状况以及特定需求,必要时就法律、商业、财务、税收等方面咨询专家的意见。客户应自 主作出投资决策并自行承担投资风险。本公司特别提示,本公司不会与任何客户以任何形式分享证券投资收益或分担 证券投资损失,任何形式的分享证券投资收益或者分担证券投资损失的书面或口头承诺均为无效。市场有风险,投资 须谨慎。对依据或者使用本报告所造成的一切后果,本公司和关联人员均不承担任何法律责任。"
doc = Document(text=text, metadata={"doc_id": "示例文档1"})
nodes = node_parser.get_nodes_from_documents([doc])
# 打印每个节点及其 metadata(调试用)
for i, node in enumerate(nodes[:2]):
print("---- NODE", i, "----")
print("node.text:", repr(node.text)) # node.text 应为单句
print("=" * 80)
#print("relationships:", node.relationships)
# 存在的 metadata key
#print("metadata keys:", node.metadata.keys())
# window 存放左右句(字符串或列表,取决于实现;通常是字符串)
print("window metadata:", node.metadata.get("window"))
# original_text 为可选,存放原始文本
print("original_text:", node.metadata.get("original_text"))
print()
# 4) 在检索/查询时把 window 替换回去
# 用于 QueryEngine 的 node_postprocessor 示例:把 node.text 替换为 metadata['window'](如果存在)
postproc = MetadataReplacementPostProcessor(target_metadata_key="window")
# 当你用 index.as_query_engine(..., node_postprocessors=[postproc]) 时,
# 被检索到的 node.text 会被 postproc 替换为 metadata['window'](即包含左右句的文本)
# 之后传给 LLM 的就是带上下文的窗口文本,而不是单个句子
使用注意事项
- 处理长文档:
-
- 对于多页PDF,有时需要先将所有页面的文本内容合并为一个文档,以确保句子窗口能跨页面边界正确划分。
- 窗口大小的选择:
-
- 太小:可能无法提供足够的上下文,影响LLM的理解。
- 太大:会增加传递给LLM的token数量,可能导致成本上升和处理延迟,甚至可能触及模型上下文长度限制。
- 通常建议从3开始尝试,并根据效果调整。
- 适用场景:
-
- 处理结构清晰的文档:对于技术文档、学术论文、法律合同等逻辑性强、句子间关联紧密的文档,能有效避免固定分块可能导致的答案被割裂的问题。
- 对答案准确性要求高的应用:当需要LLM生成的答案严格基于文档上下文,减少"幻觉"时,提供更完整的窗口上下文有助于提升答案的准确性和可靠性。
HierarchicalNodeParser 结构切分器
HierarchicalNodeParser结合文档结构(标题、章节、段落)和语义边界进行多层次切分,适合处理Markdown、PDF、Word等结构化文档,尤其适用于说明书、规约、设计文档等场景。其核心优势在于保留文档原生逻辑层级,支持父节点(章节标题+简介)与子节点(具体段落)的嵌套组织。
默认设置优先原则与场景适配
- 结构文档(如.md、.pdf、.docx):默认使用HierarchicalParser,优先保留文档结构
- 非结构文档(如.txt、.csv):默认使用SentenceSplitter,平衡效率与基础语义完整性
# -*- coding: utf-8 -*-
import re
import json
from typing import List
from llama_index.core import Document
from llama_index.core.node_parser.relational.hierarchical import HierarchicalNodeParser
# 构造示例中文较长文档
text = (
"第一章:公司发展背景。公司成立于2005年,最初是一家小型软件外包公司。"
"随着云计算与大数据的兴起,公司在2010年转型为云服务提供商,"
"并在2015年完成了A轮融资,融资金额达数千万美元。"
"第二章:产品与服务。公司主要提供数据分析平台、实时流处理系统和人工智能模型服务。"
"其中,数据分析平台支持海量日志处理;流处理系统可实现毫秒级别延迟。"
"第三章:市场与竞争。国内外竞争者众多,我们面临来自大型互联网公司的压力,"
"但我们的优势在于垂直行业深耕与定制化服务。"
"第四章:未来展望。我们计划在2026年前进入国际市场,并开展亚太地区的业务。"
)
print("=== 切分前(原文) ===")
print(text)
print("\n")
# 创建 HierarchicalNodeParser
node_parser = HierarchicalNodeParser.from_defaults(
chunk_sizes=[300, 120], # 例:300字符/120字符级别
chunk_overlap=30, # 重叠区域大小
include_metadata=True, # 是否包含metadata
include_prev_next_rel=True # 是否包含前后关系
)
# 注意:默认内部会为每个级别创建 SentenceSplitter 用于拆分
nodes = node_parser.get_nodes_from_documents([Document(text=text, metadata={"doc_id":"示例文档"})])
print("=== 切分后(nodes) ===")
for idx, node in enumerate(nodes):
print(f"--- node {idx} ---")
print("relationships:", node.relationships)
print("=" * 60)
print("text:", repr(node.text))
print("metadata keys:", list(node.metadata.keys()))
# 可显示 chunk size 或 parent id
print("metadata sample:", {k: node.metadata.get(k) for k in ["chunk_size","chunk_level","doc_id"] if k in node.metadata})
print()
混合策略
结合多种切分策略,发挥各自优势:
from unstructured.partition.pdf import partition_pdf
# 使用 partition_pdf 函数解析 PDF 文档 为elements类型
elements = partition_pdf(
filename="甬兴证券-AI行业点评报告:海外科技巨头持续发力AI,龙头公司中报业绩亮眼.pdf",
strategy="hi_res", # 使用高精度模式
extract_images_in_pdf=False,
)
Unstructured的chunk_by_title
核心思想: 利用Title元素作为分段标志,将Title与其后的内容组合成语义完整的chunk
- 优势:
-
- 保留文档结构边界
- 自动合并小段落
- 保留元数据层级信息
- 避免跨章节混合
# 导入unstructured的chunk_by_title切分方法
from unstructured.chunking.title import chunk_by_title
# 文档切分
chunked_elements = chunk_by_title(
elements, # 读取的元素列表
max_characters=800, # 每个chunk的最大字符数
combine_text_under_n_chars=150, # 小于该字符数的文本块会合并
)
print(f"✓ 解析出 {len(elements)} 个元素")
print(f"✓ 切分成 {len(chunked_elements)} 个chunks")
HierarchicalNodeParser 结合 SemanticSplitterNodeParser语义切分
核心思想: 创建多层级的chunk结构,对长文本再进一步使用SemanticSplitterNodeParser或其他切分器进行切分
- 优势:
-
- 提供多粒度检索
- 自动保留父子关系
- 适合长文档
from llama_index.readers.file.unstructured import UnstructuredReader
from unstructured.partition.auto import partition
from llama_index.core import Document
from pathlib import Path
def smart_load(file_path):
"""
智能文档加载器:根据文件类型选择最佳解析策略
Args:
file_path: 文件路径
Returns:
解析后的Document对象列表
"""
file_path = Path(file_path)
file_ext = file_path.suffix.lower()
# 定义复杂文件类型(需要高精度解析)
complex_types = {
'.pdf', # PDF文档(可能包含表格、图像、复杂布局)
'.png', '.jpg', '.jpeg', '.gif', '.bmp', '.tiff', # 图片文件(需要OCR)
'.docx', '.doc', # Word文档(可能包含复杂格式)
'.pptx', '.ppt', # PowerPoint(复杂布局)
'.xlsx', '.xls' # Excel(表格结构)
}
# 简单文件类型(可以用Reader直接处理)
simple_types = {
'.txt', '.md', '.csv', '.html', '.xml', '.json'
}
if file_ext in complex_types:
# 复杂文件使用底层解析,获得更好的结构识别
print(f"检测到复杂文件类型 {file_ext},使用partition高精度解析")
try:
elements = partition(
filename=str(file_path),
# 使用hi_res模式进行高精度解析
strategy="hi_res",
# 支持中文、英文
languages=["eng", "chi_sim"],
# 推断表格结构
infer_table_structure=True
)
# 将解析元素转换为Document对象
return [Document(text=e.text, metadata={
"source": str(file_path),
"element_type": type(e).__name__,
"file_type": file_ext
}) for e in elements if e.text.strip()] # 过滤空文本
except Exception as e:
print(f"高精度解析失败,回退到Reader: {e}")
# 回退到Reader
reader = UnstructuredReader()
return reader.load_data(file=file_path)
else:
# 简单文件或未知类型优先使用Reader
print(f"检测到简单文件类型 {file_ext},使用Reader解析")
try:
# 直接使用Reader进行简单解析
reader = UnstructuredReader()
# 加载解析后的文档,返回 Document 对象列表
docs = reader.load_data(file=file_path)
return docs
except Exception as e:
print(f"Reader解析失败,回退到partition: {e}")
# 回退到底层解析
elements = partition(filename=str(file_path), strategy="auto")
return [Document(text=e.text, metadata={"source": str(file_path)}) for e in elements]
AtomGit 是由开放原子开源基金会联合 CSDN 等生态伙伴共同推出的新一代开源与人工智能协作平台。平台坚持“开放、中立、公益”的理念,把代码托管、模型共享、数据集托管、智能体开发体验和算力服务整合在一起,为开发者提供从开发、训练到部署的一站式体验。
更多推荐



所有评论(0)