Intégration de Vertex AI

Cette page explique comment configurer l'intégration entre AlloyDB pour PostgreSQL et Vertex AI, ce qui vous permet d'appliquer des requêtes à des grands modèles de langage (LLM) stockés dans le cloud pour vos données.

Ces instructions concernent spécifiquement l'utilisation d'AlloyDB, et non d'AlloyDB Omni. Pour intégrer une installation locale d'AlloyDB Omni à l'aide de Vertex AI, consultez Installer AlloyDB Omni avec AlloyDB/AI.

Pour en savoir plus sur l'utilisation de modèles de ML avec AlloyDB, consultez Créer des applications d'IA générative à l'aide d'AlloyDB AI.

Pour en savoir plus sur Vertex AI, consultez la Présentation de Vertex AI.

Avant de commencer

  1. Sign in to your Google Cloud account. If you're new to Google Cloud, create an account to evaluate how our products perform in real-world scenarios. New customers also get $300 in free credits to run, test, and deploy workloads.
  2. In the Google Cloud console, on the project selector page, select or create a Google Cloud project.

    Roles required to select or create a project

    • Select a project: Selecting a project doesn't require a specific IAM role—you can select any project that you've been granted a role on.
    • Create a project: To create a project, you need the Project Creator (roles/resourcemanager.projectCreator), which contains the resourcemanager.projects.create permission. Learn how to grant roles.

    Go to project selector

  3. Verify that billing is enabled for your Google Cloud project.

  4. Enable the Vertex AI API.

    Roles required to enable APIs

    To enable APIs, you need the Service Usage Admin IAM role (roles/serviceusage.serviceUsageAdmin), which contains the serviceusage.services.enable permission. Learn how to grant roles.

    Enable the API

  5. Install the Google Cloud CLI.

  6. Si vous utilisez un fournisseur d'identité (IdP) externe, vous devez d'abord vous connecter à la gcloud CLI avec votre identité fédérée.

  7. Pour initialiser la gcloud CLI, exécutez la commande suivante :

    gcloud init
  8. In the Google Cloud console, on the project selector page, select or create a Google Cloud project.

    Roles required to select or create a project

    • Select a project: Selecting a project doesn't require a specific IAM role—you can select any project that you've been granted a role on.
    • Create a project: To create a project, you need the Project Creator (roles/resourcemanager.projectCreator), which contains the resourcemanager.projects.create permission. Learn how to grant roles.

    Go to project selector

  9. Verify that billing is enabled for your Google Cloud project.

  10. Enable the Vertex AI API.

    Roles required to enable APIs

    To enable APIs, you need the Service Usage Admin IAM role (roles/serviceusage.serviceUsageAdmin), which contains the serviceusage.services.enable permission. Learn how to grant roles.

    Enable the API

  11. Install the Google Cloud CLI.

  12. Si vous utilisez un fournisseur d'identité (IdP) externe, vous devez d'abord vous connecter à la gcloud CLI avec votre identité fédérée.

  13. Pour initialiser la gcloud CLI, exécutez la commande suivante :

    gcloud init
  14. Accorder l'autorisation Utilisateur Vertex AI à l'agent de service AlloyDB

    Pour activer l'intégration de la base de données avec Vertex AI, procédez comme suit pour accorder à l'agent de service AlloyDB les autorisations Identity and Access Management (IAM) pour accéder à Vertex AI :

    Ajoutez des autorisations Vertex AI à l'agent de service AlloyDB pour le projet où se trouve le cluster de la base de données AlloyDB :

    Console

    1. Accédez à la page Bienvenue dans la console Google Cloud , puis copiez le numéro du projet qui contient des clusters ou des instances AlloyDB. Vous utiliserez ce numéro de projet dans les étapes suivantes.

      Accéder à la page d'accueil

    2. Dans la console Google Cloud , accédez à la page IAM.

      Accéder à IAM

    3. Sélectionnez le projet contenant les points de terminaison Vertex AI.

    4. Activez l'option Inclure les attributions de rôles fournies par Google.

    5. Cliquez sur Accorder l'accès.

    6. Dans le champ Nouveaux comptes principaux, saisissez les éléments suivants :

          service-PROJECT_NUMBER@gcp-sa-alloydb.iam.gserviceaccount.com
          

      Remplacez PROJECT_NUMBER par le numéro du projet.

    7. Dans le champ Rôle, saisissez Utilisateur Vertex AI.

    8. Cliquez sur Enregistrer.

    gcloud

    Pour utiliser la gcloud CLI, vous pouvez installer et initialiser Google Cloud CLI, ou utiliser Cloud Shell.

            gcloud projects add-iam-policy-binding PROJECT_ID 
    --member="serviceAccount:service-PROJECT_NUMBER@gcp-sa-alloydb.iam.gserviceaccount.com"
    --role="roles/aiplatform.user"

    Remplacez les éléments suivants :

    • PROJECT_ID : ID du projet contenant le point de terminaison Vertex AI.
    • PROJECT_NUMBER : numéro du projet contenant des clusters ou des instances AlloyDB.

    Vérifier l'extension installée

    Vérifiez si google_ml_integration est installé dans la base de données contenant les données sur lesquelles vous souhaitez exécuter des prédictions :

    Console

    1. Dans la console Google Cloud , accédez à la page Clusters.

      accéder aux clusters

    2. Pour afficher la page Présentation du cluster, cliquez sur le nom du cluster AlloyDB dans la colonne Nom de la ressource.

    3. Dans le menu de navigation, cliquez sur AlloyDB Studio.

    4. Sur la page Se connecter à AlloyDB Studio, authentifiez-vous à l'aide du nom de votre base de données, de votre nom d'utilisateur et de votre mot de passe.

    5. Dans l'onglet Éditeur 1, procédez comme suit :

      1. Vérifiez que l'extension google_ml_integration version 1.4.2 ou ultérieure est installée :

        SELECT extversion FROM pg_extension WHERE extname = 'google_ml_integration';
      2. Cliquez sur Exécuter. Attendez que la version de l'extension s'affiche dans le volet Résultats.

    psql

    1. Connectez un client psql à l'instance principale du cluster, comme décrit dans Connecter un client psql à une instance.

    2. Lorsque l'invite de commande psql s'affiche, connectez-vous à la base de données :

      \c DB_NAME

      Remplacez DB_NAME par le nom de la base de données sur laquelle vous souhaitez installer l'extension.

    3. Vérifiez que l'extension google_ml_integration version 1.4.2 ou ultérieure est installée :

      SELECT extversion FROM pg_extension WHERE extname = 'google_ml_integration';

    Étapes suivantes