Pular para o conteúdo principal
Última atualização em

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

info

Beta

Este recurso está em Beta. Administradores de conta podem gerenciar o acesso a este recurso a partir da página Previews do console da conta. Consulte Gerenciar prévias do Databricks.

Esta página descreve como rotear agentes de codificação como Claude Code e Codex CLI através de um serviço de provedor de modelo. O agente funciona com um provedor externo que você registrou no Unity Catalog, como sua própria conta OpenAI, Anthropic ou Amazon Bedrock. O Unity AI 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 AI Gateway do Unity 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 modelos com ucode​

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

Passo 1: Instalar o ucode​

Bash
uv tool install git+https://github.com/databricks/ucode

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
ucode claude --provider main.default.openai_prod    # Claude Code
ucode codex --provider main.default.openai_prod     # Codex CLI

Na primeira abertura, ucode solicita o URL do seu workspace 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. ucode anexa o cabeçalho Databricks-Model-Provider-Service a cada solicitação. O Unity AI Gateway resolve o serviço, fornece as credenciais armazenadas e encaminha a solicitação ao provedor externo.

nota

--provider é compatível com ucode claude e ucode codex. Se o recurso de serviço do provedor de modelo não estiver ativado em seu workspace, ou se você não tiver EXECUTE no serviço, ucode imprimirá 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
ucode configure

Para Claude Code e CLI do Codex, ucode 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 ucode --help.

Como o roteamento funciona​

Quando você encaminha um agente através de um serviço de provedor de modelo, ucode define a URL base do agente para seu endpoint do Unity AI Gateway e adiciona o cabeçalho Databricks-Model-Provider-Service que nomeia o serviço:

  • Claude Code autentica-se contra https://<workspace-url>/ai-gateway/anthropic e transporta o cabeçalho em ANTHROPIC_CUSTOM_HEADERS. Para um serviço apoiado por Amazon Bedrock, ucode fixa os 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 modelo permite. Consulte Crie e gerencie serviços de provedor de modelo para allow_all_targets e targets.

Próximos os passos​

Nesta página