2026年7月9日 周四晚上19:30,报名腾讯会议了解“如何构建自进化的动态知识库(Brain)”(限30人)
免费POC, 零成本试错
FDE知识库

FDE知识库

学习大模型的前沿技术与行业落地应用


收藏

让用户与数据库对话:使用Streamlit构建基于RAG的SQL助手

发布日期:2025-05-09 19:08:31 浏览次数: 2268
作者:PyTorch研习社

微信搜一搜,关注“PyTorch研习社”

推荐语

利用RAG技术,构建一个能与数据库对话的SQL助手,让自然语言查询成为可能。

核心内容:
1. RAG系统如何将自然语言转化为SQL查询
2. 技术栈介绍:LangChain、LangGraph、Streamlit
3. 系统架构和内部工作流程解析

杨芳贤
53AI创始人/腾讯云(TVP)最具价值专家

RAG(检索增强生成)系统通过利用大语言模型(LLM)并将其与特定数据源集成,使用户可以使用自然语言提出问题。

我将在本文重点介绍 RAG 的一个具体应用:将用户的自然语言转化成 SQL 查询并在数据库引擎上执行,最后以自然语言的形式返回结果。

最终我们将会得到一个能执行单表和联表查询的 SQL 助手:

我们的主要技术栈:

  • LangChain 用于连接 LLM 与数据库
  • LangGraph 用于管理多步骤工作流和内存
  • Streamlit 作为聊天机器人用户界面
  • 以 OpenAI API 兼容的形式调用阿里千问模型 

用于 SQL 的 RAG 系统

SQL 的 RAG 系统通过将 LLM 与真实数据库上下文结合,帮助生成更准确的结果。LLM 不仅仅依赖通用训练数据,而是利用检索到的上下文生成精确、实时的响应。这些上下文来自数据库的模式元数据。

当用户提出问题时,系统会自动将上下文与问题一起作为提示的一部分提供。这帮助 LLM 理解 SQL 方言、可用表、关系和列数据,从而构建语法和语义正确的查询。

这些查询随后提交给数据库引擎,检索到的结果被转化为自然语言。

我在本文的示例是使用 SQLite 数据库查询 Northwind 数据集,SQLite 是一个轻量级本地数据库,常用于教程和演示。Northwind 数据集是一个经典的零售数据集,包含客户、产品、订单和员工数据。

系统概览:工作原理

下图展示了整体系统架构——用户、数据库和大型语言模型如何通过应用层交互。

工作原理:内部工作流程和工具

应用层内的工作流程通常分为以下几个阶段:

工作流程背后的关键工具

  • LangChain 是一个框架,旨在帮助将 LLM 与外部系统(如 SQL 数据库、API 和文件存储)集成,简化 AI 应用的构建。

  • LangGraph 是 LangChain 的扩展,是一个编排框架,支持有状态的工作流程,允许系统在工作流程的多个步骤中保持上下文。

  • Streamlit 是一个开源 Python 库,可用于构建简单、交互式的 Web 应用,完全使用纯 Python。

技术细节

现在让我们逐步了解其构建过程——从项目设置到数据库连接、模型编排和前端集成。

