使用SQLDatabase Toolkit构建智能数据库问答系统

bhawfgrcbtwny

于 2024-10-08 10:46:24 发布

阅读量243

点赞数 3

文章标签：数据库 python

本文链接：https://blog.csdn.net/bhawfgrcbtwny/article/details/142753879

版权

# 使用SQLDatabase Toolkit构建智能数据库问答系统

## 引言

在当今的数据驱动世界中，通过自然语言查询数据库已成为一种趋势。本文将介绍如何使用SQLDatabase Toolkit，将AI代理与SQL数据库结合，构建一个智能数据库问答系统。

## 主要内容

### 为什么选择SQLDatabase Toolkit

SQLDatabase Toolkit为开发者提供了一种与SQL数据库进行智能交互的方式。它允许创建能够迭代修复错误的问答代理，从而提高了系统的健壮性。

### 安全注意事项

构建这样的系统需特别注意安全，因为模型会生成并执行SQL查询。确保数据库连接权限最小化，以减少潜在风险。

### 安装与设置

要使用SQLDatabase Toolkit，请确保安装`langchain-community`软件包：

```bash
%pip install --upgrade --quiet langchain-community

对于完整的功能启用，还需要安装一些额外的包，如langchainhub和langgraph。

实例化工具包

要实例化SQLDatabaseToolkit，需要一个SQLDatabase对象和一个LLM（大语言模型）。以下是如何创建一个基于Chinook示例数据库的数据库对象：

import sqlite3
import requests
from langchain_community.utilities.sql_database import SQLDatabase
from sqlalchemy import create_engine
from sqlalchemy.pool import StaticPool

def get_engine_for_chinook_db():
    url = "https://raw.githubusercontent.com/lerocha/chinook-database/master/ChinookDatabase/DataSources/Chinook_Sqlite.sql"
    response = requests.get(url)
    sql_script = response.text

    connection = sqlite3.connect(":memory:", check_same_thread=False)
    connection.executescript(sql_script)
    return create_engine(
        "sqlite://",
        creator=lambda: connection,
        poolclass=StaticPool,
        connect_args={"check_same_thread": False},
    )

engine = get_engine_for_chinook_db()
db = SQLDatabase(engine)

与代理结合

可以通过以下方式将SQL工具包与问答代理结合：

from langchain_community.agent_toolkits.sql.toolkit import SQLDatabaseToolkit

toolkit = SQLDatabaseToolkit(db=db, llm=llm)  # 使用API代理服务提高访问稳定性

from langgraph.prebuilt import create_react_agent

agent_executor = create_react_agent(
    llm, toolkit.get_tools(), state_modifier=system_message
)

example_query = "Which country's customers spent the most?"
events = agent_executor.stream(
    {"messages": [("user", example_query)]},
    stream_mode="values",
)
for event in events:
    event["messages"][-1].pretty_print()

代码示例

下面是一个简单的代码示例，展示如何使用SQLDatabase Toolkit来回答客户消费问题：

example_query = "Which country's customers spent the most?"

events = agent_executor.stream(
    {"messages": [("user", example_query)]},
    stream_mode="values",
)
for event in events:
    event["messages"][-1].pretty_print()