ai-knowledge-graph

上传文档,AI 自动生成可视化知识图谱

开源 🏷️完全免费 🔧 仅开源项目,无在线服务,需自行部署 开源模型 🍴 复刻 420 🐛 待处理 8 👀 关注 25
GitHub ↗
GitHub 星标 ★ 3136
维护状态 维护中
是否开源 开源
定价模式 🏷️完全免费 开源项目,Apache-2.0许可证,可免费使用和部署。

项目预览

ai-knowledge-graph 项目界面预览图
图片来源:GitHub 官方仓库名片

项目数据

分类开源模型
开发团队robert-mcdermott
定价模式🏷️完全免费
价格说明开源项目,Apache-2.0许可证,可免费使用和部署。
访问状态🔧 仅开源项目,无在线服务,需自行部署
是否开源是
开源协议Apache-2.0
主要语言Python
技术栈/模型artificial-intelligence,knowledge-distillation,knowledge-graph,llm,networkx,pyvis,visualization
GitHub 星标★ 3136
上线时间2025-03-23 00:00:00
代码更新2026-09-13
仓库体积3.1 MB
本站数据更新2026-11-09
维护状态维护中
收录时间2026-08-09
浏览次数3

项目介绍

ai-knowledge-graph 是一个基于大语言模型自动生成知识图谱的开源工具。它解决的是从非结构化文本中手工构建知识图谱耗时费力的问题,利用 AI 自动抽取实体、关系并生成可视化图谱。核心能力包括:支持多种文档格式输入(如 PDF、Word、Markdown),通过 LLM 进行实体识别和关系抽取,提供交互式图谱可视化界面,支持图谱导出和增量更新。项目采用 Python 编写,易于集成到现有数据管道中。它适合需要快速从文档中提取结构化知识的场景,如企业知识管理、研究报告分析、学术文献梳理等。

Star 增长趋势

本站每日记录该项目 GitHub 星标变化,观测越久曲线越完整

数据来源:GitHub API 逐日快照 · 收录于 2026-08-09 · 当前星标 ★ 3136

使用教程

难度:入门 约 10 分钟 部署方式:命令行工具 5 步

环境要求

  • Python 3.11 或更高版本
  • pip(安装 requirements.txt)或 uv 包管理器
  • 一个 OpenAI 兼容的 LLM API 端点(Ollama、LM Studio、OpenAI、vLLM、LiteLLM 等任选其一)

安装与启动步骤

  1. 1克隆仓库

    把项目代码下载到本地并进入项目目录,后续所有命令都在该目录下执行。

    git clone https://github.com/robert-mcdermott/ai-knowledge-graph.git
    cd ai-knowledge-graph
  2. 2安装依赖

    README 提供两种方式:pip 安装 requirements.txt,或使用 uv 同步依赖,二者选一即可。

    pip install -r requirements.txt
  3. 3配置 config.toml

    编辑项目根目录的 config.toml,填写模型名、api_key 和 base_url,指向你的 OpenAI 兼容接口。

  4. 4生成知识图谱

    把待处理文本换成你的文件路径,运行脚本后会依次执行抽取、实体标准化和关系推断。

    python generate-graph.py --input your_text_file.txt --output knowledge_graph.html
  5. 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···

★ 197 2026-10-02
openchatbi 开源

用聊天代替写SQL,一句话查数出图

OpenChatBI is an intelligent chat-based BI tool powered by large language models, de···

★ 677 2026-09-12
Resume-Screening-RAG-Pipeline 开源

用聊天的方式,几秒筛完一堆简历

An LLM Chatbot that dynamically retrieves and processes resumes using RAG to perform···

★ 188 2026-08-09
kg-rag 开源

跟着书学 GraphRAG,用图数据库做出更聪明的聊天机器人

Supporting repository for the Essential GraphRAG book published by Manning

★ 193 2026-08-09
personal-ai-memory 开源

自动记住所有 AI 对话,本地私密搜索回顾

A local-first Chrome extension that passively captures ChatGPT, Gemini, Claude, Grok···

★ 71 2026-08-09
LightRAG 开源

轻量级 RAG 框架,用知识图谱让问答更准更快

[EMNLP2025] LightRAG: Simple and Fast Retrieval-Augmented Generation

★ 40610 2026-08-09
RAG_Techniques 开源

一站式学透 RAG 进阶技巧,从入门到实战快速上手

This repository showcases various advanced techniques for Retrieval-Augmented Genera···

★ 30273 2026-08-09
colette 开源

本地多模态 RAG,让技术文档里的图和表也能被搜到、问到

Multimodal RAG to search and interact locally with technical documents of any kind

★ 301 2026-10-12