
ai-knowledge-graph
上传文档,AI 自动生成可视化知识图谱
项目预览
项目数据
项目介绍
Star 增长趋势
本站每日记录该项目 GitHub 星标变化,观测越久曲线越完整
数据来源:GitHub API 逐日快照 · 收录于 2026-08-09 · 当前星标 ★ 3136
使用教程
环境要求
- Python 3.11 或更高版本
- pip(安装 requirements.txt)或 uv 包管理器
- 一个 OpenAI 兼容的 LLM API 端点(Ollama、LM Studio、OpenAI、vLLM、LiteLLM 等任选其一)
安装与启动步骤
-
1克隆仓库
把项目代码下载到本地并进入项目目录,后续所有命令都在该目录下执行。
git clone https://github.com/robert-mcdermott/ai-knowledge-graph.git cd ai-knowledge-graph -
2安装依赖
README 提供两种方式:pip 安装 requirements.txt,或使用 uv 同步依赖,二者选一即可。
pip install -r requirements.txt -
3配置 config.toml
编辑项目根目录的 config.toml,填写模型名、api_key 和 base_url,指向你的 OpenAI 兼容接口。
-
4生成知识图谱
把待处理文本换成你的文件路径,运行脚本后会依次执行抽取、实体标准化和关系推断。
python generate-graph.py --input your_text_file.txt --output knowledge_graph.html -
5打开可视化结果
运行结束会在 --output 指定位置生成 HTML 文件,用浏览器打开即可查看交互式知识图谱。
关键配置
| 配置项 | 必填 | 说明 | 示例 |
|---|---|---|---|
llm.model | 是 | 使用的模型名称,需与所连端点提供的模型一致 | gemma3 |
llm.api_key | 是 | 调用 LLM 的密钥,本地 Ollama 也需填占位值 | sk-1234 |
llm.base_url | 是 | OpenAI 兼容接口地址,可指向本地或云端服务 | http://localhost:11434/v1/chat/completions |
llm.max_tokens | 否 | 单次请求最大输出 token 数 | 8192 |
chunking.chunk_size | 否 | 文本分块大小,按词数计算 | 200 |
chunking.overlap | 否 | 相邻分块之间的重叠词数,用于保持上下文 | 20 |
如何确认成功
控制台最后打印 Knowledge graph visualization saved to ... 并生成 HTML 文件,浏览器打开该文件能看到可交互图谱。
常见问题
Q:可以用哪些大模型?
A:任何 OpenAI 兼容端点都可以,例如 Ollama、LM Studio、OpenAI、vLLM、LiteLLM,通过 base_url 和 api_key 指定即可。
Q:不想做实体标准化或关系推断怎么办?
A:运行命令时加上 --no-standardize 和 --no-inference 参数即可关闭,也可在 config.toml 中把对应 enabled 设为 false。
Q:只想先看看效果,没有文本文件怎么办?
A:使用 --test 参数可以基于示例数据生成一张测试可视化图,无需提供输入文件。
Q:出现抽取结果异常如何排查?
A:加上 --debug 参数运行,会输出 LLM 的原始响应和抽取出的 JSON,便于定位问题。
Q:可以用 uv 或作为模块安装吗?
A:可以。uv 方式执行 uv run generate-graph.py --input ... --output ...;模块方式先 pip install --upgrade -e . 再执行 generate-graph --input ... --output ...。
注意事项
- Python 版本必须为 3.11 及以上,否则可能无法运行。
- config.toml 默认在项目根目录,可用 --config 参数指定其他路径的配置文件。
- 长文档会分成多个块逐块调用 LLM,处理时间较长且会产生相应 API 消耗。
- 生成的 HTML 图谱与同名 JSON 原始数据一并输出,JSON 可用于后续处理或增量更新。
核心亮点
- 自动抽取实体与关系,省去手工建模时间
- 支持 PDF/Word/Markdown 等多格式输入
- 提供交互式可视化界面,图谱直观可探索
不足之处
- 依赖 LLM API,调用成本可能较高
- 文档/社区待观察
适用场景
- 企业知识库构建与维护
- 学术文献综述与关系梳理
- 市场研究报告结构化分析
替代项目
GraphGPT、Knowledge Graph Generator (KGG)、Text2KG
上一篇:arxiv-mcp-server
下一篇:contextgem
同类项目推荐
CatWiki
开源
把团队文档变成能对话的知识库,问一句就有答案
A fully functional knowledge base platform offering robust content management, AI-po···
openchatbi
开源
用聊天代替写SQL,一句话查数出图
OpenChatBI is an intelligent chat-based BI tool powered by large language models, de···
Awesome-Generative-Recommendatio···
开源
一页看懂生成式推荐工业落地论文脉络
Awesome Generative Recommendation papers primarily focused on industry-level applica···
Resume-Screening-RAG-Pipeline
开源
用聊天的方式,几秒筛完一堆简历
An LLM Chatbot that dynamically retrieves and processes resumes using RAG to perform···
kg-rag
开源
跟着书学 GraphRAG,用图数据库做出更聪明的聊天机器人
Supporting repository for the Essential GraphRAG book published by Manning
personal-ai-memory
开源
自动记住所有 AI 对话,本地私密搜索回顾
A local-first Chrome extension that passively captures ChatGPT, Gemini, Claude, Grok···
LightRAG
开源
轻量级 RAG 框架,用知识图谱让问答更准更快
[EMNLP2025] LightRAG: Simple and Fast Retrieval-Augmented Generation
RAG_Techniques
开源
一站式学透 RAG 进阶技巧,从入门到实战快速上手
This repository showcases various advanced techniques for Retrieval-Augmented Genera···
colette
开源
本地多模态 RAG,让技术文档里的图和表也能被搜到、问到
Multimodal RAG to search and interact locally with technical documents of any kind