大家好,我是知白。一个专注于输出 AI+ 编程内容的大厂资深程序员,全国最大付费社群 AI 破局初创合伙人,关注我一起进步。

本文将从零开始构建本地知识库,从而辅助 ChatGPT 基于知识库内容生成回答。

对向量检索相关内容可以查看之前的文章:

初识 Embedding,为何大家都基于它搭建私人智能客服?

搭建 Embedding 向量数据库,手把手教你创造自己的私人智能助理

这里再重复下部分核心观点:

  1. 向量:将人类的语言(文字、图片、视频等)转换为计算机可识别的语言(数组)。

  2. 向量相似度:计算两个向量之间的相似度,表示两种语言的相似程度。

  3. 语言大模型的特性:上下文理解、总结和推理。

这三个概念结合起来,就构成了 "向量搜索 + 大模型 = 知识库问答" 的公式。

1.介绍

FastGPT 是目前 Prompt 串接做的最好的项目,知识库核心流程图如下:

从官方简介也可以看出很牛逼:

FastGPT 是一个基于 LLM 大语言模型的知识库问答系统,提供开箱即用的数据处理、模型调用等能力。同时可以通过 Flow 可视化进行工作流编排,从而实现复杂的问答场景!

FastGPT 官网地址:https://doc.fastgpt.in/docs/intro/

FastGPT 在线体验:https://fastgpt.run

2. 安装 Docker

话不多说直接开干,首选需要我们安装 Docker 应用,这里以 Windows 安装为例(其他系统可自行百度解决)。

Docker 官网 :https://www.docker.com/get-started/

进入官网双击下载的 Docker for Windows Installer 安装文件,一路 Next,点击 Finish 完成安装。

安装完成后,Docker 会自动启动。通知栏上会出现个小鲸鱼的图标,这表示 Docker 正在运行。

我装的 docker-desktop 自带了 docker-compose,如果未安装可以去官网进行下载。

Docker-Compose 官网:https://docs.docker.com/desktop/install/windows-install/

安装 Docker 还是比较简单,如果遇到大家可直接百度解决,网上这类文章很多。

3. 配置文件

先创建一个文件夹

`# 创建文件夹
mkdir fastgpt

# 进入文件夹
cd fastgpt

`

创建 config.json,内容如下:

{  "FeConfig": {    "show_emptyChat": false,    "show_contact": false,    "show_git": false,    "show_doc": true,    "systemTitle": "个人知识库",    "limit": {      "exportLimitMinutes": 0    },    "scripts": []  },  "SystemParams": {    "vectorMaxProcess": 15,    "qaMaxProcess": 15,    "pgIvfflatProbe": 20  },  "ChatModels": [    {      "model": "gpt-3.5-turbo",      "name": "GPT35-4k",      "contextMaxToken": 4000,      "quoteMaxToken": 2000,      "maxTemperature": 1.2,      "price": 0,      "defaultSystem": ""    },    {      "model": "gpt-3.5-turbo-16k",      "name": "GPT35-16k",      "contextMaxToken": 16000,      "quoteMaxToken": 8000,      "maxTemperature": 1.2,      "price": 0,      "defaultSystem": ""    },    {      "model": "gpt-4",      "name": "GPT4-8k",      "contextMaxToken": 8000,      "quoteMaxToken": 4000,      "maxTemperature": 1.2,      "price": 0,      "defaultSystem": ""    }  ],  "VectorModels": [    {      "model": "text-embedding-ada-002",      "name": "Embedding-2",      "price": 0,      "defaultToken": 500,      "maxToken": 3000    }  ],  "QAModel": {    "model": "gpt-3.5-turbo-16k",    "name": "GPT35-16k",    "maxToken": 16000,    "price": 0  },  "ExtractModel": {    "model": "gpt-3.5-turbo-16k",    "functionCall": true,    "name": "GPT35-16k",    "maxToken": 16000,    "price": 0,    "prompt": ""  },  "CQModel": {    "model": "gpt-3.5-turbo-16k",    "functionCall": true,    "name": "GPT35-16k",    "maxToken": 16000,    "price": 0,    "prompt": ""  },  "QGModel": {    "model": "gpt-3.5-turbo",    "name": "GPT35-4k",    "maxToken": 4000,    "price": 0,    "prompt": "",    "functionCall": false  }}

