深度学习-145-Text2SQL之基于官方提示词模板进行交互
文章目录
- 1 基于sqlite
-
- 1.1 数据库Chinook
-
- 1.1.1 创建并载入数据
- 1.1.2 SQLDatabase
- 1.2 数据库中的表
-
- 1.2.1 获取表的字段
- 1.2.2 翻译字段
- 1.3 建表语句
- 2 操作单表
-
- 2.1 大语言模型
- 2.2 数据库连接
- 2.3 官方提示词模板
-
- 2.3.1 一般输出
- 2.3.2 结构化输出
- 2.4 执行SQL查询
- 2.5 大模型整理结果
- 2.6 整体代码
- 3 操作多表
-
- 3.1 基本工具
- 3.2 用户交互
- 4 参考附录
1 基于sqlite
1.1 数据库Chinook
Chinook 数据库是作为 NorthWind 数据库的替代品创建的。它代表一个数字媒体商店,包括艺术家、专辑、媒体曲目、发票和客户的表格。
1.1.1 创建并载入数据
以下示例将使用SQLite与Chinook数据库的连接。
按照安装步骤在与此notebook相同的目录中创建Chinok.db:
(1)下载文件Chinook_Sqlite.sql。
(2)运行sqlite3 Chinook.db
(3)运行 .read Chinook_Sqlite.sql
(4)测试 SELECT * FROM Artist LIMIT 10
1.1.2 SQLDatabase
现在Chinhook.db已经在我们的目录中,可以使用SQLAlchemy驱动的SQLDatabase类与它进行接口对接:
from langchain_community