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 comme 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, tel que votre propre compte OpenAI, Anthropic ou Amazon Bedrock. Unity AI Gateway fournit les identifiants stockés au moment de la requête et applique la gouvernance ainsi que le suivi de l’utilisation, afin 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 AI Gateway capture les requêtes dans le suivi de l'utilisation et, une fois configurées, les tables d'inférence.

Exigences​

Acheminer un agent de codage vers un service de fournisseur de modèle avec ucode​

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

Étape 1 : Installer ucode​

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

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

À la première ouverture, ucode vous invite à saisir 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. ucode joint l’en-tête Databricks-Model-Provider-Service à chaque requête. Unity AI 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 ucode claude et ucode codex. Si la fonctionnalité du service de fournisseur de modèle n'est pas activée sur votre workspace, ou si vous manquez EXECUTE sur le service, ucode 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
ucode configure

Pour Claude Code et Codex CLI, ucode 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 ucode --help.

Fonctionnement du routage​

Lorsque vous acheminez un agent via un service de fournisseur de modèle, ucode définit l’URL de base de l’agent sur votre Endpoint Unity AI 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, ucode pin 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