Intégrer à Agent Platform

L'intégration d'AlloyDB pour PostgreSQL à Gemini Enterprise Agent Platform vous permet d'exécuter des requêtes SQL qui transmettent les données de votre base de données directement aux modèles Agent Platform pour la génération d'embeddings et les prédictions en ligne.

Ces instructions sont spécifiques à l'utilisation d'AlloyDB, et non d'AlloyDB Omni. Pour intégrer une installation locale d'AlloyDB Omni à l'aide d'Agent Platform, 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 Agent Platform, consultez la présentation d'Agent Platform.

Avant de commencer

Connectez-vous à votre compte Google Cloud . Si vous débutez sur Google Cloud, créez un compte pour évaluer les performances de nos produits en conditions réelles. Les nouveaux clients bénéficient également de 300 $ de crédits sans frais pour exécuter, tester et déployer des charges de travail.

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 role (roles/resourcemanager.projectCreator), which contains the resourcemanager.projects.create permission. Learn how to grant roles.

Note: If you don't plan to keep the resources that you create in this procedure, create a project instead of selecting an existing project. After you finish these steps, you can delete the project, removing all resources associated with the project.

Go to project selector

Verify that billing is enabled for your Google Cloud project.

Enable the Agent Platform API, if it is not already enabled.

Roles required to enable APIs

To enable APIs, you need the serviceusage.services.enable permission. If you created the project, then you likely already have this permission through the Owner role (roles/owner). Otherwise, you can get this permission through the Service Usage Admin role (roles/serviceusage.serviceUsageAdmin). Learn how to grant roles.

Enable the API

Install the Google Cloud CLI.

If you're using an external identity provider (IdP), you must first sign in to the gcloud CLI with your federated identity.

To initialize the gcloud CLI, run the following command:

gcloud init

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 role (roles/resourcemanager.projectCreator), which contains the resourcemanager.projects.create permission. Learn how to grant roles.

Note: If you don't plan to keep the resources that you create in this procedure, create a project instead of selecting an existing project. After you finish these steps, you can delete the project, removing all resources associated with the project.

Go to project selector

Verify that billing is enabled for your Google Cloud project.

Enable the Agent Platform API, if it is not already enabled.

Roles required to enable APIs

To enable APIs, you need the serviceusage.services.enable permission. If you created the project, then you likely already have this permission through the Owner role (roles/owner). Otherwise, you can get this permission through the Service Usage Admin role (roles/serviceusage.serviceUsageAdmin). Learn how to grant roles.

Enable the API

Install the Google Cloud CLI.

If you're using an external identity provider (IdP), you must first sign in to the gcloud CLI with your federated identity.

To initialize the gcloud CLI, run the following command:

gcloud init

À propos de l'agent de service AlloyDB

Lorsque vous utilisez AlloyDB avec d'autres services Google Cloud , AlloyDB utilise un agent de service Identity and Access Management (IAM) pour s'authentifier auprès de ces services. Cet agent de service est un compte principal IAM contrôlé par AlloyDB. Vous attribuez des rôles IAM à cet agent de service pour lui permettre d'accéder à d'autres services Google Cloud en votre nom.

Attribuer le rôle Vertex AI User à l'agent de service AlloyDB

Pour activer l'intégration de la base de données à Agent Platform, vous devez attribuer le rôle Vertex AI User à l'agent de service AlloyDB. Cela permet à l'agent de service AlloyDB d'accéder à Agent Platform en votre nom.

Remarque : Pour attribuer des rôles IAM, vous devez disposer de l'autorisation resourcemanager.projects.setiamPolicy. Les rôles "Propriétaire" et "Administrateur IAM du projet" incluent cette autorisation. Le rôle d'administrateur AlloyDB inclut également les autorisations requises pour attribuer le rôle d'utilisateur Vertex AI à l'agent de service. Si vous ne disposez pas des autorisations nécessaires, contactez l'administrateur de votre organisation.

Pour attribuer le rôle Vertex AI User à l'agent de service AlloyDB, procédez comme suit :

Ajoutez des autorisations Agent Platform à 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 de la console Google Cloud , puis copiez le numéro du projet contenant 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 qui doit appeler les points de terminaison Agent Platform.

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 informations suivantes :

       [e-mail]
       

   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 la Google Cloud CLI, ou utiliser Cloud Shell.

        gcloud projects add-iam-policy-binding PROJECT_ID 

        --member="serviceAccount:[e-mail]" 

        --role="roles/aiplatform.user"
  

Remplacez les éléments suivants :

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

Remarque : La modification de la règle prend effet dans un délai de 60 secondes à sept minutes.

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

   Remarque : AlloyDB Studio se connecte à l'instance principale de votre cluster, où sont gérées la gestion des extensions et les prédictions initiales, même si vous prévoyez d'appeler des prédictions à partir d'instances de pool de lecture ultérieurement.

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

   1. Vérifiez que la version 1.4.2 ou ultérieure de l'extension google_ml_integration 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.

   Remarque : Vous devez vous connecter à l'instance principale, même si vous avez l'intention d'appeler des prédictions en étant connecté à une instance de pool de lecture.

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

• Utiliser des embeddings

• Appeler des prédictions

• Quotas et limites d'Agent Platform

• Appeler des points de terminaison de modèles distants