Políticas de uso da Pesquisa de AI
Visualização
Esse recurso está em Prévia Pública.
Este artigo descreve como usar políticas de uso para monitorar os custos de busca de IA.
Políticas de uso permitem aos administradores agrupar e filtrar registros de cobrança em todos os produtos serverless do Databricks e fornecem uma interface de usuário dedicada para acompanhamento de gastos. Políticas de uso são criadas por administradores de workspace e podem ser aplicadas a endpoints e índices de Pesquisa de AI. Para obter informações gerais e detalhes sobre como criar e gerenciar políticas de uso, consulte Uso de atributos com políticas de uso serverless.
Para obter informações gerais sobre como gerenciar seus custos ao usar a Pesquisa de IA, consulte guia de gerenciamento de custos da Pesquisa de IA.
Aplicar uma política de uso a um endpoint
Você pode aplicar uma política de uso ao criar um endpoint, ou editar o endpoint mais tarde para aplicar uma política.
Aplicar política de uso na criação de endpoint
- Databricks UI
- Python SDK
- REST API
Siga estes passos para criar um endpoint de Pesquisa de IA e aplicar uma política de uso:
-
Na barra lateral esquerda, clique em Compute .
-
Clique na **tab AI Search** 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 de Uso**.
-
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 de uso para um endpoint
Para editar uma política de uso para um endpoint com a interface do usuário do Databricks:
-
Na página do endpoint, clique no ícone de lápis.

-
Selecione uma política no menu suspenso e clique em Salvar .
Para editar uma política usando o SDK Python ou a API REST, consulte as tabs em Aplicar política de uso na criação do endpoint.
Aplicar uma política de uso a um índice
É possível aplicar uma política de uso ao criar um índice, ou editar o índice posteriormente para aplicar uma política.
Aplicar política de uso na criação do índice
- Databricks UI
- Python SDK
- REST API
Ao criar um índice de Pesquisa de AI na UI do Databricks, é possível selecionar uma política de uso no dropdown **Política de Uso** 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"
}
}
Editar política de uso para um índice
Para editar uma política de uso para um índice usando o SDK para Python:
client.update_index_budget_policy(
index_name="index_name",
budget_policy_id="12345678-1234-1234-1234-1234567890ab",
)
Consultar gastos por política de uso
Para consultar gastos por política de uso, é necessário possuir o ID da política de uso. Para obter o ID da política de uso usando a IU, siga estes passos como um usuário administrador:
- Clique no seu nome de usuário na barra superior do workspace do Databricks e clique em **Configurações**.
- Click compute .
- Ao lado de **políticas de uso serverless**, clique em **Gerenciar**.
- Selecione uma política de uso serverless. O ID da política de uso aparece na seção superior direita, **Sobre esta política**.
Para obter o ID da política de uso usando a API REST, consulte Listar políticas.
A seguinte consulta agrega gastos pelo ID da política de uso. Substituir <INSERT USAGE POLICY ID> pelo ID da política de uso.
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 USAGE 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 Pesquisa de AI, as políticas de uso têm as seguintes limitações:
- Políticas de uso não impõem limites de gastos para AI Pesquisa.