项目简介
这是一个借助ChatGLM和Qdrant向量数据库搭建的党建问答助手系统,支持用户通过Web界面与党建知识库交互,获取党建相关知识及答案,为用户学习党建知识提供便利。
项目的主要特性和功能
- 实时聊天:用户能通过Web界面实时与系统交流,获取党建解答。
- 知识库查询:可查询党建知识库,返回相关答案和解释。
- 句子向量化:利用句子嵌入模型将用户输入句子转为向量,便于在知识库搜索匹配。
- 数据库支持:采用Qdrant向量数据库存储和检索党建知识信息。
- 自定义模型:支持使用自定义的ChatGLM模型和句子嵌入模型。
- 界面友好:提供简洁的Web界面,方便用户操作。
安装使用步骤
前期准备
- 安装Python环境并配置好环境变量。
- 使用pip安装依赖库,在项目根目录运行
pip install -r requirements.txt
。 - 安装Qdrant向量数据库,使用docker启动,命令如下:
bash docker run -p 6333:6333 \ -v $(pwd)/qdrant_storage:/qdrant/storage:z \ qdrant/qdrant
- 下载模型文件:
- ChatGLM模型:从https://aistudio.baidu.com/datasetdetail/246362 下载并解压到
model
文件夹。 - 句子向量化模型:使用sentenceTransformer,放置在
model/EmbeddingModel
文件夹。
- ChatGLM模型:从https://aistudio.baidu.com/datasetdetail/246362 下载并解压到
运行程序
- 启动数据库后,运行
data_import
文件夹中的数据添加程序。 - 可在
chat_app.py
中指明模型路径。 - 在终端运行
python test_with_web.py
启动Web服务器。 - 在浏览器输入
http://localhost:8080/
访问系统界面。
下载地址
点击下载 【提取码: 4003】【解压密码: www.makuang.net】