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