Merge branch 'dev' of https://github.com/csunny/DB-GPT into dev

This commit is contained in:
csunny 2023-05-03 16:26:09 +08:00
commit 917cf37e9f
6 changed files with 22 additions and 6 deletions

2
.gitignore vendored
View File

@ -129,3 +129,5 @@ dmypy.json
# Pyre type checker
.pyre/
.DS_Store
logs

View File

@ -3,10 +3,24 @@ A Open Database-GPT Experiment
![GitHub Repo stars](https://img.shields.io/github/stars/csunny/db-gpt?style=social)
DB-GPT is an experimental open-source application that builds upon the fastchat model and uses vicuna as its base model. Additionally, it looks like this application incorporates langchain and llama-index embedding knowledge to improve Database-QA capabilities.
Overall, it appears to be a sophisticated and innovative tool for working with databases. If you have any specific questions about how to use or implement DB-GPT in your work, please let me know and I'll do my best to assist you.
Run on an RTX 4090 GPU (The origin mov not sped up!, [YouTube地址](https://www.youtube.com/watch?v=1PWI6F89LPo))
- 运行演示
![](https://github.com/csunny/DB-GPT/blob/dev/asserts/演示.gif)
- SQL生成示例
<img src="https://github.com/csunny/DB-GPT/blob/dev/asserts/sql_generate.png" width="600" margin-left="auto" margin-right="auto" >
- 数据库QA示例
<img src="https://github.com/csunny/DB-GPT/blob/dev/asserts/DB_QA.png" margin-left="auto" margin-right="auto" width="600">
# Install
1. Run model server
```
@ -16,10 +30,10 @@ python vicuna_server.py
2. Run gradio webui
```
python app.py
python webserver.py
```
# Featurs
- SQL-Generate
- Database-QA Based Knowledge
- SQL-diagnosis
- SQL-diagnosis

BIN
asserts/DB_QA.png Normal file

Binary file not shown.

After

Width:  |  Height:  |  Size: 243 KiB

BIN
asserts/sql_generate.png Normal file

Binary file not shown.

After

Width:  |  Height:  |  Size: 164 KiB

BIN
asserts/演示.gif Normal file

Binary file not shown.

After

Width:  |  Height:  |  Size: 3.0 MiB

View File

@ -89,8 +89,8 @@ class Conversation:
conv_one_shot = Conversation(
system="A chat between a curious human and an artificial intelligence assistant."
"The assistant gives helpful, detailed and polite answers to the human's questions. ",
system="A chat between a curious human and an artificial intelligence assistant, who very familiar with database related knowledge. "
"The assistant gives helpful, detailed, professional and polite answers to the human's questions. ",
roles=("Human", "Assistant"),
messages=(
(
@ -123,8 +123,8 @@ conv_one_shot = Conversation(
)
conv_vicuna_v1 = Conversation(
system = "A chat between a curious user and an artificial intelligence assistant. "
"The assistant gives helpful, detailed and polite answers to the user's questions. ",
system = "A chat between a curious user and an artificial intelligence assistant. who very familiar with database related knowledge. "
"The assistant gives helpful, detailed, professional and polite answers to the user's questions. ",
roles=("USER", "ASSISTANT"),
messages=(),
offset=0,