mcpField
D
databricks-unity-gateway

databricks-unity-gateway

github.com/databricks其他37341 次安装1 天前更新
代码仓库

关于 databricks-unity-gateway

databricks-unity-gateway 是通过 Databricks CLI 管理 Unity Gateway 服务的技能,涵盖模型、MCP 和模型提供方服务,并指导生命周期操作、访问控制与查询。

主要功能

  1. 01管理模型、MCP 和模型提供方服务
  2. 02创建、查看、更新和删除服务
  3. 03使用 CLI 配置与身份验证配置文件
  4. 04管理访问权限和支出控制
  5. 05查询模型和模型提供方服务

适用场景

  1. 01注册或更新 MCP 服务
  2. 02管理 Unity Gateway 模型服务资源
  3. 03查看或更改服务访问权限
  4. 04查询模型或模型提供方服务

MCP 相关

该技能通过 `databricks ai-gateway` 管理 Unity Gateway MCP 服务,包括创建、获取、列出、更新和删除;也提供注册外部 MCP 服务的指引。

安装

复制下面的命令,在终端中运行。

$npx skills add https://github.com/databricks/databricks-agent-skills --skill databricks-unity-gateway

安装后在 Claude Code 中输入/databricks-unity-gateway确认已加载。

包含的文件

除 SKILL.md 外,安装时一起下载的参考资料和脚本。

agents/
openai.yaml
assets/
databricks.svg
references/
account-access.md
budgets.md
mcp-services.md
model-provider-services.md
model-services.md
model-serving-migration.md
permissions.md
querying.md

详细介绍

Unity Gateway

使用父级 databricks-core 技能处理 CLI 身份验证和配置文件选择。

Unity Gateway 提供 Unity Catalog 可保护对象,用于受管控地访问 AI 服务。CLI 命令组保留了早期产品名称:databricks ai-gateway --help。

选择正确的 Gateway

请求使用
Unity Gateway 模型服务、MCP 服务或模型提供商服务此技能和 databricks ai-gateway
Model Serving 端点上的 AI Gateway 配置,包括 put-ai-gatewaydatabricks-model-serving 和 databricks serving-endpoints
Lakeflow Connect 摄取网关databricks-lakeflow-connect

当资源是 Unity Catalog 可保护对象时,将 Unity AI Gateway 视为 Unity Gateway 的早期名称。除非用户明确要求迁移,否则不要将旧版 Model Serving AI Gateway 资源转换为 Unity Gateway 资源。

CLI 工作流

使用 Databricks CLI 管理资源生命周期和权限。不要用直接 REST 调用或 Databricks SDK 替代这些操作。支持使用 Python 客户端查询模型服务和模型提供商服务;请阅读查询服务。

明确要求使用 Unity Gateway、Unity AI Gateway 或 databricks ai-gateway 的请求属于此技能。不要仅因为请求涉及模型服务或工作区,就切换到 databricks-model-serving。只有在处理旧版的单端点 AI Gateway 操作,或预配吞吐量目标需要 Model Serving 端点详情时,才使用该技能。

  1. 确认 CLI 满足最低版本要求:

    databricks --version
    
  2. 在构造负载之前,检查命令和相关操作:

    databricks ai-gateway --help
    databricks ai-gateway <operation> --help
    
  3. 通过 --json 传递创建和更新配置。需要基于配置文件进行身份验证时,请显式指定配置文件:

    databricks ai-gateway <operation> --json @payload.json --profile <PROFILE>
    

不要根据名称相似的旧版 API 臆造 JSON 字段。请根据已安装 CLI 的帮助信息和所选服务类型对应的 Unity Gateway 文档构建负载。

服务类型

服务类型CLI 操作
模型服务create-model-service、get-model-service、list-model-services、update-model-service、delete-model-service;请阅读模型服务
MCP 服务create-mcp-service、get-mcp-service、list-mcp-services、update-mcp-service、delete-mcp-service;请阅读MCP 服务
模型提供商服务create-model-provider-service、get-model-provider-service、list-model-provider-services、update-model-provider-service、delete-model-provider-service;请阅读模型提供商服务

查询和权限

对于这些参考资料已涵盖的字段,不要查阅公开文档。只有在参考资料中缺少必需字段,或用户明确要求查看最新文档时,才查阅权威文档。除非已安装的 CLI 帮助信息或当前文档明确说明,否则不要将功能标记为 Beta 或预览版。

从 Model Serving 迁移到 Unity Gateway

从工作区范围的 Model Serving 端点迁移到 Unity Catalog 范围的 Unity Gateway 服务时,请阅读 references/model-serving-migration.md。

权威文档

评论

支持 Markdown