Skip to content

Latest commit

 

History

History
80 lines (55 loc) · 1.75 KB

README.md

File metadata and controls

80 lines (55 loc) · 1.75 KB

DatabaseQueryGPT

通过自然语言询问,能够自动判断数据库中的表结构及数据结构。,生成SQL语句在数据库中检索信息,展示数据。 然后判断是否适合绘制图表并绘制,并做出最后总结。

一、效果展示

展示效果图1
展示效果图2
展示效果图3
展示效果图4
展示效果图5

二、代码简介

项目为简单Prompt工程,看似简单但功能实用而又全面。

背后所使用的大模型可以自行根据需求更换,当前举例了GPT4和KIMI。

  1. main.py: 主程序入口。仅包含前端页面。
  2. inference.py: 核心推理部分,包含了对话推理的主要逻辑。
  3. structure.py: 数据库结构分析部分,负责提取数据库所有表结构及执行SQL。

其他文件较为简单,不再赘述。

三、代码使用

  1. 安装依赖 openai matplotlib sqlalchemy pandas pymysql
pip install -r requirements.txt

  1. 填写大模型key,和数据库连接信息。信息放于secret目录下的txt文件。

  1. 确保数据库中有表,如果没有,可以使用tools/excel_to_sql.py将Excel转换为数据库表。

  1. 执行主程序
python main.py