GraphRAG 源码学习
微软图增强 RAG:模块化图谱索引与全局/局部问答。这不是 API 速查表,而是一条可以重复走的源码路线:每天用 30 分钟,从一个真实相对路径出发,先用类比建立直觉,再把它接回整条运行链路。
开篇 · 这是个什么项目
GraphRAG 可以先想成一间社区摘要知识图谱。它的代码不是一堆孤立文件:入口接到需求,核心对象做判断,工具或模型完成工作,状态把结果交给下一个环节。
架构全景(动起来看)
发光扫过的每一层都是一次任务可能经过的站点。不同仓库命名不同,但「接入 → 编排 → 核心能力 → 资源」的责任边界很相似。
怎么使用这份教程
每天 30 分钟
6 讲,每讲先问问题,再给本质和大白话。
真实相对路径
所有路径都从项目根目录算,不把你的机器路径写进源码引用。
先验证再推理
每页都有可复制 bash 命令;文件版本变化时以本地实际内容为准。
可迁移方法
学会从入口、数据结构、循环和扩展点读任何 AI 项目。
从 README.md 认识项目大门
README.md:围绕 README.md 建立因果链:谁调用、处理什么、交给谁;对应阶段是「入口与地图」。
配置与依赖:pyproject.toml
pyproject.toml:围绕 pyproject.toml 建立因果链:谁调用、处理什么、交给谁;对应阶段是「入口与地图」。
顶层入口与启动路径
packages:围绕 packages 建立因果链:谁调用、处理什么、交给谁;对应阶段是「入口与地图」。
核心对象:template_engine_config.py
packages/graphrag-llm/graphrag_llm/config/template_engine_config.py:围绕 packages/graphrag-llm/graphrag_llm/config/template_engine_config.py 建立因果链:谁调用、处理什么、交给谁;对应阶段是「核心循环」。
运转循环里的 index.py
packages/graphrag/graphrag/api/index.py:围绕 packages/graphrag/graphrag/api/index.py 建立因果链:谁调用、处理什么、交给谁;对应阶段是「核心循环」。
协作主线穿过 index.py
packages/graphrag/graphrag/cli/index.py:围绕 packages/graphrag/graphrag/cli/index.py 建立因果链:谁调用、处理什么、交给谁;对应阶段是「核心循环」。
能力模块:main.py
packages/graphrag/graphrag/cli/main.py:围绕 packages/graphrag/graphrag/cli/main.py 建立因果链:谁调用、处理什么、交给谁;对应阶段是「核心能力」。
业务动作落在 cli.py
packages/graphrag/graphrag/utils/cli.py:围绕 packages/graphrag/graphrag/utils/cli.py 建立因果链:谁调用、处理什么、交给谁;对应阶段是「核心能力」。
把任务交给 pyproject.toml
packages/graphrag-cache/pyproject.toml:围绕 packages/graphrag-cache/pyproject.toml 建立因果链:谁调用、处理什么、交给谁;对应阶段是「核心能力」。
扩展点附近的 pyproject.toml
packages/graphrag-chunking/pyproject.toml:围绕 packages/graphrag-chunking/pyproject.toml 建立因果链:谁调用、处理什么、交给谁;对应阶段是「核心能力」。
状态如何进入 state.py
packages/graphrag/graphrag/index/typing/state.py:围绕 packages/graphrag/graphrag/index/typing/state.py 建立因果链:谁调用、处理什么、交给谁;对应阶段是「状态与协作」。
消息与协作:update_clean_state.py
packages/graphrag/graphrag/index/workflows/update_clean_state.py:围绕 packages/graphrag/graphrag/index/workflows/update_clean_state.py 建立因果链:谁调用、处理什么、交给谁;对应阶段是「状态与协作」。
持久化视角下的 state
unified-search-app/app/state:围绕 unified-search-app/app/state 建立因果链:谁调用、处理什么、交给谁;对应阶段是「状态与协作」。
模型侧连接:pyproject.toml
packages/graphrag-common/pyproject.toml:围绕 packages/graphrag-common/pyproject.toml 建立因果链:谁调用、处理什么、交给谁;对应阶段是「模型与扩展」。
工具/MCP 桥:pyproject.toml
packages/graphrag-input/pyproject.toml:围绕 packages/graphrag-input/pyproject.toml 建立因果链:谁调用、处理什么、交给谁;对应阶段是「模型与扩展」。
检索与知识:pyproject.toml
packages/graphrag-llm/pyproject.toml:围绕 packages/graphrag-llm/pyproject.toml 建立因果链:谁调用、处理什么、交给谁;对应阶段是「模型与扩展」。
集成层:pyproject.toml
packages/graphrag-storage/pyproject.toml:围绕 packages/graphrag-storage/pyproject.toml 建立因果链:谁调用、处理什么、交给谁;对应阶段是「模型与扩展」。
样例与复盘:config.toml
packages/graphrag-common/example_notebooks/config_files/config.toml:围绕 packages/graphrag-common/example_notebooks/config_files/config.toml 建立因果链:谁调用、处理什么、交给谁;对应阶段是「复盘与扩展」。
测试与边界:config.yaml
packages/graphrag-common/example_notebooks/config_files/config.yaml:围绕 packages/graphrag-common/example_notebooks/config_files/config.yaml 建立因果链:谁调用、处理什么、交给谁;对应阶段是「复盘与扩展」。
收官:沿 settings.yaml 串回全图
packages/graphrag-common/example_notebooks/config_files/settings.yaml:围绕 packages/graphrag-common/example_notebooks/config_files/settings.yaml 建立因果链:谁调用、处理什么、交给谁;对应阶段是「复盘与扩展」。