通过Llama Index实现大模型的SQL生成的三种方法详解
文中使用了chatglm的llm和embedding modle,利用的智谱的免费token
Text2SQL其实就是从文本到SQL,也是NLP中的一种实践,这可以降低用户和数据库交互的门槛,无需懂SQL就可以拿到数据库数据。Text2SQL实现了从自然语言到SQL语言的生成,更加进一步的是直接给出数据中的结果。在开始之前我们需要一个能够操作的数据库,本文以sqlite为例。
0、创建demo数据库
为了演示,我们创建了一个简单的数据库(sqlite相关操作参考sqlite建立数据库,sqlalchemy的Session
),数据库中建立了简单的表2,语句如下:
CREATE TABLE `t2` (
`id` integer PRIMARY KEY autoincrement ,
`testcase_name` varchar(80) NOT NULL,
`create_user` varchar(80) DEFAULT NULL,
`create_date` varchar(80) DEFAULT NULL
) ;