当前位置：首页 > news >正文

GraphRAG 别急着上：先把图谱血缘理清，比调大模型重要十倍

news 2026/7/20 19:53:23

这篇我按“先跑起来、再讲取舍”的方式写《一次GraphRAG项目复盘，问题最后出在流程而不是模型》。概念会讲，但重点放在代码怎么组织、哪里容易踩坑。

摘要

先把这篇文章的目标说清楚：看完之后，你应该能判断这件事值不值得做，以及从哪里动手。

摘要：很多团队引入 GraphRAG 后发现效果不如预期，甚至导致系统更慢。本文复盘一次将知识图谱融入 RAG 的实战经历，指出核心痛点往往不在模型精度，而在数据治理。通过具体的实体关系抽取和图查询优化案例，分享如何避开“死图”陷阱，构建可维护的企业级知识库。

传统 RAG 的瓶颈：为什么向量检索会“迷路”？

在前段时间的团队技术分享会上，我们讨论了一个典型场景：用户问“A 项目的上游依赖 B 模块，B 模块又受 C 库版本影响，请问 A 项目的潜在风险是什么？”

如果用传统的 Vector RAG（基于向量检索的生成增强），我们面临的最大问题是逻辑断裂。向量检索擅长语义匹配，比如找到包含“A 项目”、“风险”、“依赖”的文档片段。但它很难跨越多个文档节点，精准地串联起 A->B->C 的链条。即使你把所有文档都向量化，检索出来的 Top-K 碎片往往是孤立的，LLM 需要在上下文窗口里强行拼凑，这就导致了“幻觉”或者回答模棱两可。

这就是我们决定尝试 GraphRAG 的直接动因。我们不是要抛弃向量检索，而是试图用知识图谱（Knowledge Graph, KG）来补齐 RAG 在“结构化逻辑”和“全局视野”上的短板。但在动手之前，我们必须承认一个反直觉的事实：对于大多数中小团队，GraphRAG 的维护成本远高于它带来的收益，除非你的数据存在强烈的实体关联性。

知识图谱建模：别贪多，先抓主干

很多初学者（包括之前的我）容易犯的错误是试图把企业所有信息都塞进图谱里。结果就是图谱变得巨大且稀疏，查询延迟飙升。

在我们的实战中，我们做了一个关键的取舍：只建模“强关系”和“高价值实体”。

我们定义的 Schema 非常简单，只包含三类核心节点和两种关系：
1. 文档块 (Chunk)：经过语义切分的原始内容。
2. 实体 (Entity)：人名、项目名、代码库名、API 接口名。
3. 关系 (Relation)：DEPENDS_ON(依赖),MENTIONS(提及),PART_OF(属于)。

我们没有去建模复杂的继承树或详细的配置参数，因为那些更适合放在向量索引里。图谱负责的是“骨架”，向量负责的是“血肉”。

# Neo4j Cypher 示例：创建基础实体和关系 CREATE (c:Chunk {id: 'chunk_001', content: 'A项目依赖B模块...'}) CREATE (e1:Entity {name: 'A项目', type: 'Project'}) CREATE (e2:Entity {name: 'B模块', type: 'Component'}) CREATE (e1)-[:MENTIONS]->(c) CREATE (c)-[:MENTIONS]->(e2) CREATE (e1)-[:DEPENDS_ON]->(e2)

这一步看似简单，但实际上决定了后续检索的效率。如果实体提取不准，或者关系定义过于宽泛，整个图谱就会变成一张“蜘蛛网”，检索时根本无从下手。

实体关系抽取：自动化还是半自动？

这是整个流程中最容易踩坑的地方。理论上，我们可以直接用 LLM 从非结构化文本中提取三元组(Head, Relation, Tail)。但在实际生产中，直接让 LLM 全量抽取面临着两个问题：
1. 一致性差：今天叫“User Service”，明天叫“UserService”，图谱里就会分裂成两个实体。
2. 成本高昂：全量处理历史文档的 Token 消耗巨大。

我们的解决方案是“半自动 + 规则清洗”。

首先，利用现有的 OCR 和日志系统，预定义一批高频实体词典。然后，只对新增或更新频繁的文档进行 LLM 抽取。抽取后，必须经过一个标准化层，将所有变体映射回标准实体 ID。

此外，我们发现一个有趣的现象：对于代码类知识库，静态分析（Static Analysis）提取的关系比 LLM 更准确。比如，通过 AST（抽象语法树）解析得到的函数调用关系，远比让 LLM 读几行代码猜出来的“调用关系”靠谱。因此，我们将代码仓库的依赖树直接导入图谱，而将技术文档、Wiki 条目交给 LLM 抽取实体关系。这种混合策略大大降低了噪声。

图检索增强：Hybrid Search 的真正玩法

有了图谱，怎么检索？这里我们要区分两种查询模式：

1. 局部查询：用户问“B 模块的最新 API 文档在哪？”。
* 这时直接用向量检索Chunk节点最快，图谱只用来做实体消歧。
2. 全局/推理查询：用户问“如果 C 库升级，会对 A 项目产生什么影响？”。
* 这才是 GraphRAG 的主场。我们需要进行子图提取（Subgraph Extraction）。

具体做法是：
1. 先将用户问题的关键词转化为实体 ID。
2. 在图谱中进行 $k$-hop 游走，找到与这些实体相连的子图。
3. 将子图中的实体描述、关系类型以及关联的文档片段，重新组织成 Prompt 上下文。

注意，不要直接把整个子图扔给 LLM。我们需要对子图进行摘要（Summarization）。利用 LLM 对每个实体周围的邻居节点生成简短的描述，比如：“B 模块：负责用户认证，依赖 C 库 v1.2+，最近修复了 XX 漏洞”。这样既保留了拓扑结构的信息，又控制了上下文长度。

# 伪代码：构建图感知检索请求 def graph_aware_rag(query): entities = extract_entities(query) # 获取 A, B, C 实体ID subgraph = neo4j.query_graph(entities, hop=2) # 获取两跳内的子图 # 对子图中的每个实体生成摘要 summaries = [] for node in subgraph.nodes: summary = llm.summarize(node.neighbors()) summaries.append(summary) # 组装最终 Prompt prompt = f"Query: {query}\nContext:\n" + "\n".join(summaries) return llm.generate(prompt)