メインコンテンツまでスキップ

モデルプロバイダーサービスとコーディングエージェントを統合する

備考

ベータ版

この機能はベータ版です。アカウント管理者は、アカウントコンソールの [プレビュー] ページからこの機能へのアクセスを管理できます。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 は、使用状況追跡でリクエストをキャプチャし、設定されている場合は推論テーブルでもキャプチャします。

要件

ucodeでコーディングエージェントをモデルプロバイダーサービスにルーティングする

ucode (Unity AI Gateway Coding CLI)は、サポートされているコーディングエージェントをインストール、認証、設定して、Unity AI Gateway 経由でルーティングします。Databricks が提供するモデルの代わりに、モデルプロバイダーサービスを介してエージェントをルーティングするには、そのサービスの3部構成の名前を--providerに渡します。

ステップ1: ucode をインストール

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

ステップ 2:モデルプロバイダーサービスに対してエージェントを開きます

モデルプロバイダーサービスの完全修飾名(<catalog>.<schema>.<name>)を--providerに渡します:

Bash
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を渡す代わりに、以下を実行します:

Bash
ucode configure

Claude Code と Codex CLI では、ucode configure に利用可能なモデルプロバイダーサービスが一覧表示され、その中から選択できます。どれも選択しない場合、エージェントはDatabricksが提供するモデル経由でルーティングされます。

完全なコマンドリファレンスについては、ucode --help を実行してください。

ルーティングの仕組み

モデルプロバイダーサービスを介してエージェントをルーティングすると、ucodeがエージェントのベースURLをUnity AI Gatewayエンドポイントに設定し、サービスの名前を指定するDatabricks-Model-Provider-Serviceヘッダーを追加します:

  • Claude Codehttps://<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/v1wire_api = "responses" を使用して に対して認証を行い、プロバイダーのhttp_headers にヘッダーを格納します。

エージェントが送信するモデルは、モデルプロバイダーサービスが許可するものでなければなりません。allow_all_targetstargetsについては、モデルプロバイダーサービスを作成および管理するをご覧ください。

次のステップ