Integración con Vertex AI

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

Estas instrucciones son específicas para usar AlloyDB y no AlloyDB Omni. Para integrar una instalación local de AlloyDB Omni con Vertex AI, consulta Instalar AlloyDB Omni con AlloyDB AI.

Para obtener más información sobre cómo usar modelos de aprendizaje automático con AlloyDB, consulta el artículo Crea aplicaciones de IA generativa con AlloyDB AI.

Para obtener más información sobre Vertex AI, consulta la introducción a Vertex AI.

Antes de empezar

  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 utilizas un proveedor de identidades (IdP) externo, primero debes iniciar sesión en la CLI de gcloud 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 utilizas un proveedor de identidades (IdP) externo, primero debes iniciar sesión en la CLI de gcloud con tu identidad federada.

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

    gcloud init

    14. Conceder 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 conceder los permisos de gestión de identidades y accesos (IAM) del agente de servicio de AlloyDB para acceder a Vertex AI:

    Añade permisos de Vertex AI a la cuenta de servicio de AlloyDB del proyecto en el que se encuentra el clúster de la base de datos de AlloyDB:

    Consola

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

      Ir a Bienvenida

    2. En la consola, ve a la página Gestión de identidades y accesos. Google Cloud

      Ir a IAM

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

    4. Habilita Incluir concesiones de roles proporcionadas por Google.

    5. Haz clic en Conceder acceso.

    6. En el campo New principals (Nuevos principales), introduce lo siguiente:

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

      Sustituye PROJECT_NUMBER por el número del proyecto.

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

    8. Haz clic en Guardar.

    gcloud

    Para usar la CLI de gcloud, puedes instalar e inicializar Google Cloud CLI o 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"

    Haz los cambios siguientes:

    • PROJECT_ID: el ID del proyecto que tiene el endpoint de Vertex AI.
    • PROJECT_NUMBER: número del proyecto que tiene clústeres o instancias de AlloyDB.

    Verificar la extensión instalada

    Verifica si google_ml_integration está instalado en la base de datos que contiene los datos con los que quieres generar predicciones:

    Consola

    1. En la Google Cloud consola, ve a la página Clusters.

      Ir a Clústeres

    2. Para mostrar la página Resumen del clúster, haga 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 Iniciar sesión en 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, haz lo siguiente:

      1. Comprueba que la extensión google_ml_integration tenga la versión 1.4.2 o una 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 de psql a la instancia principal del clúster, tal como se describe en Conectar un cliente de psql a una instancia.

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

      \c DB_NAME

      Sustituye DB_NAME por el nombre de la base de datos en la que quieras instalar la extensión.

    3. Comprueba que la extensión google_ml_integration tenga la versión 1.4.2 o una posterior: 

      SELECT extversion FROM pg_extension WHERE extname = 'google_ml_integration';

    Siguientes pasos