Aller au contenu principal
Dernière mise à jour le

Intégrez des agents de codage avec des services de fournisseur de modèle

Acheminez les agents de codage tels que Claude Code et Codex CLI via un service de fournisseur de modèles pour les exécuter sur un fournisseur externe que vous avez enregistré dans Unity Catalog, comme votre propre compte OpenAI, Anthropic ou Amazon Bedrock. Unity Gateway fournit les identifiants stockés au moment de la requête et applique la gouvernance ainsi que le suivi de l’utilisation, de sorte que l’agent de codage ne manipule pas le secret du fournisseur.

Pour acheminer un agent vers des modèles fournis par Databricks, consultez Intégrer avec des agents de codage.

Fonctionnalités​

  • Utilisez votre propre fournisseur. Pointez Claude Code ou Codex CLI vers votre propre OpenAI, Anthropic, Amazon Bedrock ou tout autre fournisseur enregistré via un seul élément sécurisable Unity Catalog gouverné.
  • Gouvernance centralisée. Le privilège EXECUTE permet d'accéder au service de fournisseur de modèles, et vous pouvez appliquer des limites de débit et des politiques de service au niveau du service.
  • Observabilité. Unity Gateway capture les requêtes dans le suivi de l’utilisation et, lorsqu’il est configuré, dans les tables d’inférence.

Exigences​

Acheminer un agent de codage vers un service de fournisseur de modèles avec la CLI Unity Gateway​

La CLI Unity Gateway installe, authentifie et configure les agents de codage pris en charge pour acheminer via Unity Gateway. Pour acheminer un agent via un service de fournisseur de modèle au lieu des modèles fournis par Databricks, transmettez le nom en trois parties du service à --provider.

Étape 1 : Installez la CLI Unity Gateway​

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

Étape 2 : Ouvrez un agent sur un service de fournisseur de modèles​

Transmettez le nom entièrement qualifié du service de fournisseur de modèle (<catalog>.<schema>.<name>) à --provider:

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

À la première ouverture, la CLI Unity Gateway vous demande l’URL de votre workspace Databricks, s’authentifie et écrit le fichier de configuration de l’agent afin que les requêtes soient acheminées via le service de fournisseur de modèle. Elle joint l’en-tête Databricks-Model-Provider-Service à chaque requête. Unity Gateway résout le service, fournit les identifiants stockés et transmet la requête au fournisseur externe.

remarque

--provider est pris en charge pour ug claude et ug codex. Si la fonctionnalité de service de fournisseur de modèle n'est pas activée sur votre Workspace, ou si vous manquez de EXECUTE sur le service, la CLI Unity Gateway affiche une erreur descriptive.

Étape 3 (facultatif) : Configurer les agents de manière interactive​

Pour sélectionner un service de fournisseur de modèle via un flux guidé au lieu de passer --provider, exécutez :

Bash
ug configure

Pour Claude Code et Codex CLI, ug configure liste les services de fournisseur de modèle qui vous sont accessibles et vous permet d’en sélectionner un. Choisir « aucune » achemine l’agent via les modèles fournis par Databricks.

Pour la référence complète de la commande, exécutez ug --help.

Fonctionnement du routage​

Lorsque vous acheminez un agent via un service de fournisseur de modèles, la CLI Unity Gateway définit l'URL de base de l'agent sur votre endpoint Unity Gateway et ajoute l'en-tête Databricks-Model-Provider-Service nommant le service :

  • Claude Code s’authentifie auprès de https://<workspace-url>/ai-gateway/anthropic et transporte l’en-tête dans ANTHROPIC_CUSTOM_HEADERS. Pour un service pris en charge par Amazon Bedrock, l’interface CLI Unity Gateway pins les ID de modèle côté fournisseur (par exemple, us.anthropic.claude-sonnet-4-6).
  • Codex CLI s'authentifie auprès de https://<workspace-url>/ai-gateway/codex/v1 avec wire_api = "responses" et transmet l'en-tête dans le http_headers du fournisseur.

Le modèle que l'agent envoie doit être un de ceux que le service de fournisseur de modèle autorise.

Étapes suivantes​

Sur cette page