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
EXECUTEcontrola 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
-
Um serviço de provedor de modelos que você pode query. As seguintes permissões são necessárias. Consulte Descobrir e controlar o acesso a provedores de modelos externos (serviços de provedor de modelos).
EXECUTEno serviço de provedor de modeloUSE CATALOGno catálogoUSE SCHEMAno esquema
-
A CLI do Unity Gateway. Requer Python 3.12 ou posterior e
uv.
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
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:
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.
--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:
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/anthropice carrega o cabeçalho emANTHROPIC_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/v1comwire_api = "responses"e carrega o cabeçalho nohttp_headersdo provedor.
O modelo que o agente envia deve ser um que o serviço de provedor de modelos permita.