Dataherald
Dataherald 是一个自然语言到SQL的工具。
本页面介绍了如何在LangChain中使用Dataherald API
。
安装与设置
- 使用以下命令安装所需软件包
pip install dataherald
- 前往dataherald并注册 这里
- 创建一个应用程序并获取您的
API KEY
- 将您的
API KEY
设置为环境变量DATAHERALD_API_KEY
包装器
实用工具
存在一个DataheraldAPIWrapper工具,它封装了这个API。要导入这个工具:
from langchain_community.utilities.dataherald import DataheraldAPIWrapper
API Reference:DataheraldAPIWrapper
有关此包装器的更详细演练,请参见此笔记本。
工具
你可以像这样在代理中使用该工具:
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, load_tools
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, tools, prompt)
agent_executor = AgentExecutor(agent=agent, tools=tools, verbose=True, handle_parsing_errors=True)
agent_executor.invoke({"input":"Return the sql for this question: How many employees are in the company?"})
API Reference:DataheraldAPIWrapper | DataheraldTextToSQL | ChatOpenAI | hub | AgentExecutor | create_react_agent | load_tools
输出
> 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?Answer: SELECT
COUNT(*) FROM employeesI now know the final answer
Final Answer: SELECT
COUNT(*)
FROM
employees
> Finished chain.
{'input': 'Return the sql for this question: How many employees are in the company?', 'output': "SELECT \n COUNT(*)\nFROM \n employees"}
有关工具的更多信息,请参见此页面。