Grok与数据库连接:利用SQLAgent实现自然语言查询数据库

小辰大大_2100

小辰大大_2100

2026-06-16

854人浏览

原创

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

grok支持自然语言查询数据库,自动解析为sql并执行;需配置数据库连接信息、安装langchain-community等依赖、初始化sqldatabase与chatopenai模型,并调用create_sql_agent构建agent。

用自然语言向数据库提问,比如“上个月销售额最高的三个产品是什么”,让grok自动解析成sql并执行返回结果,无需手动写查询语句。

准备数据库连接信息

确认数据库类型(PostgreSQL/MySQL/SQLite)、主机地址、端口、数据库名、用户名和密码。这些信息将用于构建SQLAlchemy连接字符串。

PostgreSQL示例连接字符串格式为:postgresql+psycopg2://【用户名】:【密码】@【主机】:【端口】/【数据库名】。若密码含特殊字符(如@、/),必须先进行URL编码,否则连接会静默失败。

测试连接是否可用:在Python中运行 engine = create_engine(conn_str); engine.connect()。连接成功后才进入下一步。

安装并配置SQLAgent依赖

执行 pip install langchain-community langchain-openai sqlalchemy psycopg2-binary(PostgreSQL)或 mysql-connector-python(MySQL)。

导入必要模块:from langchain.agents import create_sql_agent;from langchain.sql_database import SQLDatabase;from langchain_openai import ChatOpenAI。

注意:SQLAgent已从langchain旧版迁移至langchain-community,直接pip install langchain会导致缺失sql_toolkit模块,必须安装langchain-community。

初始化SQLDatabase与LLM实例

第一步:用SQLAlchemy连接字符串初始化SQLDatabase对象,传入include_tables参数限定可访问的表(避免Agent扫描全部系统表拖慢响应)。

Grok
Grok

Grok是由埃隆·马斯克旗下xAI公司开发的AI助手,2023年11月推出。其最大特色是与X平台深度整合,可实时获取最新信息,并以幽默、直率的对话风格区别于其他AI。功能涵盖文本问答、图像生成、文档分析及语音交互,最新版本已具备多模态识别与深度推理能力。

下载

第二步:初始化ChatOpenAI模型,temperature设为0以保证SQL生成稳定性;model_name建议选gpt-4-turbo或gpt-3.5-turbo-16k,避免使用gpt-4o-mini——它对SQL关键词识别准确率明显下降。

第三步:调用create_sql_agent函数,传入llm、db、agent_type="openai-tools",并设置verbose=True便于调试。这一步会自动加载SQLQueryTool、SchemaRetrieverTool等内置工具。

运行自然语言查询

方法一:直接调用agent.invoke(),传入字符串问题,例如 agent.invoke("统计每个部门的员工人数")。

方法二:使用streaming输出流式响应,适合Web界面:for chunk in agent.stream({"input": "过去7天订单量Top5的用户ID"}): print(chunk.get("output", ""), end="", flush=True)。

首次运行时,Agent会先检索表结构生成schema摘要,耗时取决于表数量。若提示“no such table”,检查include_tables是否拼写错误或大小写不匹配——某些数据库(如PostgreSQL)默认区分大小写。

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

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

下载

相关标签:

grok

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

相关专题

更多
Grok使用教程大全
Grok使用教程大全

从注册登录到高级提示词应用,全面讲解Grok使用教程、常见问题解决方案和效率提升技巧。

2026.06.12

562

11

Grok工具推荐合集
Grok工具推荐合集

本专题整合了Grok工具合集,阅读专题下面的文章了解更多详细内容。

2026.06.12

482

9

Grok AI模型选择推荐
Grok AI模型选择推荐

本专题整合了Grok模型推荐指南,阅读专题下面的文章了解更多详细内容。

2026.06.12

682

19

grok搭建部署教程合集
grok搭建部署教程合集

本专题整合了grok搭建部署教程合集,阅读专题下面的文章了解更多详细内容。

2026.06.12

678

13

OpenJDK JVM参数怎么调优
OpenJDK JVM参数怎么调优

Java应用运行缓慢,如何调整JVM参数?本专题介绍OpenJDK JVM性能调优的基本流程,涵盖性能指标确认、运行数据采集、内存配置和垃圾回收分析,帮助你先定位瓶颈,再通过对照测试验证效果,避免直接套用参数而影响应用稳定性。

2026.10.16

100

10

OpenJDK怎么配置Java开发环境
OpenJDK怎么配置Java开发环境

怎样用OpenJDK搭建开发环境?本专题介绍版本选择、下载安装、环境变量配置和编译运行验证,帮助你理清JDK的作用及基本配置流程,并通过简单程序检查环境是否可用,为后续项目开发做好准备。

2026.10.16

60

10

OpenJDK怎么部署Java应用
OpenJDK怎么部署Java应用

Java项目开发完成后,怎样部署到目标环境并正常运行?本专题介绍OpenJDK应用部署流程,涵盖运行环境准备、版本兼容检查、应用打包、启动参数配置和日志查看,帮助你理清部署前后的操作顺序,减少环境差异造成的运行问题。

2026.10.16

40

10

Revel框架模板渲染与表单验证
Revel框架模板渲染与表单验证

本专题围绕Revel框架视图层开发,介绍模板语法、变量输出、布局复用、表单提交、参数绑定与验证规则,帮助开发者完成安全规范的页面交互。

2026.10.16

40

15

Revel框架MVC路由与控制器
Revel框架MVC路由与控制器

本专题聚焦Revel框架MVC核心机制,讲解路由定义、控制器Action、参数绑定、请求响应和反向路由,并结合常见场景梳理接口请求处理流程。

2026.10.16

20

15

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
Grok官方手册
Grok官方手册

共0课时 | 0人学习