LangChain怎么连接数据库_LangChain数据查询实战教程

秋浩君_4016

秋浩君_4016

2026-02-18

604人浏览

原创

LangChain支持三种数据库查询方式:一、用SQLDatabaseChain实现自然语言转SQL查询;二、用SQLDatabaseToolkit配合Agent动态多步查询;三、自定义DatabaseQueryTool实现细粒度安全控制。

☞☞☞AI 智能聊天, 问答助手, AI 智能搜索, 多模态理解力帮你轻松跨越从0到1的创作门槛☜☜☜

langchain怎么连接数据库_langchain数据查询实战教程

如果您希望在LangChain应用中执行数据库查询操作,则需要通过适配器与数据库建立连接,并将查询结果注入到语言模型的上下文中。以下是实现LangChain连接数据库并完成数据查询的具体步骤:

一、使用SQLDatabaseChain连接关系型数据库

SQLDatabaseChain是LangChain提供的内置链,专用于连接SQL数据库并执行自然语言查询。它依赖SQLAlchemy作为底层驱动,支持PostgreSQL、MySQL、SQLite等主流关系型数据库。该方法无需手动编写SQL语句,模型可直接理解用户提问并生成对应查询。

1、安装必要依赖:pip install langchain sqlalchemy psycopg2-binary

2、导入模块并初始化数据库连接:from langchain.utilities import SQLDatabase;db = SQLDatabase.from_uri("postgresql+psycopg2://user:password@localhost:5432/mydb")

3、加载大语言模型并构建SQLDatabaseChain:from langchain.llms import OpenAI;from langchain.chains import SQLDatabaseChain;llm = OpenAI(temperature=0);chain = SQLDatabaseChain(llm=llm, database=db, verbose=True)

4、调用chain.run()执行自然语言查询:chain.run("查询用户表中年龄大于30的所有姓名")

二、通过SQLDatabaseToolkit配合Agent动态查询

SQLDatabaseToolkit结合Agent可实现更灵活的交互式数据库查询,支持多步推理、错误修正与元数据感知。它会自动加载表结构信息,并在执行前验证SQL语法合法性,适合复杂业务场景下的探索式查询。

1、创建SQLDatabase实例并传入schema信息:db = SQLDatabase.from_uri("sqlite:///example.db", include_tables=["users", "orders"])

2、初始化工具集:from langchain.agents import create_sql_agent;from langchain.agents.agent_toolkits import SQLDatabaseToolkit;toolkit = SQLDatabaseToolkit(db=db, llm=llm)

AI舌诊面诊问诊医疗系统+诊疗商城系统源码
AI舌诊面诊问诊医疗系统+诊疗商城系统源码

AI 舌诊面诊问诊医疗系统 + 诊疗商城系统源码是一套结合 AI 辅助诊断与商品转化的健康类程序,基于浩创云、ThinkPHP(PHP)开发,支持微信服务号与 H5 端,适合做中医舌诊面诊问答、健康建议推送以及配套诊疗商品的运营。

下载

3、构建Agent:agent_executor = create_sql_agent(llm=llm, toolkit=toolkit, verbose=True, agent_type="openai-tools")

4、发起查询请求:agent_executor.invoke({"input": "统计每个城市的订单总数"})

三、自定义DatabaseQueryTool实现细粒度控制

当内置链无法满足权限隔离、查询限流或审计日志等企业级需求时,可通过继承BaseTool类定义专用查询工具。该方式完全掌控输入解析、SQL生成、执行与结果格式化全过程,适用于高安全要求环境。

1、定义工具类并重写_run方法:class DatabaseQueryTool(BaseTool): name = "db_query";description = "Useful for querying database with custom SQL"

2、在_run中嵌入参数校验与白名单机制:if not re.match(r'^SELECT\s.+FROM\s\w+', query): raise ValueError("Only SELECT statements allowed")

3、使用execute()执行预编译语句并返回结构化结果:result = db.engine.execute(text(query)).fetchall()

