vitali87/code-graph-rag Vibe 项目

code-graph-rag 是一款面向大型代码库的 AI 知识图谱检索工具，通过 Tree-sitter 解析多语言代码库，将函数、类、方法、模块及其关系构建为 Memgraph 知识图谱，支持用自然语言查询、理解并编辑代码库。与传统基于文本的 RAG 不同，它将代码结构显式建模为图数据库节点，支持跨语言统一查询（如 Python、TypeScript、Go、Java、Rust、C++ 等 13 种语言），并可追踪数据流（Data-Flow Tracing）找出污点传播路径。该项目包含两个核心组件：多语言 Tree-sitter 解析器负责提取代码结构并写入 Memgraph；RAG 系统负责将自然语言转换为 Cypher 查询，从图中检索相关代码并交由 AI 完成编辑和优化。系统支持 AST 级别的精准代码搜索与替换、用 ast-grep 实现结构化模式匹配，以及死代码检测和代码质量审计。适合需要处理大型单体仓库或混合语言代码库的团队，帮助 AI 编程工具更好地理解代码上下文并生成精准的编辑建议。