大模型与交互式数据库查询

之前一直有个想法,就是用户可以输入一段文字,然后可以根据用户的输入生成一段SQL,从而在数据库中查询出相应的内容。
想法是很美好,但是现实却很残酷。自从有了大模型之后,一切似乎就变得很简单了。
下面是某同事做出来的一个效果,基于DeepSeek-R1:

可以看到当用户输入一段文字后,比如最近关于WordPress出现的一些中危漏洞,此时DeepSeek-R1根据输入提取出相应的信息,将其转换为SQL语句,再调用agent进行查询,就可以实现交互式的数据库查询了。
唯一不足的地方在于当前只支持单轮交互,由于没有使用langchain的历史记录,因此效果比较一般。另外也可以使用langgraph实现类似的效果。