一个更小、更快、更干净的GraphRAG！

GraphRAG 的概念近几个月热度很高，它的能力毋庸置疑。但如果你尝试过直接阅读或修改微软官方的 GraphRAG 实现，大概会认同一个感受：代码量大、结构复杂、文档也不够友好，想基于它做二次开发或者只是理解其工作原理，都相当痛苦。

正是为了降低这个门槛，

出现了。这是一个更轻量、更干净、更易于安装和修改的 GraphRAG 实现，同时保留了最核心的功能。它的代码量（排除测试和提示部分）大约只有 800 行，采用异步设计，完全类型化，可扩展性强。

快速入门

安装非常简单，一行命令搞定：

pip install nano-graphrag

你可以拿查尔斯·狄更斯的《圣诞颂歌》来试一试：

curl https://raw.githubusercontent.com/gusye1234/nano-graphrag/main/tests/mock_data.txt > ./book.txt

然后运行下面的 Python 代码即可完成索引构建和查询：

from nano_graphrag import GraphRAG, QueryParam

graph_func = GraphRAG(working_dir="./dickens")

with open("./book.txt") as f:
    graph_func.insert(f.read())

# 执行全局 graphrag 搜索
print(graph_func.query("What are the top themes in this story?"))

# 执行局部 graphrag 搜索（这通常是更好、更具扩展性的选择）
print(graph_func.query("What are the top themes in this story?", param=QueryParam(mode="local")))

下次你从相同的工作目录初始化 GraphRAG 时，所有上下文都会自动重新加载，无需重复处理。

增量插入

nano-graphrag 的一个很实用的特性是支持增量插入，而且不会造成重复计算或数据冗余。它使用内容的 MD5 哈希作为键来去重，确保同一段内容不会被重复索引。

with open("./book.txt") as f:
    book = f.read()
half_len = len(book) // 2
graph_func.insert(book[:half_len])
graph_func.insert(book[half_len:])

需要注意的是，每次插入操作后，图的社区会被重新计算，社区报告也会重新生成——这算是目前的代价，但对于数据频繁更新的场景，整体效率仍然可控。

与微软 GraphRAG 的效果对比

为了验证 nano-graphrag 的实际效果，我们与微软官方的 GraphRAG 实现进行了一组对比测试。测试设定如下：

• 语料库：罗贯中的《三国演义》
• 官方版本：GraphRAG 的 commit 61b5eea34783c58074b3c53f1689ad8a5ba6b6ee
• 两者均使用 OpenAI 的 embedding 和 gpt-4o
• 均不使用缓存，且在同一设备和网络环境下运行
• 官方版本最大异步请求数：25；nano-graphrag 最大异步请求数：16

Index 对比

两个方案的索引构建时间都超过了 10 分钟（语料量较大），在此不做详细对比。

Local Search 对比

我们以问题“大乔和曹操的关系”为例，看看两种实现给出的答案质量。

大乔与曹操之间的关系在历史和文学中被描绘为复杂且多维的，尽管直接的互动可能并不频繁。大乔，作为东吴孙策的妻子，以及其美貌和与三国时期重要人物的关联而闻名。曹操，作为三国时期的关键整治和军事领导者，其与大乔的关系主要通过他与东吴的关系以及对大乔及其家族的兴趣来体现。

首先，大乔的美貌在三国时期广为人知，甚至引起了曹操的注意。据记载，曹操曾有意俘获大乔以供己用，这一点体现了曹操对大乔的个人兴趣，尽管这种兴趣可能更多地是基于她的美貌和象征意义，而非两人之间有实质性的个人关系 [Data: Entities (266)]。

其次，大乔作为孙策的妻子，自然成为了孙家与东吴势力中的一员。孙策和孙权是曹操在整治和军事上的对手，这使得大乔与曹操之间存在着间接的敌对关系。曹操对东吴的军事行动，包括对孙权的攻击，间接地影响到了大乔及其家族的安全和利益 [Data: Entities (266)]。

最后，尽管直接的文献记录可能不多，大乔的地位和她所处的时代背景使她与曹操之间的关系充满了历史和文化上的意义。她的故事和曹操的野心相交织，展现了三国时期女性角色与整治军事活动之间的复杂联系。

综上所述，大乔与曹操之间的关系主要是通过她的家庭背景、曹操的整治军事活动，以及文化记忆中对她美貌的追捧来定义的。这种关系更多地体现在历史和文学的叙述中，而不是基于两人之间有直接的个人互动。

# 大乔和曹操的关系

