Integración con Agent Platform

En esta página, se detalla cómo configurar la integración entre AlloyDB para PostgreSQL y la plataforma de agentes de Gemini Enterprise, lo que te permite emitir consultas que pasan tus datos a los modelos de la plataforma de agentes.

Estas instrucciones son específicas para el uso de AlloyDB, no de AlloyDB Omni. Si, en cambio, deseas integrar una instalación local de AlloyDB Omni con la Plataforma de agentes, consulta Cómo instalar AlloyDB Omni con AlloyDB AI.

Para obtener más información sobre el uso de modelos de AA con AlloyDB, consulta Compila aplicaciones de IA generativa con AlloyDB AI.

Para obtener más información sobre Agent Platform, consulta la Descripción general de Agent Platform.

Antes de comenzar

  1. Accede a tu cuenta de Google Cloud . Si eres nuevo en Google Cloud, crea una cuenta para evaluar el rendimiento de nuestros productos en situaciones reales. Los clientes nuevos también obtienen $300 en créditos gratuitos para ejecutar, probar y, además, implementar cargas de trabajo.

  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 role (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 Agent Platform 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. Instala Google Cloud CLI.

  6. Si usas un proveedor de identidad externo (IdP), primero debes acceder a la gcloud CLI con tu identidad federada.

  7. Para inicializar gcloud CLI, ejecuta el siguiente comando: 

    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 role (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 Agent Platform 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. Instala Google Cloud CLI.

  12. Si usas un proveedor de identidad externo (IdP), primero debes acceder a la gcloud CLI con tu identidad federada.

  13. Para inicializar gcloud CLI, ejecuta el siguiente comando: 

    gcloud init

Acerca del agente de servicio de AlloyDB

Cuando usas AlloyDB con otros servicios de Google Cloud , AlloyDB usa un agente de servicio de Identity and Access Management (IAM) para autenticarse en esos servicios. Este agente de servicio es una principal de IAM que controla AlloyDB. Otorgas roles de IAM a este agente de servicio para permitirle acceder a otros servicios de Google Cloud en tu nombre.

Otorga el rol Vertex AI User al agente de servicio de AlloyDB

Para habilitar la integración de la base de datos con Agent Platform, otorga al agente de servicio de AlloyDB el rol Vertex AI User. Esto permite que el agente de servicio de AlloyDB acceda a Agent Platform en tu nombre.

Sigue estos pasos para otorgar al agente de servicio de AlloyDB el rol de Vertex AI User:

Agrega permisos de Agent Platform al agente de servicio de AlloyDB para el proyecto en el que se encuentra el clúster de la base de datos de AlloyDB:

Console

  1. Ve a la página Welcome en la consola de Google Cloud y copia el número del proyecto que tiene clústeres o instancias de AlloyDB. Usarás este número de proyecto en los próximos pasos.

    Ir a Bienvenida

  2. En la consola de Google Cloud , dirígete a la página IAM.

    Ir a IAM

  3. Selecciona el proyecto que necesita llamar a los extremos de Agent Platform.

  4. Habilita Incluir asignaciones de roles proporcionadas por Google.

  5. Haz clic en Otorgar acceso.

  6. En el campo Principales nuevas, ingresa lo siguiente:

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

    Reemplaza PROJECT_NUMBER por el número del proyecto.

  7. En el campo Rol, ingresa Usuario de Vertex AI.

  8. Haz clic en Guardar.

gcloud

Para usar gcloud CLI, puedes instalar y, luego, inicializar Google Cloud CLI, o bien usar 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"

Reemplaza lo siguiente:

  • PROJECT_ID: Es el ID del proyecto que tiene el extremo de Agent Platform.
  • PROJECT_NUMBER: Es el número del proyecto que tiene clústeres o instancias de AlloyDB.

Verifica la extensión instalada

Verifica si google_ml_integration está instalado en la base de datos que contiene los datos sobre los que deseas ejecutar predicciones:

Console

  1. En la consola de Google Cloud , ve a la página Clústeres.

    Ir a los clústeres

  2. Para mostrar la página Descripción general del clúster, haz clic en el nombre del clúster de AlloyDB en la columna Nombre del recurso.

  3. En el menú de navegación, haz clic en AlloyDB Studio.

  4. En la página Accede a AlloyDB Studio, autentícate con el nombre de tu base de datos, tu nombre de usuario y tu contraseña.

  5. En la pestaña Editor 1, completa lo siguiente:

    1. Verifica que esté instalada la extensión google_ml_integration versión 1.4.2 o posterior:

      SELECT extversion FROM pg_extension WHERE extname = 'google_ml_integration';

    2. Haz clic en Ejecutar. Espera a que se muestre la versión de la extensión en el panel Resultados.

psql

  1. Conecta un cliente psql a la instancia principal del clúster, como se describe en Conecta un cliente psql a una instancia.

  2. En el símbolo del sistema de psql, conéctate a la base de datos: 

    \c DB_NAME

    Reemplaza DB_NAME por el nombre de la base de datos en la que deseas instalar la extensión.

  3. Verifica que esté instalada la extensión google_ml_integration versión 1.4.2 o posterior: 

    SELECT extversion FROM pg_extension WHERE extname = 'google_ml_integration';

¿Qué sigue?