前置准备

  1. # 源码下载
  2. $ git clone https://github.com/eosphoros-ai/DB-GPT.git
  3. # 进入到源码目录
  4. $ cd DB-GPT

环境安装

  1. # 创建Python环境,注意版本3.10以上
  2. $ conda create -n dbgpt_env python=3.10
  3. # 激活虚拟环境
  4. $ conda activate dbgpt_env

安装依赖

  1. # 安装默认依赖
  2. pip install -e ".[default]"

模型下载

下载语言模型与Embedding模型

:::danger ⚠️ 如果没有GPU资源,建议使用代理模型,如果OpenAI、千文、文心等

:::

  1. $ mkdir models && cd models
  2. # download embedding model, eg: text2vec-large-chinese
  3. $ git clone https://huggingface.co/GanymedeNil/text2vec-large-chinese

:::color1 设置代理API,修改.env配置

:::

  1. #set LLM_MODEL TYPE
  2. LLM_MODEL=proxyllm
  3. #set your Proxy Api key and Proxy Server url
  4. PROXY_API_KEY={your-openai-sk}
  5. PROXY_SERVER_URL=https://api.openai.com/v1/chat/completions

:::danger ⚠️ 有GPU资源,可以使用本地模型来部署

:::

  1. $ mkdir models && cd models
  2. # # download embedding model, eg: vicuna-13b-v1.5 or
  3. $ git clone https://huggingface.co/lmsys/vicuna-13b-v1.5
  4. # download embedding model, eg: text2vec-large-chinese
  5. $ git clone https://huggingface.co/GanymedeNil/text2vec-large-chinese
  6. $ popd

命令行一键启动

  1. LLM_MODEL=vicuna-13b-v1.5
  2. dbgpt start webserver --port 6006

默认情况下,dbgpt start webserver命令将启动webservermodel controllermodel worker通过一个单Python进程。 如上命令中,指定了6006端口。

查看并验证模型服务

:::color2 查看并展示所有的模型服务

:::

  1. dbgpt model list
  1. +-----------------+------------+------------+------+---------+---------+-----------------+----------------------------+
  2. | Model Name | Model Type | Host | Port | Healthy | Enabled | Prompt Template | Last Heartbeat |
  3. +-----------------+------------+------------+------+---------+---------+-----------------+----------------------------+
  4. | vicuna-13b-v1.5 | llm | 172.17.0.9 | 6006 | True | True | | 2023-10-16T19:49:59.201313 |
  5. | WorkerManager | service | 172.17.0.9 | 6006 | True | True | | 2023-10-16T19:49:59.246756 |
  6. +-----------------+------------+------------+------+---------+---------+-----------------+----------------------------+

其中WorkerManager是Model Workers的管理进程

:::color2 检查并验证模型服务

:::

  1. dbgpt model chat --model_name vicuna-13b-v1.5

上述命令将启动一个交互式页面,可以通过终端与模型对话。

  1. Chatbot started with model vicuna-13b-v1.5. Type 'exit' to leave the chat.
  2. You: Hello
  3. Bot: Hello! How can I assist you today?
  4. You: