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

Mosaic AI Vector Search: 予算ポリシー

備考

プレビュー

この機能は パブリック プレビュー段階です。

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

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

Mosaic AI Vector Searchを使用する場合のコスト管理方法に関する一般的な情報については、「Mosaic AI Vector Search: コスト管理ガイド」を参照してください。

予算ポリシーを適用する

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

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

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

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

  2. ベクトル検索 」タブをクリックし、「 作成 」をクリックします。

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

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

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

  5. 確認 をクリックします。

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

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 に置き換えます。

SQL
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

制限

ベクトル検索の場合、予算ポリシーには次の制限があります。

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