使用Langchain构建简单的数据库Agent

news2025/1/11 23:41:10

这篇文章我们介绍一个使用LangChain实现SQLagent的方法，LangChain直接内置了自己的SQLagent实现-—SQLDatabaseChain。这个方法使用 SQL Alchemy 与数据库交互。感兴趣的可以考虑一下这两个方案是否可以融合，这样保证SQL的准确性从而提升最终结果的准确率。下面将介绍今天主要内容的实现步骤：

实现步骤

首先，我们得有一个数据库。这里我们用的是ClickHouse，一个开源的、面向列的SQL数据库管理系统，特别适合大数据分析。安装起来超级简单，一行命令就搞定。

接下来，我们得让Agent学会和ClickHouse交流。这里就需要用到clickhouse-sqlalchemy这个包了。安装它之后，我们就可以建立数据库连接，然后初始化我们的Toolkit。这个Toolkit就像是我们的Agent的工具箱，里面装满了有用的工具。

现在，让我们来创建一个SQLDatabaseChain的Agent。这个Agent可以直接和数据库交互，回答问题。我们只需要定义一些工具，比如查询表结构、执行SQL查询等。然后，我们就可以快速创建并运行这个Agent了。

环境搭建

今天用到的主要安装包如下，其它可以根据提示安装：

# 安装 clickhouse-sqlalchemy 包来连接 ClickHouse``pip install clickhouse-sqlalchemy langchain

运行代码

代码大概是这样的：

from langchain.agents import AgentExecutor, AgentType``from langchain.sql_database_toolkit import SQLDatabaseToolkit``   ``   ``# 设置数据库连接``uri = 'clickhouse+native://localhost/ecommerce'``db = SQLDatabase.from_uri(uri)``   ``   ``# 初始化工具箱``toolkit = SQLDatabaseToolkit(db=db, llm=OpenAI(temperature=0))``   ``   ``# 创建并运行基于 OpenAI 函数的 Agent``agent_executor = create_sql_agent(`    `llm=ChatOpenAI(temperature=0.1, model='gpt-4-1106-preview'),`    `toolkit=toolkit,`    `verbose=True,`    `agent_type=AgentType.OPENAI_FUNCTIONS``)``# 运行 Agent，回答数据相关问题``agent_executor.run("我们有多少来自上海的活跃客户？")

运行这段代码后，Agent就会根据我们的问题去数据库里找到答案，然后告诉我们。比如，它会告诉我们有多少来自上海的活跃客户。这个Agent的好处在于，它可以直接理解我们的问题，然后去数据库里找到答案。我们不需要自己写SQL查询，也不需要担心查询的语法问题。Agent会帮我们搞定一切。

那么，如何系统的去学习大模型LLM？

我在一线互联网企业工作十余年里，指导过不少同行后辈。帮助很多人得到了学习和成长。

作为一名热心肠的互联网老兵，我意识到有很多经验和知识值得分享给大家，也可以通过我们的能力和经验解答大家在人工智能学习中的很多困惑，所以在工作繁忙的情况下还是坚持各种整理和分享。

但苦于知识传播途径有限，很多互联网行业朋友无法获得正确的资料得到学习提升，故此将并将重要的AI大模型资料包括AI大模型入门学习思维导图、精品AI大模型学习书籍手册、视频教程、实战学习等录播视频免费分享出来。

所有资料 ⚡️ ，朋友们如果有需要全套《LLM大模型入门+进阶学习资源包》，扫码获取~
在这里插入图片描述

篇幅有限，部分资料如下：

👉LLM大模型学习指南+路线汇总👈

💥大模型入门要点，扫盲必看！
在这里插入图片描述
💥既然要系统的学习大模型，那么学习路线是必不可少的，这份路线能帮助你快速梳理知识，形成自己的体系。

👉大模型入门实战训练👈

💥光学理论是没用的，要学会跟着一起做，要动手实操，才能将自己的所学运用到实际当中去，这时候可以搞点实战案例来学习。
在这里插入图片描述

👉国内企业大模型落地应用案例👈

💥《中国大模型落地应用案例集》 收录了52个优秀的大模型落地应用案例，这些案例覆盖了金融、医疗、教育、交通、制造等众多领域，无论是对于大模型技术的研究者，还是对于希望了解大模型技术在实际业务中如何应用的业内人士，都具有很高的参考价值。 （文末领取）
在这里插入图片描述
💥《2024大模型行业应用十大典范案例集》 汇集了文化、医药、IT、钢铁、航空、企业服务等行业在大模型应用领域的典范案例。