用途:为 Databricks 上的智能体添加工具并授予权限。适用于向 agent 添加 MCP 服务器、Genie 空间、向量搜索或 Unity Catalog 函数等资源,并在 databricks.yml 中配置访问权限。
主要能力:在 agent_server/agent.py 中通过 DatabricksMCPServer 与 DatabricksMultiServerMCPClient 注册 MCP 服务器并获取工具;在 databricks.yml 的 resources 下声明资源及权限(如 genie_space 的 CAN_RUN、app 的 CAN_USE);支持 UC 函数、UC 连接、向量搜索索引、SQL 仓库、模型服务端点、Genie 空间、Lakebase、MLflow 实验、Databricks App 等资源类型;支持用 value_from 在 config.env 中引用已声明资源;支持 handle_tool_error 与 timeout 处理 MCP 调用错误。
使用方式:所有 databricks CLI 命令需带 .env 中的 profile,即 databricks <command> --profile <profile>。先在 agent.py 添加 MCP 服务器,再在 databricks.yml 授予权限,最后执行 databricks bundle deploy 与 databricks bundle run agent_langgraph。自定义 MCP 服务器需 Databricks CLI v0.298.0+。
限制:若资源尚不存在需先参考 create-tools 技能;添加 MCP 服务器后必须在 databricks.yml 授权,否则会报权限错误;config.env 中每个 value_from 必须匹配 resources 中的 name;各资源权限取值不同;MLflow 实验已在模板中配置无需操作。