利用Dataherald API在LangChain中实现自然语言到SQL的转换

用户57716616653
64 阅读2分钟

引言

在现代数据驱动的世界中，自然语言到SQL的转换工具受到越来越多的关注。这些工具可以帮助用户轻松查询数据库，而无需深入了解SQL语法。本文将探讨Dataherald API在LangChain中的使用，展示如何将自然语言转换为SQL查询。

主要内容

安装与设置

要使用Dataherald API，首先需要进行安装和设置：

  1. 使用以下命令安装Dataherald的Python包：

    pip install dataherald

  2. 在Dataherald官网注册并获取API KEY。

  3. 将API KEY设置为环境变量：

    export DATAHERALD_API_KEY='your_api_key'

API包装器

LangChain提供了一个DataheraldAPIWrapper工具，可简化API的集成：

  • 导入方法： 
    from langchain_community.utilities.dataherald import DataheraldAPIWrapper

工具的使用

可以在LangChain代理中使用Dataherald工具，如下所示：

from langchain_community.utilities.dataherald import DataheraldAPIWrapper
from langchain_community.tools.dataherald.tool import DataheraldTextToSQL
from langchain_openai import ChatOpenAI
from langchain import hub
from langchain.agents import AgentExecutor, create_react_agent

api_wrapper = DataheraldAPIWrapper(db_connection_id="<db_connection_id>")
tool = DataheraldTextToSQL(api_wrapper=api_wrapper)
llm = ChatOpenAI(model="gpt-3.5-turbo", temperature=0)
prompt = hub.pull("hwchase17/react")
agent = create_react_agent(llm, [tool], prompt)
agent_executor = AgentExecutor(agent=agent, tools=[tool], verbose=True)

# 使用API代理服务提高访问稳定性
agent_executor.invoke({"input":"Return the sql for this question: How many employees are in the company?"})

输出示例

运行以上代码，将输出对应的SQL查询：

> Entering new AgentExecutor chain...
I need to use a tool that can convert this question into SQL.
Action: dataherald
Action Input: How many employees are in the company?

Final Answer: 
SELECT COUNT(*) FROM employees

> Finished chain.

常见问题和解决方案

  1. API访问失败： 由于网络限制，API访问可能会失败，建议使用API代理服务确保稳定连接。

  2. 环境变量未设置： 确保API KEY已正确设置为环境变量，否则API调用将失败。

总结和进一步学习资源

Dataherald API为自然语言到SQL的转换提供了一种简单而有效的方法。通过LangChain的工具支持，用户可以无缝集成该功能来增强应用程序的交互性。

进一步学习资源：

参考资料

  1. Dataherald官方文档
  2. LangChain官方文档

如果这篇文章对你有帮助，欢迎点赞并关注我的博客。您的支持是我持续创作的动力！

---END---

avatar
文章
阅读
粉丝