# ChatSQL
基于ChatGLM-6B/MOSS,实现nl2sql,直连数据库并返回查询结果
目前仅支持MYSQL语法,后续支持多数据库语法查询
* 目前观察下来,ChatGLM-6B多轮交互上限在3-4轮,之后就无法正确生成准确的SQL语句
## ✨ 整体思路
![](figure/ChatSQL技术流程图.png)
整体思路如上,目前采用yaml文件代替Table_info表结构
## 🎬 开始
```
git clone git@github.com:yysirs/ChatSQL.git
cd ChatSQL
conda create -n chatsql python=3.9
conda activate chatsql
pip install -r requirements.txt
# 新建文件夹
mkdir DB
mkdir logs
# 生成本地数据库+插入数据
python local_database.py
# 基于GLM生成SQL
python main_gui.py
或者 基于MOSS生成SQL
python main_gui_moss.py
```
## 😁 效果演示
![](figure/ChatSQL演示图2.png)
## 👍 特性
- 🛒 支持多表联查
- 🖼️ 2023/04/24 支持web前端
- 🎉 2023/04/24 支持yaml自定义数据库schema
- 😁 2023/04/25 支持yaml自定义数据
- 🎗️ 2023/04/25 支持直连本地数据库查询,验证SQL是否正确
- 👌 2023/04/30 支持MOSS大模型
## 🔍 各种类型的查询
```
# 单表多条件查询
请帮我查询在2019年的货物销售的净收益率大于10的货物名称
# 两表联查
请帮我查询在2019年的净收益率大于10并且销售量大于100的销售负责人名字
# 两表多条件联查
请帮我查询在2019年的货物的净收益率大于10并且销售量大于100并且销售负责人业绩大于1000的销售负责人名字
# max
请帮我查询货物销售量最大的货物名称
# min
请帮我查询货物销售量最小的货物名称
# COUNT
请帮我查询在2019年的货物销售的净收益率大于10的货物名称的数目
# AVG
请帮我查询2019年以及2020年货物销售量的平均值
# GROUP BY
请根据年份进行分组查询货物销售量和年份
# ORDER BY
请帮我按照数量大小对货物名称进行排序
# SUM
请帮我查询货物的销售量总和是多少
```
没有合适的资源?快使用搜索试试~ 我知道了~
基于ChatGLM的数据库查询代码NL2SQL.zip
共25个文件
py:17个
png:3个
txt:1个
1.该资源内容由用户上传,如若侵权请联系客服进行举报
2.虚拟产品一经售出概不退款(资源遇到问题,请及时私信上传者)
2.虚拟产品一经售出概不退款(资源遇到问题,请及时私信上传者)
版权申诉
5星 · 超过95%的资源 5 下载量 127 浏览量
2023-05-24
16:05:44
上传
评论 11
收藏 351KB ZIP 举报
温馨提示
本项目是研究大语言模型的应用案例,内含详细的教程以及docker环境,可以直接快速部署应用
资源推荐
资源详情
资源评论
收起资源包目录
基于ChatGLM的数据库查询代码NL2SQL.zip (25个子文件)
基于ChatGLM的数据库查询代码NL2SQL
utils.py 2KB
main_gui_moss.py 8KB
main_gui.py 5KB
utility
utils.py 1KB
__init__.py 0B
db_tools.py 4KB
constant.py 227B
loggers.py 1KB
local_database.py 2KB
main.py 4KB
figure
ChatSQL-HF.png 16KB
ChatSQL技术流程图.png 64KB
ChatSQL演示图2.png 85KB
ChatSQL演示图.jpg 219KB
config.yaml 7KB
prompt.py 10KB
requirements.txt 125B
models
tokenization_moss.py 16KB
configuration_moss.py 5KB
quantization.py 18KB
modeling_moss.py 31KB
custom_autotune.py 7KB
config.cfg 0B
demo.py 241B
README.md 2KB
共 25 条
- 1
资源评论
- 一帆风顺7122024-04-06这个资源总结的也太全面了吧,内容详实,对我帮助很大。
- weixin_481174282023-06-28感谢资源主分享的资源解决了我当下的问题,非常有用的资源。
- m0_515689792024-04-21终于找到了超赞的宝藏资源,果断冲冲冲,支持!
- laimiaohua2023-10-28资源内容总结地很全面,值得借鉴,对我来说很有用,解决了我的燃眉之急。
- bbmpg1234562023-10-28非常有用的资源,有一定的参考价值,受益匪浅,值得下载。
小黑码蚁
- 粉丝: 2406
- 资源: 1976
上传资源 快速赚钱
- 我的内容管理 展开
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功