NL2SQL工具¶

描述¶

此工具用于将自然语言转换为SQL查询。当传递给代理时，它将生成查询，然后使用这些查询与数据库进行交互。

这支持多种工作流程，例如让代理访问数据库，根据目标获取信息，然后使用这些信息生成响应、报告或其他任何输出。此外，还提供了代理根据其目标更新数据库的能力。

注意：确保代理可以访问只读副本，或者代理在数据库上运行插入/更新查询是可以接受的。

要求¶

SqlAlchemy
任何兼容的数据库库（例如 psycopg2, mysql-connector-python）

安装¶

安装crewai_tools包

pip install 'crewai[tools]'

使用¶

为了使用NL2SQLTool，您需要将数据库URI传递给工具。URI应采用以下格式：dialect+driver://username:password@host:port/database。

from crewai_tools import NL2SQLTool

# 安装psycopg2以使用此示例与PostgreSQL
nl2sql = NL2SQLTool(db_uri="postgresql://example@localhost:5432/test_db")

@agent
def researcher(self) -> Agent:
    return Agent(
        config=self.agents_config["researcher"],
        allow_delegation=False,
        tools=[nl2sql]
    )