DB-GPT is an experimental open-source project that uses localized GPT large models to interact with your data and environment. With this solution, you can be assured that there is no risk of data leakage, and your data is 100% private and secure.
Run on an RTX 4090 GPU.
Currently, we have released multiple key features, which are listed below to demonstrate our current capabilities:
Private KBQA & data processing
The DB-GPT project offers a range of features to enhance knowledge base construction and enable efficient storage and retrieval of both structured and unstructured data. These include built-in support for uploading multiple file formats, the ability to integrate plug-ins for custom data extraction, and unified vector storage and retrieval capabilities for managing large volumes of information.
Multiple data sources & visualization
The DB-GPT project enables seamless natural language interaction with various data sources, including Excel, databases, and data warehouses. It facilitates effortless querying and retrieval of information from these sources, allowing users to engage in intuitive conversations and obtain insights. Additionally, DB-GPT supports the generation of analysis reports, providing users with valuable summaries and interpretations of the data.
Multi-Agents&Plugins
It supports custom plug-ins to perform tasks, natively supports the Auto-GPT plug-in model, and the Agents protocol adopts the Agent Protocol standard.
Fine-tuning text2SQL
An automated fine-tuning lightweight framework built around large language models, Text2SQL data sets, LoRA/QLoRA/Pturning, and other fine-tuning methods, making TextSQL fine-tuning as convenient as an assembly line. DB-GPT-Hub
Multi LLMs Support, Supports multiple large language models, currently supporting
Massive model support, including dozens of large language models such as open source and API agents. Such as LLaMA/LLaMA2, Baichuan, ChatGLM, Wenxin, Tongyi, Zhipu, etc.
Etc.
Privacy and security
The privacy and security of data are ensured through various technologies, such as privatized large models and proxy desensitization.
Support Datasources
| DataSource | support | Notes |
|---|---|---|
| MySQL | Yes | |
| PostgreSQL | Yes | |
| Spark | Yes | |
| DuckDB | Yes | |
| Sqlite | Yes | |
| MSSQL | Yes | |
| ClickHouse | Yes | |
| Oracle | No | TODO |
| Redis | No | TODO |
| MongoDB | No | TODO |
| HBase | No | TODO |
| Doris | No | TODO |
| DB2 | No | TODO |
| Couchbase | No | TODO |
| Elasticsearch | No | TODO |
| OceanBase | No | TODO |
| TiDB | No | TODO |
| StarRocks | No | TODO |
The architecture of the entire DB-GPT is shown.
The core capabilities mainly consist of the following parts:
In the .env configuration file, modify the LANGUAGE parameter to switch to different languages. The default is English (Chinese: zh, English: en, other languages to be added later).
black . before submitting the code. Contributing guidelines, how to contribute
Multi Documents
RAG
Graph Database
Multi-Vector Database
Testing and Evaluation Capability Building
support llms
SFT Accuracy
As of October 10, 2023, by fine-tuning an open-source model of 13 billion parameters using this project, the execution accuracy on the Spider evaluation dataset has surpassed that of GPT-4!
| name | Execution Accuracy | reference |
|---|---|---|
| GPT-4 | 0.762 | numbersstation-eval-res |
| ChatGPT | 0.728 | numbersstation-eval-res |
| CodeLlama-13b-Instruct-hf_lora | 0.789 | sft train by our this project,only used spider train dataset ,the same eval way in this project with lora SFT |
| CodeLlama-13b-Instruct-hf_qlora | 0.774 | sft train by our this project,only used spider train dataset ,the same eval way in this project with qlora and nf4,bit4 SFT |
| wizardcoder | 0.610 | text-to-sql-wizardcoder |
| CodeLlama-13b-Instruct-hf | 0.556 | eval in this project default param |
| llama2_13b_hf_lora_best | 0.744 | sft train by our this project,only used spider train dataset ,the same eval way in this project |
More Information about Text2SQL finetune
The MIT License (MIT)
Content type
Image
Digest
sha256:9bb269844…
Size
7.1 GB
Last updated
about 1 month ago
docker pull eosphorosai/dbgpt