モデルプロバイダーサービスとコーディングエージェントを統合する
Claude CodeやCodex CLIなどのコーディングエージェントをモデルプロバイダーサービス経由でルーティングし、Unity Catalogに登録した外部プロバイダー(独自のOpenAI、Anthropic、またはAmazon Bedrockアカウントなど)に対して実行します。Unity Gatewayはリクエスト時に保存された認証情報を提供し、ガバナンスと使用状況の追跡を適用するため、コーディングエージェントがプロバイダーのシークレットを直接扱うことはありません。
代わりにDatabricks提供モデルにエージェントをルーティングするには、コーディングエージェントとの統合を参照してください。
機能
- 独自のプロバイダーを持ち込む。 Claude CodeまたはCodex CLIを、単一の統制されたUnity Catalogセキュリティ保護可能オブジェクトを通じて、独自のOpenAI、Anthropic、Amazon Bedrock、またはその他の登録済みプロバイダーにポイントします。
- 一元的なガバナンス。
EXECUTEの権限はモデルプロバイダーサービスへのアクセスを管理し、サービスレベルでレート制限とサービスポリシーを適用できます。 - 可観測性 Unity Gatewayは、使用状況追跡にリクエストを取り込みます。また、構成されている場合は推論テーブルにも取り込みます。
要件
-
クエリーを実行できるモデルプロバイダーサービス。以下の権限が必要です。「外部モデルプロバイダー(モデルプロバイダーサービス)の検出とアクセス管理」を参照してください。
EXECUTEモデルプロバイダーサービス上USE CATALOGカタログ上USE SCHEMAスキーマ上で
-
Unity Gateway CLI。Python 3.12以降および
uvが必要です。
Unity Gateway CLIを使用してコーディングエージェントをモデルプロバイダーサービスにルーティングする
Unity Gateway CLI は、サポートされているコーディングエージェントをインストール、認証、および構成して、Unity Gateway を通じてルーティングできるようにします。Databricks が提供するモデルの代わりにモデルプロバイダーサービスを通じてエージェントをルーティングするには、そのサービスの 3 部構成の名前を --provider に渡します。
ステップ1:Unity Gateway CLIをインストールする
uv tool install git+https://github.com/databricks/unity-gateway
ステップ 2:モデルプロバイダーサービスに対してエージェントを開きます
モデルプロバイダーサービスの完全修飾名(<catalog>.<schema>.<name>)を--providerに渡します:
ug claude --provider main.default.openai_prod # Claude Code
ug codex --provider main.default.openai_prod # Codex CLI
初回起動時に、Unity Gateway CLIはDatabricksワークスペースURLの入力を求め、認証を行い、リクエストがモデルプロバイダーサービスを経由するようにエージェントの構成ファイルを書き込みます。すべてのリクエストに Databricks-Model-Provider-Service ヘッダーを付加します。Unity Gatewayはサービスを解決し、保存された資格情報を提供して、リクエストを外部プロバイダーに転送します。
--provider は ug claude および ug codex でサポートされています。ワークスペースでモデルプロバイダーサービス機能が有効になっていない場合、またはサービスに対する EXECUTE が不足している場合、Unity Gateway CLIは詳細なエラーを出力します。
ステップ3(オプション):エージェントをインタラクティブに構成します
ガイド付きフローを通じてモデルプロバイダーサービスを選択するには、--providerを渡す代わりに、以下を実行します:
ug configure
Claude Code と Codex CLI では、ug configure に利用可能なモデルプロバイダーサービスが一覧表示され、その中から選択できます。どれも選択しない場合、エージェントはDatabricksが提供するモデル経由でルーティングされます。
完全なコマンドリファレンスについては、ug --help を実行してください。
ルーティングの仕組み
エージェントをモデルプロバイダーサービス経由でルーティングする場合、Unity Gateway CLIはエージェントのベースURLをUnity GatewayのEndpointに設定し、サービス名を指定するDatabricks-Model-Provider-Serviceヘッダーを追加します:
- Claude Code は
https://<workspace-url>/ai-gateway/anthropicに対して認証を行い、ANTHROPIC_CUSTOM_HEADERSにヘッダーを保持します。Amazon Bedrockを基盤とするサービスの場合、Unity Gateway CLIはプロバイダー側のモデルID(例:us.anthropic.claude-sonnet-4-6)をピン留めします。 - **Codex CLI**
https://<workspace-url>/ai-gateway/codex/v1はwire_api = "responses"を使用して に対して認証を行い、プロバイダーのhttp_headersにヘッダーを格納します。
エージェントが送信するモデルは、モデルプロバイダーサービスが許可するものでなければなりません。