次の方法で共有

モザイク AI ベクター検索: 予算ポリシー

重要

この機能はパブリック プレビュー段階にあります。

この記事では、予算ポリシーを使用してベクター検索コストを追跡する方法について説明します。

予算ポリシーを使用すると、管理者はすべての Azure Databricks サーバーレス製品の課金レコードをグループ化してフィルター処理し、支出を追跡するための専用 UI を提供できます。 予算ポリシーはワークスペース管理者によって作成され、ユーザーに割り当てられます。 予算ポリシーを作成および管理する方法に関する一般的な情報と詳細については、「 サーバーレス予算ポリシーを使用した属性の使用」を参照してください。

予算ポリシーを適用する

エンドポイントを作成するときに予算ポリシーを適用することも、後でエンドポイントを編集してポリシーを適用することもできます。

エンドポイントの作成時に予算ポリシーを適用する

Databricks ユーザーインターフェース

ベクター検索エンドポイントを作成し、予算ポリシーを適用するには、次の手順に従います。

  1. 左側のサイドバーで、[ コンピューティング] をクリックします。

  2. [ ベクター検索 ] タブをクリックし、[ 作成] をクリックします。

  3. [詳細設定] の右側にある矢印をクリックして、ダイアログのそのセクションを開きます。

    ベクター検索エンドポイントの作成ダイアログ。

  4. [予算ポリシー] ドロップダウン メニューからポリシーを選択します。

  5. [Confirm](確認) をクリックします。

Python SDK

client.create_endpoint(
   name="vector_search_endpoint_name",
   endpoint_type="STANDARD",
   budget_policy_id="12345678-1234-1234-1234-1234567890ab",
)

client.update_endpoint_budget_policy(
   name="vector_search_endpoint_name",
   budget_policy_id="12345678-1234-1234-1234-1234567890ab",
)

REST API

{
	name="vector_search_endpoint_name",
	budget_policy_id=”12345678-1234-1234-1234-1234567890ab”
}

エンドポイントの予算ポリシーを編集する

Databricks UI を使用してエンドポイントの予算ポリシーを編集するには:

  1. エンドポイント ページで、鉛筆アイコンをクリックします。

    エンドポイント ページで予算ポリシーを編集する鉛筆アイコン。

  2. ドロップダウン メニューからポリシーを選択し、[ 保存] をクリックします。

Python SDK または REST API を使用してポリシーを編集するには、[ エンドポイントの作成時に予算ポリシーを適用する] のタブを参照してください。

予算ポリシー別に支出を照会する

予算ポリシー別に支出を照会するには、予算ポリシー ID が必要です。 UI を使用して予算ポリシー ID を取得するには、管理者ユーザーとして次の手順に従います。

  1. Databricks ワークスペースの上部バーでユーザー名をクリックし、[ 設定] をクリックします。
  2. [コンピューティング] をクリックします。
  3. サーバーレス予算ポリシーの横にある [管理]をクリックします。
  4. サーバーレス予算ポリシーを選択します。 [ このポリシーについて] セクションの右上に予算ポリシー ID が表示されます。

REST API を使用して予算ポリシー ID を取得するには、「 ポリシーの一覧表示」を参照してください。

次のクエリは、予算ポリシー ID ごとに支出を集計します。 <INSERT BUDGET POLICY ID>を予算ポリシー ID に置き換えます。

WITH vector_search_usage_by_budget_policy (
 SELECT *,
        CASE WHEN usage_type = "STORAGE_SPACE" THEN 'storage'
             ELSE 'serving'
       END as workload_type
   FROM system.billing.usage
  WHERE billing_origin_product = 'VECTOR_SEARCH'
  AND usage_metadata.endpoint_name IS NOT NULL
  AND usage_metadata.budget_policy_id = '<INSERT BUDGET POLICY ID>'
),
daily_usage_by_budget_policy AS (
 SELECT workspace_id,
      cloud,
      usage_date,
      workload_type,
      usage_metadata.endpoint_name as vector_search_endpoint,
      CASE WHEN workload_type = 'serving' THEN SUM(usage_quantity)
           ELSE null
           END as dbus,
      CASE WHEN workload_type = 'storage' THEN SUM(usage_quantity)
           ELSE null
           END as dsus
FROM vector_search_usage_by_budget_policy
GROUP BY all
ORDER BY 1,2,3,4,5,6 DESC
)
SELECT * FROM daily_usage_by_budget_policy

制限事項

Vector Search の場合、予算ポリシーには次の制限があります。

  • 予算ポリシーでは、ベクター検索の使用制限は適用されません。
  • 予算ポリシーは、サービス コストを追跡するためにエンドポイントにのみ適用されます。 これらはベクター インデックスには適用されません。
  • Last updated on