## 大乔的背景
大乔是江东乔公的长女，以其极美的容貌闻名于世。她最为著名的身份是孙策的妻子，这一身份在三国时期的整治和军事格局中也占有重要地位。周瑜特别提到她，显示她在周瑜与曹操局势中的重要性 [Data: Entities (1)]。

## 曹操的欲望
曹操对大乔的渴望主要源自其好色本性。他希望得到大乔和她的妹妹小乔，以满足个人欲望，并且认为获得她们可以帮助他提高整治地位和享乐。曹操曾发誓希望能得到江东二乔（大乔和小乔），并将她们置于铜雀台上，以乐晚年 [Data: Relationships (6, 7); Entities (1, 2)]。

## 谋略与整治
在三国时期的权谋斗争中，诸葛亮曾提议使用美人计，将大乔和小乔送给曹操，以此来使曹操班师回朝，减少对东吴的威胁。这一提议显示出大乔在权谋中的重要角色以及她可能对大局产生的影响 [Data: Sources (0); Relationships (12)]。

## 曹操的策略和行动
尽管曹操对大乔有很大的兴趣，但实际行动中的证据有限。周瑜得知曹操对大乔的意图后，表达了强烈的反对，并与诸葛亮商议对策，以确保她和小乔的安全 [Data: Sources (2)]。周瑜明确表示大乔是孙策的正妻，而小乔是他的妻子，拒绝将她们送给曹操。

## 总结
总体而言，曹操和大乔没有直接的互动记录，但曹操对大乔的欲望及其可能带来的整治影响却是不容忽视的。大乔作为江东乔公的长女，其美貌和身份使她成为多方势力争夺的对象，但她与曹操的关系主要体现在曹操的欲望和诸葛亮、周瑜等人的权谋计划中。曹操对大乔的欲望和权谋体现了三国时期复杂的整治动态，同时也揭示了美人计在当时战略中的重要性 [Data: Relationships (6, 12, 13); Entities (1); Sources (0, 2)].

从回答结构上看，nano-graphrag 的答案更加结构化，明确分点陈述，并且直接引用了具体的实体和关系 ID，信息透明度更高。

Global Search 对比

同样以“大乔与曹操的关系”为例，两种实现在全局搜索模式下的表现如下。

### 大乔与曹操的关系

大乔是乔公的女儿，以其非凡的美貌而闻名。这一特质使她在社区中成为一个重要人物，尤其是在曹操对她的渴望背景下 [Data: Reports (41)]。

#### 曹操的计划

曹操计划在统一江南后，将大乔安置在铜雀台。铜雀台是曹操在叶县张河上建造的一座宏伟平台，象征着好运和庆祝。曹操打算将包括大乔在内的美丽女子安置在铜雀台中 [Data: Reports (269)]。这一计划不仅反映了曹操的野心和愿景，也显示了他对大乔的特别关注。

#### 乔公与曹操的关系

乔公是大乔的父亲，同时也是曹操的朋友。这层关系不仅突显了乔公的影响力，还暗示了由于曹操对大乔的渴望，可能会产生潜在的杠杆作用或冲突 [Data: Reports (41)]。

### 结论

综上所述，大乔与曹操的关系主要体现在曹操对她的渴望以及他计划将她安置在铜雀台的意图上。此外，乔公与曹操的友谊也为这一关系增添了复杂性，可能会在未来引发更多的动态变化 [Data: Reports (41, 269)]。

## 曹操与大乔的关系

曹操对大乔展现了显著的整治和战略兴趣，这部分是由于大乔的美貌以及她在战略上的重要性。在那个时期，美丽并非单纯的外貌特征，还往往与整治资源和战略权力相关联。曹操认为拥有大乔会对他的整治野心有所帮助，并为其权势扩大增添筹码 [Data: Reports (6)]。

大乔作为孙策的妻子，再加上她的地位和美貌，使得她在权力动态中成为一个关键的目标。曹操对大乔的重视，不仅因为她个人的吸引力，更因为她所代表的权力象征。如果能将大乔纳入自己的势力范围，曹操将可能通过这种方式削弱敌对势力，同时增强自己的影响力 [Data: Reports (6)]。

总之，曹操和大乔的关系不仅仅停留在个人层面，更深层次的是整治策略和权力博弈的体现。曹操对她的兴趣，反映了他在权力斗争中的精明与野心。

在全局搜索中，nano-graphrag 的回答更强调整治与战略层面的解读，而 GraphRAG 则着重于具体计划和人物关系。两者各有侧重，但 nano-graphrag 在信息密度和逻辑连贯性上表现不错。

如果你对这个项目感兴趣，可以直接访问 GitHub 仓库获取源码和更多示例：https://github.com/gusye1234/nano-graphrag