再创建 docker-compose.yml 文件,内容如下:

`# 非 host 版本, 不使用本机代理
version: '3.3'

services:
  pg:
    # 使用阿里云的 pgvector 镜像
    image: registry.cn-hangzhou.aliyuncs.com/fastgpt/pgvector:v0.4.2
    container_name: pg
    restart: always
    # 生产环境建议不要暴露端口
    ports:
      - "5432:5432"
    networks:
      - fastgpt
    # 环境变量配置,首次运行生效,修改后需删除持久化数据再重启
    environment:
      - POSTGRES_USER=username
      - POSTGRES_PASSWORD=password
      - POSTGRES_DB=fastgpt
    # 卷挂载,包括初始化脚本和数据持久化
    volumes:
      - ./pg/init.sql:/docker-entrypoint-initdb.d/init.sh
      - ./pg/data:/var/lib/postgresql/data

  mongo:
    # 使用阿里云的 mongo 镜像
    image: registry.cn-hangzhou.aliyuncs.com/fastgpt/mongo:5.0.18
    container_name: mongo
    restart: always
    # 生产环境建议不要暴露端口
    ports:
      - "27017:27017"
    networks:
      - fastgpt
    # 环境变量配置,首次运行生效,修改后需删除持久化数据再重启
    environment:
      - MONGO_INITDB_ROOT_USERNAME=username
      - MONGO_INITDB_ROOT_PASSWORD=password
    # 卷挂载,包括数据和日志
    volumes:
      - ./mongo/data:/data/db
      - ./mongo/logs:/var/log/mongodb

  fastgpt:
    container_name: fastgpt
    # 使用阿里云的 fastgpt 镜像
    image: registry.cn-hangzhou.aliyuncs.com/david_wang/fastgpt:latest
    ports:
      - "3000:3000"
    networks:
      - fastgpt
    # 确保在 mongo 和 pg 服务启动后再启动 fastgpt
    depends_on:
      - mongo
      - pg
    restart: always
    # 可配置的环境变量
    environment:
      - DEFAULT_ROOT_PSW=123456
      - OPENAI_BASE_URL=https://api.openai.com/v1
      - CHAT_API_KEY=sk-*****
      - DB_MAX_LINK=5
      - TOKEN_KEY=wenwenai
      - ROOT_KEY=wenwenai
      - FILE_TOKEN_KEY=filetoken
      - MONGODB_URI=mongodb://username:password@mongo:27017/fastgpt?authSource=admin
      - PG_URL=postgresql://username:password@pg:5432/fastgpt
    # 配置文件的卷挂载
    volumes:
      - ./config.json:/app/data/config.json

# 定义使用的网络
networks:
  fastgpt:

`

注意修改 docker-compose.yml 中的 CHAT_API_KEY 为你的 OpenAI Key 即可。

4. 启动

执行命令启动本地知识库:

# 在 docker-compose.yml 同级目录下执行 docker-compose pull docker-compose up -d

执行完成后就可以在浏览器上通过 http://localhost:3000/ 网址来访问个人知识库了。

基于上述操作我们已经成功访问到个人知识库页面,接下来带大家创建导入个人数据进行访问。

登录用户名为  root,密码为 docker-compose.yml 环境变量里设置的 DEFAULT_ROOT_PSW。

1. 创建知识库

成功登录后,新建一个知识库,这里将我的个人经历导入,所以取名为知白个人经历

通过文件将个人经历导入到知识库中。

确认后就开始将当前数据转化为向量数据。

全文大约2300多字,大概3~5分钟就导入完成了。由于文本限制问题,按照固定字数拆分为了8个数据集。

至此,我们的个人知识库已经建好了。我们尝试进行问答,这里的 0.7881 就是向量相似度,相似度越高的越靠前。

2. 使用知识库

创建一个应用来使用知识库。

这里简单设置了一下开场白,选择并绑定对应知识库。

开始对话,效果展示如下:

我们从零到一完成了本地个人知识库的搭建,整体花费时间也较短,刨除安装 Docker 的时间预计在10分钟左右。

后续对召回内容从多方面进行详细分析,从而使回答内容更加符合我们的预期。

关注我

一起做AI时代的弄潮儿!