PDF 本地知识库

项目说明

基于 FastAPI 快速开发框架，整合了 LangChain、OpenAI、FAISS等技术链，实现了基于PDF文档构建问答知识库

如何使用

环境配置

• 使用 python3 版本，测试通过版本为3.10.x

OpenAI

修改项目下的 .env 文件，项目中使用到 api-key 和 api-base 代理，如果不需要使用代理，可以注释tool目录下 llm.py 和 store.py 的相应配置

创建虚拟环境

#当前目录创建虚拟环境
python3 -m venv .venv

#激活虚拟环境
source venv/bin/activate

#安装依赖
pip3 install -r requirements.txt

初始化PDF文件

修改 init_pdf.py 脚本的 pdf_path 路径，然后执行此脚本，成功后默认会生成 faiss 目录为向量数据库文件

python init_pdf.py

项目启动

#其中python为虚拟环境bin目录下的
python -m uvicorn main:app --reload 

接口调用

详情可查看 chat.py 文件

# 调用问答接口
http://127.0.0.1:8000/chat/question