Databricks SQL 查询工具

DatabricksQueryTool

描述

可通过 CLI profile 或直接的 host/token 认证，对 Databricks 工作区表运行 SQL。

安装

uv add crewai-tools[databricks-sdk]

环境变量

• DATABRICKS_CONFIG_PROFILE 或 (DATABRICKS_HOST + DATABRICKS_TOKEN)

在 Databricks 工作区的 User Settings → Developer 中创建个人访问令牌并查找 host 信息。
文档：https://docs.databricks.com/en/dev-tools/auth/pat.html

示例

from crewai import Agent, Task, Crew
from crewai_tools import DatabricksQueryTool

tool = DatabricksQueryTool(
    default_catalog="main",
    default_schema="default",
)

agent = Agent(
    role="数据分析师",
    goal="查询 Databricks",
    tools=[tool],
    verbose=True,
)

task = Task(
    description="SELECT * FROM my_table LIMIT 10",
    expected_output="10 rows",
    agent=agent,
)

crew = Crew(
    agents=[agent],
    tasks=[task],
    verbose=True,
)
result = crew.kickoff()

print(result)

参数

• query（必需）：要执行的 SQL 查询
• catalog（可选）：覆盖默认 catalog
• db_schema（可选）：覆盖默认 schema
• warehouse_id（可选）：覆盖默认 SQL warehouse
• row_limit（可选）：返回的最大行数（默认：1000）

初始化默认值

• default_catalog
• default_schema
• default_warehouse_id

错误处理与提示

• 认证错误：请验证 DATABRICKS_HOST 是否以 https:// 开头，并确认 token 有效。
• 权限：确保你的 SQL warehouse 和 schema 可被 token 访问。
• 限制：避免在 agent 循环中运行长时间查询；请添加过滤条件或 LIMIT。