Pular para o conteúdo principal

Integre agentes de codificação com serviços de provedor de modelo

Encaminhe agentes de codificação como o Claude Code e a CLI do Codex por meio de um serviço de provedor de modelos para a execução deles em um provedor externo que você registrou no Unity Catalog, como sua própria account OpenAI, Anthropic ou Amazon Bedrock. O Unity Gateway fornece as credenciais armazenadas no momento da solicitação e aplica governança e acompanhamento de uso, para que o agente de codificação não manipule o segredo do provedor.

Para rotear um agente para modelos fornecidos pela Databricks, em vez disso, consulte Integrar com agentes de codificação.

recurso

  • Traga seu próprio provedor. Aponte o Claude Code ou o Codex CLI para seu próprio provedor registrado do OpenAI, Anthropic, Amazon Bedrock ou outro por meio de um único recurso protegível do Unity Catalog regido.
  • Governança centralizada. O privilégio EXECUTE controla o acesso ao serviço de provedor de modelos, e é possível aplicar limites de taxa e políticas de serviço no nível de serviço.
  • Observabilidade. O Unity Gateway captura solicitações no acompanhamento de uso e, quando configurado, em tabelas de inferência.

Requisitos

Encaminhe um agente de codificação para um serviço de provedor de modelo com a CLI do Unity Gateway

A CLI do Unity Gateway instala, autentica e configura agentes de codificação compatíveis para rotear através do Unity Gateway. Para rotear um agente através de um serviço de provedor de modelos em vez de modelos fornecidos pelo Databricks, passe o nome de três partes do serviço para --provider.

Etapa 1: Instale a CLI do Unity Gateway

Bash
uv tool install git+https://github.com/databricks/unity-gateway

O passo 2: Abra um agente para um serviço de provedor de modelos

Passe o nome totalmente qualificado do serviço de provedor de modelo (<catalog>.<schema>.<name>) para --provider:

Bash
ug claude --provider main.default.openai_prod    # Claude Code
ug codex --provider main.default.openai_prod # Codex CLI

Na primeira abertura, a CLI do Unity Gateway solicita o URL do seu workspace do Databricks, autentica e grava o arquivo de configuração do agente para que as solicitações sejam roteadas pelo serviço de provedor de modelo. Ele anexa o cabeçalho Databricks-Model-Provider-Service a cada solicitação. O Unity Gateway resolve o serviço, fornece as credenciais armazenadas e encaminha a solicitação para o provedor externo.

nota

--provider é compatível com ug claude e ug codex. Se o recurso de serviço de provedor de modelo não estiver ativado em seu workspace, ou se você não tiver EXECUTE no serviço, a CLI do Unity Gateway exibirá um erro descritivo.

Etapa 3 (opcional): Configure agentes interativamente

Para selecionar um serviço de provedor de modelos por meio de um fluxo guiado em vez de passar --provider, execute:

Bash
ug configure

Para Claude Code e CLI do Codex, ug configure lista os serviços de provedor de modelo disponíveis para você e permite que você selecione um. Não escolher nenhum encaminha o agente pelos modelos fornecidos pelo Databricks.

Para a referência completa do comando, execute ug --help.

Como o roteamento funciona

Quando você roteia um agente por meio de um serviço de provedor de modelo, a CLI do Unity Gateway define a URL base do agente para o seu endpoint do Unity Gateway e adiciona o cabeçalho Databricks-Model-Provider-Service nomeando o serviço:

  • O Claude Code autentica-se no https://<workspace-url>/ai-gateway/anthropic e carrega o cabeçalho em ANTHROPIC_CUSTOM_HEADERS. Para um serviço com suporte da Amazon Bedrock, a CLI do Unity Gateway faz o pin dos IDs de modelo do lado do provedor (por exemplo, us.anthropic.claude-sonnet-4-6).
  • Codex CLI autentica contra https://<workspace-url>/ai-gateway/codex/v1 com wire_api = "responses" e carrega o cabeçalho no http_headers do provedor.

O modelo que o agente envia deve ser um que o serviço de provedor de modelos permita.

Próximos os passos