レジスタエージェント(エージェントサービス)

エージェントをUnity Catalogに登録し、すべてのチームのエージェントが一か所に集まるようにしましょう。 エージェントサービスをエージェントサービスとして登録すると、テーブル、モデル、関数と一緒に閲覧・発見でき、他のUnity Catalog資産を保護するのと同じ権限で権限を設定することができます。

カタログ エクスプローラーでエージェントを検出する

エージェントサービスが登録されると、作成されたスキーマの下にカタログエクスプローラーに表示されます。 エージェントサービスを見るには、ユーザーが EXECUTEなどの明示的な権限が必要です。 エージェント サービスを選択して、その名前、説明、接続、および現在のアクセス許可を表示します。

comment フィールドを使用して、エージェント サービスを作成または更新するときに説明を追加します。 この説明はカタログ エクスプローラーに表示され、メンバーがエージェントを参照するときに実行される内容を理解するのに役立ちます。

これにより、組織は、存在するエージェント、エージェントの所有者、アクセス権を持つユーザーを、個別のシステムまたはドキュメントで追跡する必要なく、1 つの場所で確認できます。

Requirements

• Unity AI Gateway ベータ機能があなたのアカウントで有効になっています。 Manage Azure Databricks プレビューを参照してください。
• エージェント サービスを 作成 するには、親カタログとスキーマの USE CATALOG と USE SCHEMA 、スキーマに CREATE SERVICE し、エージェント サービスが参照する接続に USE CONNECTION する必要があります。
• エージェント サービスの アクセス許可を管理 するには、エージェント サービスの所有者であるか、 MANAGE 特権を持っている必要があります。

エージェント サービスを作成する

エージェント サービスは、エージェントのホストと資格情報を保持する既存の Unity カタログ接続を参照します。 エージェント サービスを作成する前に、接続を作成します。

次の例では、support_agent スキーマに main.default という名前のエージェント サービスを作成します。

databricks api post \
  "/api/2.1/unity-catalog/agent-services?parent=schemas/main.default&agent_service_id=support_agent" \
  --json '{
    "agent_service_type": "AGENT_SERVICE_TYPE_EXTERNAL",
    "comment": "Support agent for the customer team",
    "config": {
      "source_connection": {
        "name": "connections/main.default.my_agent_connection"
      },
      "base_path": "/v1/chat",
      "system_prompt": "You are a helpful support assistant."
    }
  }'

応答には、エージェント サービスのリソース名が含まれます。

{
  "name": "agent-services/main.default.support_agent",
  "agent_service_type": "AGENT_SERVICE_TYPE_EXTERNAL",
  "created_by": "you@company.com",
  "config": {
    "source_connection": { "name": "connections/main.default.my_agent_connection" },
    "base_path": "/v1/chat",
    "system_prompt": "You are a helpful support assistant."
  }
}

エージェント サービスの取得と一覧表示

単一のエージェント サービスを完全名で取得します。

databricks api get "/api/2.1/unity-catalog/agent-services/main.default.support_agent"

スキーマ内のすべてのエージェント サービスを一覧表示します。

databricks api get "/api/2.1/unity-catalog/agent-services?parent=schemas/main.default"

メタストア全体でエージェント サービスを一覧表示するには、スキーマ フィルターを省略します。

databricks api get "/api/2.1/unity-catalog/agent-services"

エージェント サービスを更新する

他のフィールドを上書きせずに特定のフィールドを更新するには、 PATCH 要求を使用し、変更するフィールドを指定します。 次の例では、他のすべてのフィールドを変更せずに、システム プロンプトを更新します。

databricks api patch \
  "/api/2.1/unity-catalog/agent-services/main.default.support_agent?update_mask=config.system_prompt" \
  --json '{ "config": { "system_prompt": "You are a concise support assistant." } }'

更新できるフィールド: comment、 config.system_prompt、 config.base_path。

アクセス権を付与する

他の Unity カタログのセキュリティ保護可能なオブジェクトを保護する同じ許可モデルを使用して、エージェント サービスに対するアクセス許可を設定します。 ユーザーとサービス プリンシパルに EXECUTE を付与します。

databricks api patch \
  "/api/2.1/unity-catalog/permissions/AGENT_SERVICE/main.default.support_agent" \
  --json '{
    "changes": [
      { "principal": "teammate@company.com", "add": ["EXECUTE"] }
    ]
  }'

現在の権限を確認:

databricks api get "/api/2.1/unity-catalog/permissions/AGENT_SERVICE/main.default.support_agent"

アクセスの取り消し:

databricks api patch \
  "/api/2.1/unity-catalog/permissions/AGENT_SERVICE/main.default.support_agent" \
  --json '{
    "changes": [
      { "principal": "teammate@company.com", "remove": ["EXECUTE"] }
    ]
  }'

割り当て可能な特権: EXECUTE、 READ METADATA、 MANAGE、 ALL PRIVILEGES。

エージェント サービスを削除する

databricks api delete "/api/2.1/unity-catalog/agent-services/main.default.support_agent"

Limitations

ベータ期間中は、次の制限事項が適用されます。

• ランタイム呼び出しは使用できません。 登録済みエージェント サービスを介してエージェントを呼び出すことはできません。 登録とアクセス許可の管理ができるようになりました。
• サービス ポリシーとレート制限 は、API ではサポートされていません。 これらのフィールドを設定しようとすると、エラーが返されます。
• エージェント サービスの SQL DDL は使用できません。 REST API を使用してエージェント サービスを作成および管理します。
• full_nameフィールドとownerフィールドは、このリリースのnull応答でGETを返します。
• Unity カタログ グローバル検索では、エージェント サービスは表示されません。
• BROWSE特権は支持されていません。 エージェントサービスを見るには、ユーザーが EXECUTEなどの明示的な権限が必要です。

次のステップ

• 中央の場所からの AI トラフィックを管理するための Unity AI Gateway を使用した AI ガバナンス。