Politiques d'utilisation de la recherche IA
Aperçu
Cette fonctionnalité est en Aperçu public.
Vous pouvez appliquer une politique d'utilisation à un endpoint ou un index de recherche IA pour suivre ses dépenses serverless par équipe ou par projet.
Les politiques d'utilisation permettent aux administrateurs de regrouper et de filtrer les enregistrements de facturation sur tous les produits Databricks serverless, et de fournir une interface utilisateur dédiée pour le suivi des dépenses. Les politiques d'utilisation sont créées par les administrateurs de workspace et peuvent être appliquées aux endpoints et aux index de recherche IA. Pour des informations générales et des détails sur la création et la gestion des politiques d'utilisation, consultez Attribuer l'utilisation avec des politiques d'utilisation serverless.
Pour des informations générales sur la gestion de vos coûts lors de l'utilisation de la recherche IA, consultez le guide de gestion des coûts de la recherche IA.
Appliquer une politique d'utilisation à un endpoint
Vous pouvez appliquer une politique d'utilisation lorsque vous créez un endpoint, ou vous pouvez modifier l'endpoint plus tard pour appliquer une politique.
Appliquer une politique d'utilisation lors de la création de l'Endpoint
- Databricks UI
- Python SDK
- REST API
Suivez ces étapes pour créer un Endpoint de recherche IA et appliquer une politique d'utilisation :
-
Dans la barre latérale gauche, cliquez sur Compute .
-
Cliquez sur le tab **AI Search** et cliquez sur **Créer**.
-
Cliquez sur la flèche à droite de Paramètres avancés pour ouvrir cette section de la boîte de dialogue.

-
Sélectionnez une politique dans le menu déroulant Politique d'utilisation .
-
Cliquez sur Confirmer .
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”
}
Modifier une politique d'utilisation pour un Endpoint
Pour modifier une politique d'utilisation pour un Endpoint à l'aide de l'interface utilisateur Databricks :
-
Sur la page de l'Endpoint, cliquez sur l'icône de crayon.

-
Sélectionnez une politique dans le menu déroulant et cliquez sur **Enregistrer**.
Pour modifier une politique à l'aide du SDK Python ou de l'API REST, consultez les onglets sous Appliquer une politique d'utilisation lors de la création d'un endpoint.
Appliquer une politique d'utilisation à un index
Vous pouvez appliquer une politique d'utilisation lorsque vous créez un index, ou vous pouvez modifier l'index plus tard pour appliquer une politique.
Appliquer la politique d'utilisation lors de la création de l'index
- Databricks UI
- Python SDK
- REST API
Lors de la création d’un index de recherche IA dans l’interface utilisateur Databricks, vous pouvez sélectionner une politique d’utilisation dans la liste déroulante Politique d’utilisation , sous Paramètres avancés dans la boîte de dialogue de création d’index. Consultez Créer un index à l’aide de l’interface utilisateur.
from databricks.ai_search.client import AISearchClient
client = AISearchClient()
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"
}
}
Modifier une politique d'utilisation pour un index
Pour modifier une politique d'utilisation pour un index à l'aide du SDK Python :
client.update_index_budget_policy(
index_name="index_name",
budget_policy_id="12345678-1234-1234-1234-1234567890ab",
)
Query spending by politique d'utilisation
Pour query les dépenses par politique d'utilisation, vous devez avoir l'ID de la politique d'utilisation. Pour obtenir l’identifiant de la politique d’utilisation à l’aide de l’interface utilisateur, suivez ces étapes en tant qu’utilisateur administrateur :
- Cliquez sur votre nom d'utilisateur dans la barre supérieure du workspace Databricks et cliquez sur **Paramètres**.
- Click compute .
- À côté de Politiques d'utilisation Serverless , cliquez sur Gérer .
- Sélectionnez une politique d'utilisation Serverless. L'ID de la politique d'utilisation apparaît dans la section supérieure droite, À propos de cette politique .
Pour obtenir l'ID de la politique d'utilisation à l'aide de l'API REST, consultez Lister les politiques.
La query suivante agrège les dépenses par ID de politique d'utilisation. Remplacez <INSERT USAGE POLICY ID> par l'ID de la politique d'utilisation.
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
Limitations
Pour la recherche IA, les politiques d'utilisation ont les limitations suivantes :
- Les politiques d'utilisation n'appliquent pas de limites de dépenses pour AI Search.