项目结构和前提条件

    核心代码

    db_connections.py:负责管理数据库链接,以及获取数据库中表相关重要元数据的代码。

    sql_query_graph.py定义了从接收用户问题、编写 SQL 查询、执行查询到生成最终答案的 4 个核心函数,最后使用 LangGraph 将这 4 个核心函数组织成逻辑工作流。

    1. write_query:生成语法正确的 SQL 查询。

    • 使用选定的数据库,提取其方言和表信息。
    • 使用 SQL 特定的提示模板指导 LLM 生成与数据库结构匹配的 SQL 查询。
    2. execute_query:使用 QuerySQLDatabaseTool 函数对选定数据库执行生成的 SQL 查询,并检索结果。
    3. generate_answer:使用 SQL 查询结果生成对原始问题的自然语言回答。

    4. answer_question:创建基于图的执行流程,按逻辑顺序连接每个函数:

    • write_query → execute_query → generate_answer

    • 使用 StateGraph 创建图工作流,编译并运行。

    53AI,企业落地大模型首选服务商

    产品:场景落地咨询+大模型应用平台+行业解决方案

    承诺:免费POC验证,效果达标后再合作。零风险落地应用大模型,已交付160+中大型企业

    联系我们

    售前咨询
    186 6662 7370
    预约演示
    185 8882 0121

    微信扫码

    添加专属顾问

    回到顶部

    加载中...

    扫码咨询

    扫码登录
    登录即表示您同意《53AI网站服务协议》
    服务协议

    欢迎您使用【53AI 官方网站】(以下简称“本网站”或“我们”)。本《会员服务协议》(以下简称“本协议”)是您(以下简称“会员”或“用户”)与【深圳市博思协创网络科技有限公司】之间关于注册、登录及使用本网站会员服务所订立的法律协议。

    在您注册或登录前,请务必审慎阅读、充分理解各条款内容,特别是免除或限制责任的条款、知识产权条款、争议解决条款等。此类条款将以加粗形式提示您注意。 当您通过微信公众号授权、手机验证码验证或其他方式成功登录本网站时,即视为您已完全理解并同意接受本协议的全部内容。

    一、 定义

    本网站:指由【深圳市博思协创网络科技有限公司】运营的,域名为【53ai.com】的网站及相关移动端页面。

    会员服务:指本网站向注册会员提供的知识库文章查阅、内容检索及其他相关增值服务。

    知识库内容:指本网站发布的包括但不限于文字、图表、数据、研究报告、行业分析等数字化内容资源。

    二、 账号注册与登录

    登录方式:本网站支持以下登录方式,您可根据实际情况选择:

    微信公众号授权登录:您同意将您的微信OpenID信息授权给本网站,用于创建或关联会员账号。

    手机验证码登录:您需提供真实有效的手机号码,并通过短信验证码完成身份验证与登录/注册。

    账号安全:您的账号仅限您本人使用,禁止赠与、借用、租用、转让或售卖。因您保管不善导致的账号被盗、密码泄露等损失,由您自行承担。

    实名认证:根据相关法律法规要求,我们可能要求您在特定功能下完成实名认证。如您拒绝提供,可能无法使用部分或全部服务。

    未成年人保护:若您未满18周岁,请在法定监护人的陪同下阅读本协议,并在征得监护人同意后使用本服务。

    三、 服务内容与规范

    知识库查阅权限:会员登录后,有权按照其会员等级对应的权限范围,在线浏览、检索本网站知识库中的相关文章及内容。

    服务变更:我们有权根据业务发展需要,调整、变更或终止部分服务内容,并将以网站公告、公众号消息等方式提前通知。

    禁止行为:您在使用服务时不得实施以下行为:

    利用技术手段批量爬取、下载、转存知识库内容;

    将知识库内容用于商业目的或未经授权地向第三方传播;

    干扰本网站正常运行或侵犯其他用户合法权益;

    发布违法违规信息或从事违反公序良俗的活动。

    四、 知识产权声明

    权利归属:本网站知识库中的排版设计、软件代码等内容的知识产权均归【公司全称】或原权利人所有,受《中华人民共和国著作权法》等法律保护。

    有限许可:本网站授予会员一项非独占、不可转让、不可转授权的普通许可,仅限于个人学习、研究之目的在线查阅知识库内容。

    侵权追责:未经书面许可,任何单位或个人不得以任何形式复制、转载、摘编、镜像、汇编或以其他方式使用上述内容。一经发现,我们保留追究其法律责任的权利。

    五、 个人信息保护

    我们重视对您个人信息的保护。关于我们如何收集、使用、存储和保护您的个人信息,请单独阅读 《隐私政策》。

    您通过微信公众号授权或手机号验证所提供的信息,我们将严格按照《个人信息保护法》的规定处理,仅用于身份识别、服务提供及安全验证等必要用途。

    您可以随时通过网站设置或联系客服行使查阅、更正、删除个人信息及撤回授权同意的权利。

    六、 免责声明

    内容准确性:知识库内容仅供参考,不构成专业建议。我们不对其完整性、准确性、时效性作任何明示或暗示的保证,您应自行判断并承担使用风险。

    不可抗力:因自然灾害、政策法规变化、网络故障、第三方平台接口异常(如微信接口维护、运营商短信通道故障)等不可抗力导致的服务中断或延迟,我们不承担违约责任。

    第三方链接:本网站可能包含指向第三方网站的链接,该等网站的内容和服务不受我们控制,请您自行甄别风险。

    七、 违约责任

    如您违反本协议约定,我们有权视情节采取警告、限制功能、暂停服务、注销账号等措施,并保留要求赔偿损失的权利。

    如因您的违约行为导致我们遭受行政处罚、第三方索赔或商誉损失,您应承担全部赔偿责任(包括但不限于罚款、赔偿金、律师费、公证费等)。

    八、 法律适用与争议解决

    本协议的订立、执行和解释均适用中华人民共和国大陆地区法律。

    因本协议产生的或与本协议有关的任何争议,双方应友好协商解决;协商不成的,任何一方均可向【公司所在地】有管辖权的人民法院提起诉讼。

    九、 其他

    本协议构成双方就本服务达成的完整协议,取代此前任何口头或书面约定。

    本协议任一条款被认定为无效或不可执行的,不影响其他条款的效力。

    我们对本协议享有最终解释权,并在法律允许的范围内保留随时修改的权利。修改后的协议一经公布即生效,继续使用服务即视为同意修订内容。


    已查阅