4、将工具注册至Agent工具列表:tools = [DatabaseQueryTool(db=db)];agent = initialize_agent(tools, llm, agent="chat-zero-shot-react-description", verbose=True)

相关文章

PHP速学视频免费教程(入门到精通)
PHP速学视频免费教程(入门到精通)

PHP怎么学习?PHP怎么入门?PHP在哪学?PHP怎么学才快?不用担心,这里为大家提供了PHP速学教程(入门到精通),有需要的小伙伴保存下载就能学习啦!

下载

相关标签:

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

更多
Tailwind CSS 响应式布局怎么写
Tailwind CSS 响应式布局怎么写

Tailwind CSS 响应式布局通常从移动端样式写起,再用 sm、md、lg 等前缀调整不同屏幕宽度下的网格、间距和导航。本专题整理断点配置、Grid 与 Flex 自适应布局及常见溢出问题,帮助你逐步完成多设备适配。

2026.11.03

0

19

Tailwind CSS 常用类名怎么记
Tailwind CSS 常用类名怎么记

Tailwind CSS 类名很多,先按布局、间距、文字、颜色和状态分组记忆,再通过官方文档检索具体写法。本专题结合类名排序、动态拼接失效、自定义实用类等问题,帮助你建立常用类名的速查方法。

2026.11.03

0

12

码上飞入口地址合集
码上飞入口地址合集

本专题汇总了码上飞(CodeFlying)的官方入口地址合集,涵盖官网主站及网页版在线使用入口,支持注册登录后直接在线体验零代码AI应用生成。同时整理了APP端下载渠道与小程序入口,助你快速访问平台,通过自然语言描述需求即可生成应用、小程序或网站。

2026.11.03

0

10

Uvicorn 报错与问题解决教程
Uvicorn 报错与问题解决教程

本专题汇总了Uvicorn服务器常见问题与报错解决的教程合集,涵盖启动失败、端口占用、模块导入错误、事件循环冲突及热重载失效等典型故障的排查思路。同时整理了多进程与Worker配置、SSL/HTTPS证书报错、跨域与代理配置等实战解决方案,助你快速定位并解决Uvicorn运行中的各类问题。

2026.11.02

20

14

Uvicorn使用教程
Uvicorn使用教程

本专题汇总了Uvicorn使用教程与启动运行指南,涵盖其作为Python ASGI服务器的定位与特点,以及命令行启动(uvicorn main:app --reload)、参数配置(host/port/workers)、程序内嵌运行与Gunicorn结合部署等核心用法。同时整理了热重载调试、日志配置与生产环境部署优化等技巧,助你快速掌握Uvicorn的运行与配置。

2026.11.02

20

17

Uvicorn安装配置教程
Uvicorn安装配置教程

本专题汇总了Uvicorn的安装配置教程与部署指南,涵盖通过pip安装Uvicorn、启动FastAPI/Starlette应用、开发模式热重载(--reload)及多进程工作模式配置等核心操作。同时整理了结合Gunicorn的生产部署方案、HTTPS与反向代理配置及性能调优技巧,助你快速完成Uvicorn服务的安装与上线部署。

2026.11.02

20

17

Vite 项目如何打包和部署
Vite 项目如何打包和部署

Vite项目打包部署专题汇集生产构建、产物预览、输出目录、资源路径、Nginx与GitHub Pages部署及常见白屏和路由404问题的已审核文章,帮助读者完成上线流程。

2026.10.28

120

16

Vite 如何搭配 Vue.js
Vite 如何搭配 Vue.js

Vite搭配Vue.js专题汇集创建Vue 3项目、配置开发环境、路径别名、Vue插件、代理、多环境构建与调试的已审核文章,帮助读者完成从初始化到日常开发的配置。

2026.10.28

100

15

Vite 项目目录怎么看
Vite 项目目录怎么看

Vite项目目录专题汇集入口文件、src源码、public静态资源、路径别名、项目根目录和构建输出目录相关的已审核文章,帮助读者理解各文件夹的用途与引用方式。

2026.10.28

300

10

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程