CourseGraph: 使用大模型自动构建课程知识图谱

中文 | English

CourseGraph 使用大模型，利用多种 prompt 优化技术, 自动从教材、书籍中抽取知识点, 构成以课程-章节-知识点为主题的知识图谱。为增加每个知识点的信息, CourseGraph 可以为知识点链接相应的习题、扩展阅读材料等资源, 另外也可利用多模态大模型从 pptx、图片、视频中提取信息并与之相关联。

🤔 局限性

• 目前只实现了基本的知识图谱抽取和对 pptx 的解析，效果有待优化
• 对视频的解析还处于规划中

📈 未来发展方向

• 改进提示词工程，并尝试使用 Agent 完成相关工作
• 基于图谱的问答 (KBQA 或 Graph-RAG)

🚀 快速使用

首先申请阿里云通义千问 API Key，然后选择使用本地安装

方式一：本地安装

安装依赖

请确保已安装 Anaconda (或 Miniconda) 和 Rust ，然后执行：

git clone [email protected]:wangtao2001/CourseGraph.git
cd CourseGraph
conda create -n cg python=3.10 -y
conda activate cg
pip install poetry
poetry config virtualenvs.create false
poetry install
cd rust
maturin develop
cd ..

linux 下还需安装 libreoffice 以完成文档转换，以 Debian 系为例: sudo apt install libreoffice

然后定位到文件 examples/get_knowledge_graph.py 中

修改图数据库信息

图数据库使用 Neo4j，需要提供连接地址和账号密码，如未安装请参考 Neo4j 文档

执行

python examples/get_knowledge_graph.py

📚 文档

可以在 docs 目录下查看文档, 也可以访问 在线文档 (由于项目功能仍处于快速开发中，故在线文档暂时还没有准备好)。如果你希望自定义在线文档请依照以下步骤：

依赖安装和预览

文档使用 VitePress 构建, 需安装 Node.js 18 或以上版本，然后执行：

cd docs
npm i
npm run docs:dev

使用浏览器打开 http://localhost:5173/ 即可进行预览

部署

在线文档使用 Github Actions + Github Pages 部署，描述文件在 .github/workflows/docs.yaml

🛠️ 贡献和引用

欢迎提交 PR 或 Issues，也欢迎参与任何形式的贡献

如果觉得 CourseGraph 项目有助于你的工作，请考虑如下引用:

 @misc{CourseGraph,
       author = {Wang, Tao},
       year = {2024},
       note = {https://github.com/wangtao2001/CourseGraph},
       title = {CourseGraph: Automatic Construction of Course Knowledge Graphs Using Large Models}
    }