Mosaic AI Vector Search: 予算ポリシー
プレビュー
この機能は パブリック プレビュー段階です。
この記事では、予算ポリシーを使用してベクトル検索のコストを追跡する方法について説明します。
予算ポリシー を使用すると、管理者はすべての Databricks サーバレス製品の請求レコードをグループ化およびフィルタリングし、支出を追跡するための専用 UI を提供できます。 予算ポリシーは、ワークスペース管理者によって作成され、ユーザーに割り当てられます。一般的な情報と予算ポリシーの作成・管理方法については、 サーバレス 予算ポリシーによる属性の使用を参照してください。
予算ポリシーを適用する
エンドポイントの作成時に予算ポリシーを適用することも、後でエンドポイントを編集してポリシーを適用することもできます。
エンドポイント作成時の予算ポリシーの適用
- Databricks UI
- Python SDK
- REST API
Follow these steps to create a vector search endpoint and apply a budget policy:
-
In the left sidebar, click Compute.
-
Click the Vector Search tab and click Create.
-
Click the arrow to the right of Advanced Settings to open that section of the dialog.
-
Select a policy from the Budget Policy dropdown menu.
-
Click Confirm.
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",
)
{
name="vector_search_endpoint_name",
budget_policy_id=”12345678-1234-1234-1234-1234567890ab”
}
エンドポイントの予算ポリシーの編集
Databricks UI を使用してエンドポイントの予算ポリシーを編集するには:
-
エンドポイント ページで、鉛筆アイコンをクリックします。
-
ドロップダウンメニューからポリシーを選択し、[ 保存 ]をクリックします。
Python SDK または REST API を使用してポリシーを編集するには、「 エンドポイントの作成時に予算ポリシーを適用する」のタブを参照してください。
予算ポリシー別のクエリ支出
予算ポリシー別に支出をクエリするには、予算ポリシー ID が必要です。UI を使用して予算ポリシー ID を取得するには、管理者ユーザーとして次の手順を実行します。
- Databricks ワークスペースの上部バーでユーザー名をクリックし、[ 設定 ] をクリックします。
- 「コンピュート」 をクリックします。
- サーバレス 予算ポリシー の横にある 管理 をクリックします。
- サーバレス 予算ポリシーを選択します。 予算ポリシー 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
制限
ベクトル検索の場合、予算ポリシーには次の制限があります。
- 予算ポリシーでは、ベクトル検索の費用制限は適用されません。
- 予算ポリシーは、サービスコストを追跡するためにエンドポイントにのみ適用されます。ベクトルインデックスには適用されません。