DB-GPT 来了，兼职奶爸搞了个数据库创新AI工具！

昨晚，在咱数据库交流群里，有朋友分享了个项目，点开一看，瞬间被炸醒~

放大瞅瞅👇默认知识库对话

有趣的东西，值得分享给更多朋友，一起来玩~

一个数据库相关的 GPT 实验项目, 模型与数据全部本地化部署, 绝对保障数据的隐私安全。同时此 GPT 项目可以直接本地部署连接到私有数据库, 进行私有数据处理。

DB-GPT 是一个实验性的开源应用程序，它基于 FastChat，并使用 vicuna-13b 作为基础模型。此外，此程序结合了 langchain 和 llama-index 基于现有知识库进行 In-Context Learning 来对其进行数据库相关知识的增强。它可以进行 SQL 生成、SQL 诊断、数据库知识问答等一系列的工作。

Run on an RTX 4090 GPU

1. 运行演示

2. SQL 生成示例 

首先选择对应的数据库, 然后模型即可根据对应的数据库 Schema 信息生成 SQL.

3. 数据库 QA 示例

1. 首先你需要安装 python

python>=3.9
pip install -r requirements.txt
复制

或者如果你使用 conda 环境，你可以使用这个命令：

cd DB-GPT
conda env create -f environment.yml
复制

2. MySQL 安装

在这个项目示例中，我们连接 MySQL 并运行 SQL-Generate，所以你需要在本地安装 MySQL 进行测试，推荐使用 Docker。

docker run --name=mysql -p 3306:3306 -e MYSQL_ROOT_PASSWORD=aa123456 -dit mysql:latest
复制

密码只是为了测试，如果需要你可以改变。

1. 基础模型下载 

关于基础模型，可以根据 vicuna 合成教程进行合成。如果此步有困难的同学，也可以直接使用 Hugging Face上的模型进行替代。

2. Run model server

cd pilot/server
python vicuna_server.py
复制

3. Run gradio webui

python webserver.py
复制

Features

• SQL-Generate
• Database-QA Based Knowledge
• SQL-diagnosis

复制