构建一个基于SQL数据的问答系统：从零开始的全攻略

qq_37836323

于 2024-10-09 13:38:09 发布

阅读量173

点赞数 2

文章标签： sql 数据库 python

本文链接：https://blog.csdn.net/qq_29929123/article/details/142784652

版权

构建一个基于SQL数据的问答系统：从零开始的全攻略

在人工智能和编程领域，构建能够从结构化数据中得出结论的系统是一个重要的步骤。本教程将指导你如何利用语言模型(LLM)来查询SQL数据库，实现自然语言问答系统。

引言

为了从数据库中获取信息并以自然语言回答问题，我们需要使用语言模型生成SQL查询。这篇文章旨在帮助你构建一个简单的问答系统，通过SQL查询来获取数据库中的信息。

主要内容

系统架构

系统主要步骤包括：

将问题转换为SQL查询。
执行SQL查询。
使用查询结果回答问题。

设置

首先，安装所需的包并设置环境变量：

%%capture --no-stderr
%pip install --upgrade --quiet langchain langchain-community langchain-openai faiss-cpu

import getpass
import os

if not os.environ.get("OPENAI_API_KEY"):
    os.environ["OPENAI_API_KEY"] = getpass.getpass()

if not os.environ.get("LANGCHAIN_API_KEY"):
    os.environ["LANGCHAIN_API_KEY"] = getpass.getpass()
    os.environ["LANGCHAIN_TRACING_V2"] = "true"

确保数据库连接正常并可用：

from langchain_community.utilities import SQLDatabase

# 使用API代理服务提高访问稳定性
db = SQLDatabase.from_uri("sqlite:///Chinook.db")

代码示例

创建SQL查询链

from langchain.chains import create_sql_query_chain
from langchain_openai import ChatOpenAI

llm = ChatOpenAI(model="gpt-4o-mini")
chain = create_sql_query_chain(llm, db)
response = chain.invoke({"question": "How many employees are there"})
print(response)

执行SQL查询并回答问题

from langchain_community.tools.sql_database.tool import QuerySQLDataBaseTool

execute_query = QuerySQLDataBaseTool(db=db)
write_query = create_sql_query_chain(llm, db)
chain = write_query | execute_query
result = chain.invoke({"question": "How many employees are there"})
print(result)