Databricks 集成

概览

通过 Databricks 托管的 MCP 服务器，将你的 CrewAI 代理直接连接到 Databricks 工作区。Databricks 集成让你的代理可以通过 Genie 提出自然语言问题、运行受治理的 SQL、调用 Unity Catalog Functions，以及使用 Vector Search 检索文档，而且无需编写或托管任何连接器代码，所有调用都由 Unity Catalog 权限控制。

在底层，Databricks 集成是 CrewAI 自定义 MCP 服务器 支持的托管封装。Databricks 会把每项能力作为独立的 Model Context Protocol 端点公开，而 CrewAI 会代你安全地连接到这些端点。由于每个服务器是分别添加的，你可以精确启用你的团队所需的能力。

核心能力

Genie

通过 Genie 用自然语言提问，并从你的数据中获得有依据的答案。它会查询 Genie Spaces 和 Unity Catalog，并回链到 Databricks 界面。

Databricks SQL

在你的 Databricks 仓库上运行受治理的 SQL，直接通过代理查询、转换并编写数据管道。

Unity Catalog Functions

调用 Unity Catalog functions 来运行预定义的 SQL 和自定义业务逻辑，作为受治理、可复用的工具。

Vector Search

使用语义相似度从 Mosaic AI Vector Search 索引中检索相关文档，用于 RAG 和知识工作流。

每个服务器都运行在 Unity AI Gateway 之后，并强制执行 Unity Catalog 访问控制，因此你的代理只能看到它们被允许使用的数据和工具。

前提条件

在使用 Databricks 集成前，请确保你已经具备：

• 一个拥有有效订阅的 CrewAI AMP 账号
• 一个 Databricks 工作区，并启用你希望公开的能力（Genie Spaces、SQL 仓库、Unity Catalog functions 或 Vector Search 索引）
• 底层对象上具备相应的 Unity Catalog 权限
• 你的 Databricks 工作区主机名（例如 your-workspace.cloud.databricks.com）

Databricks 托管的 MCP 服务器

Databricks 会为每项能力发布一个单独的托管 MCP 服务器。CrewAI 将这些服务器作为独立连接公开，每个连接都使用你的工作区主机和相应的 Unity Catalog 标识符进行配置。端点遵循以下模式：

Server	作用	MCP URL pattern
Genie	基于 Genie Space 的自然语言问答	https://<workspace-hostname>/api/2.0/mcp/genie/{genie_space_id}
Databricks SQL	在你的仓库上执行 SQL	https://<workspace-hostname>/api/2.0/mcp/sql
Unity Catalog Functions	运行已注册的 UC 函数	https://<workspace-hostname>/api/2.0/mcp/functions/{catalog}/{schema}
Vector Search	查询 Vector Search 索引	https://<workspace-hostname>/api/2.0/mcp/vector-search/{catalog}/{schema}

Note

你不需要手动拼接这些 URL。CrewAI 会根据你在配置连接时提供的工作区主机和标识符（Genie Space ID，或 catalog/schema）生成每个端点。有关完整规范和最新端点详情，请参阅 Databricks 托管 MCP 文档。

在 CrewAI AMP 中连接 Databricks

每项 Databricks 能力 - Databricks Genie、Databricks SQL、Databricks Unity Catalog Functions 和 Databricks Vector Search - 都会作为自己的 MCP 服务器出现在 Tools & Integrations 页面上的 Databricks 分组下。按需配置这些服务器：

1. 打开 Tools & Integrations

   在 CrewAI AMP 左侧边栏中打开 Tools & Integrations，并在 Connections 列表中找到 Databricks 分组。你会在其下看到 Genie、SQL、Unity Catalog Functions 和 Vector Search 服务器。

2. 配置服务器

   点击你想启用的能力旁边的 Configure，并提供连接详情：

   • Workspace Host - 你的 Databricks 工作区主机名（例如 my-workspace.cloud.databricks.com）。
   • Genie - 要查询的 Genie Space ID。
   • Unity Catalog Functions - 包含这些函数的 catalog 和 schema。
   • Vector Search - 包含索引的 catalog 和 schema。
   • Databricks SQL - 不需要额外标识符；查询会在你工作区的 SQL 仓库上运行。
3. 选择身份验证方式

   选择 CrewAI 连接到 Databricks 的身份验证方式。推荐使用 OAuth。

   • Use OAuth - 通过 OAuth 2.0 安全连接。每个用户单独进行身份验证，Databricks 会发放作用域限定到该能力（genie、sql、unity-catalog 或 vector-search）的令牌。CrewAI 会处理授权流程并自动刷新令牌。
   • Use personal access token - 使用 Databricks personal access token 进行身份验证。使用最小权限身份以限制暴露面。
4. 完成身份验证

   完成身份验证后，该服务器的工具就会对你的团队可用。若要启用其他 Databricks 能力，请重复以上步骤。

Tip

由于每项能力都是独立连接，你可以灵活组合。例如，为研究团队启用 Genie 和 Vector Search，同时将 SQL 和 Unity Catalog Functions 保留给数据工程团队。可见性设置可以控制哪些团队成员能使用每个连接。

在你的团队中使用 Databricks 工具

连接完成后，每个 MCP 服务器公开的工具会与内置连接一起出现在 Tools & Integrations 页面上。你可以：

• 像使用其他 CrewAI 工具一样，将工具分配给代理。
• 管理可见性，控制哪些团队成员可以使用每个连接。
• 随时从 Connections 列表中编辑或移除任意连接。

现在你的代理可以向 Genie 提出有依据的答案、在仓库上运行 SQL、调用 Unity Catalog 函数，并搜索 Vector Search 索引，结果会自动回流到它们的推理过程。

Warning

Databricks 通过 Unity Catalog 和 Unity AI Gateway 实施治理：用户只能发现并调用其工作区身份被允许使用的工具。如果某次工具调用失败，请确认连接用户（或令牌身份）在 Genie Space、仓库、函数或索引上拥有所需的 Unity Catalog 权限。部分 Genie 和 SQL 查询以异步方式运行，可能需要一点时间才能返回结果。

了解更多

Databricks 托管的 MCP 服务器

有关托管版 Genie、SQL、Unity Catalog Functions 和 Vector Search MCP 服务器的官方 Databricks 文档。

CrewAI 中的自定义 MCP 服务器

了解 CrewAI 如何连接到任意 MCP 服务器，这正是 Databricks 集成的基础。

需要帮助？

如需 Databricks 集成方面的帮助或故障排查，请联系我们的支持团队。