Aller au contenu principal

Politiques d'utilisation de la recherche IA

info

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

Suivez ces étapes pour créer un Endpoint de recherche IA et appliquer une politique d'utilisation :

  1. Dans la barre latérale gauche, cliquez sur Compute .

  2. Cliquez sur le tab **AI Search** et cliquez sur **Créer**.

  3. Cliquez sur la flèche à droite de Paramètres avancés pour ouvrir cette section de la boîte de dialogue.

    Boîte de dialogue : Créer un endpoint de recherche IA.

  4. Sélectionnez une politique dans le menu déroulant Politique d'utilisation .

  5. Cliquez sur Confirmer .

Modifier une politique d'utilisation pour un Endpoint

Pour modifier une politique d'utilisation pour un Endpoint à l'aide de l'interface utilisateur Databricks :

  1. Sur la page de l'Endpoint, cliquez sur l'icône de crayon.

    Icône en forme de crayon pour modifier les politiques d'utilisation sur la page de l'Endpoint.

  2. 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

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.

Modifier une politique d'utilisation pour un index

Pour modifier une politique d'utilisation pour un index à l'aide du SDK Python :

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 :

  1. Cliquez sur votre nom d'utilisateur dans la barre supérieure du workspace Databricks et cliquez sur **Paramètres**.
  2. Click compute .
  3. À côté de Politiques d'utilisation Serverless , cliquez sur Gérer .
  4. 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.

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 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.