Integración en Vertex AI

En esta página, se detalla cómo configurar la integración entre AlloyDB para PostgreSQL y Vertex AI, lo que te permite aplicar consultas a modelos de lenguaje grandes (LLM) almacenados en la nube a tus datos.

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 Vertex AI, 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 Vertex AI, consulta Introducción a Vertex AI.

Antes de comenzar

  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 usas un proveedor de identidad externo (IdP), primero debes acceder a 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 (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 usas un proveedor de identidad externo (IdP), primero debes acceder a gcloud CLI con tu identidad federada.

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

    gcloud init

    14. Otorga permiso de usuario de Vertex AI al agente de servicio de AlloyDB

    Para habilitar la integración de la base de datos con Vertex AI, sigue estos pasos para otorgar al agente de servicio de AlloyDB permisos de Identity and Access Management (IAM) para acceder a Vertex AI:

    Agrega permisos de Vertex AI 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 , ve a la página IAM.

      Ir a IAM

    3. Selecciona el proyecto que tiene extremos de Vertex AI.

    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 Vertex AI.
    • 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?