Mosaic AI Vector Search: Políticas orçamentárias
Visualização
Este recurso está em Pré-visualização Pública.
Este artigo descreve como usar políticas orçamentárias para rastrear os custos de busca vetorial.
As políticas de orçamento permitem que os administradores agrupem e filtrem registros de faturamento em todos os produtos serverless Databricks e fornecem uma interface de usuário dedicada para o acompanhamento de gastos. As políticas de orçamento são criadas pelos administradores workspace e podem ser aplicadas ao ponto de extremidade e aos índices de pesquisa vetorial. Para informações gerais e detalhes sobre como criar e gerenciar políticas de orçamento, consulte Uso de atributos com políticas de orçamento serverless.
Para informações gerais sobre como gerenciar seus custos ao usar Mosaic AI Vector Search, consulte o guia de gerenciamento de custosMosaic AI Vector Search.
Aplicar uma política orçamentária a um endpoint
Você pode aplicar uma política de orçamento ao criar um endpoint ou pode editar o endpoint posteriormente para aplicar uma política.
Aplicar a política orçamentária na criação endpoint
- Databricks UI
- Python SDK
- REST API
Siga estes passos para criar um endpoint de pesquisa vetorial e aplicar uma política orçamentária:
-
Na barra lateral esquerda, clique em calcular .
-
Clique na tab Pesquisa Vetorial e clique em Criar .
-
Clique na seta à direita de Configurações Avançadas para abrir essa seção da caixa de diálogo.

-
Selecione uma política no menu dropdown "Política Orçamentária" .
-
Clique em Confirmar .
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”
}
Editar uma política orçamentária para um endpoint
Para editar uma política orçamentária para um endpoint usando a interface do usuário do Databricks:
-
Na página endpoint , clique no ícone de lápis.

-
Selecione uma política no menu dropdown e clique em Salvar .
Para editar uma política usando o SDK Python ou a API REST , consulte a aba em Aplicar política orçamentária na criação endpoint.
Aplicar uma política orçamentária a um índice
Você pode aplicar uma política orçamentária ao criar um índice ou pode editar o índice posteriormente para aplicar uma política.
Aplicar a política orçamentária na criação do índice.
- Databricks UI
- Python SDK
- REST API
Ao criar um índice de pesquisa vetorial na interface do usuário Databricks , você pode selecionar uma política de orçamento na dropdown "Política de orçamento sem servidor" em "Configurações avançadas " na caixa de diálogo de criação do índice. Consulte Criar índice usando a interface do usuário.
from databricks.vector_search.client import VectorSearchClient
client = VectorSearchClient()
client.create_delta_sync_index(
endpoint_name="vector_search_endpoint_name",
index_name="index_name",
source_table_name="catalog.schema.table",
primary_key="id",
pipeline_type="TRIGGERED",
budget_policy_id="12345678-1234-1234-1234-1234567890ab",
)
{
"name": "index_name",
"endpoint_name": "vector_search_endpoint_name",
"primary_key": "id",
"index_type": "DELTA_SYNC",
"delta_sync_index_spec": {
"source_table": "catalog.schema.table",
"pipeline_type": "TRIGGERED",
"budget_policy_id": "12345678-1234-1234-1234-1234567890ab"
}
}
Edite uma política orçamentária para um índice.
Para editar uma política orçamentária para um índice usando o SDK do Python:
client.update_index_budget_policy(
index_name="index_name",
budget_policy_id="12345678-1234-1234-1234-1234567890ab",
)
Consultar os gastos de acordo com a política orçamentária
Para consultar os gastos por política orçamentária, você precisa ter o ID da política orçamentária. Para obter o ID da política orçamentária usando a interface do usuário, siga estes passos como usuário administrador:
- Clique no seu nome de usuário na barra superior do workspace Databricks e clique em Configurações .
- Clique em calcular .
- Ao lado de políticas de orçamento sem servidor , clique em gerenciar .
- Selecione uma política de orçamento serverless . O ID da política orçamentária aparece na seção superior direita, "Sobre esta política" .
Para obter o ID da política orçamentária usando a API REST, consulte Listar políticas.
A consulta a seguir agrega os gastos pelo ID da política orçamentária. Substitua <INSERT BUDGET POLICY ID> pelo ID da política orçamentária.
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
Limitações
Para a busca vetorial, as políticas orçamentárias apresentam as seguintes limitações:
- As políticas orçamentárias não impõem limites de gastos para busca vetorial.