diff --git a/.gitignore b/.gitignore index 197b17d7a..4b46e7091 100644 --- a/.gitignore +++ b/.gitignore @@ -129,3 +129,5 @@ dmypy.json # Pyre type checker .pyre/ +.DS_Store +logs \ No newline at end of file diff --git a/README.md b/README.md index 97ce3b9bf..f7dfd6524 100644 --- a/README.md +++ b/README.md @@ -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生成示例 + + + +- 数据库QA示例 + + + # 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 \ No newline at end of file +- SQL-diagnosis diff --git a/asserts/DB_QA.png b/asserts/DB_QA.png new file mode 100644 index 000000000..46bd327d0 Binary files /dev/null and b/asserts/DB_QA.png differ diff --git a/asserts/sql_generate.png b/asserts/sql_generate.png new file mode 100644 index 000000000..13fb44af0 Binary files /dev/null and b/asserts/sql_generate.png differ diff --git a/asserts/演示.gif b/asserts/演示.gif new file mode 100644 index 000000000..22bab31ab Binary files /dev/null and b/asserts/演示.gif differ diff --git a/pilot/conversation.py b/pilot/conversation.py index 3a702c615..8e172e7dd 100644 --- a/pilot/conversation.py +++ b/pilot/conversation.py @@ -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,