Instalar AlloyDB Omni con AlloyDB AI

Selecciona una versión de la documentación:

En esta página se explica cómo instalar AlloyDB Omni e integrar AlloyDB AI en él.

AlloyDB AI es un conjunto de funciones incluidas en AlloyDB Omni que te permiten crear aplicaciones empresariales de IA generativa. Para obtener más información sobre las funciones de aprendizaje automático de AlloyDB, consulta Crear aplicaciones de IA generativa.

AlloyDB Omni con AlloyDB AI te permite consultar modelos de aprendizaje automático remotos para trabajar con predicciones online e incrustaciones de texto generadas a partir de modelos de aprendizaje automático. AlloyDB Omni con AlloyDB AI también puede procesar incrustaciones de vectores de otro contenido, como una imagen, por ejemplo, si usas la interfaz google_ml.predict_row y haces la traducción tú mismo en la consulta.

En función de dónde quieras instalar AlloyDB Omni con AlloyDB AI, selecciona una de las siguientes opciones:

Configurar la instancia de AlloyDB Omni para consultar modelos remotos

Para configurar AlloyDB Omni de forma que consulte modelos remotos, sigue estos pasos:

  1. Crea una cuenta de servicio con Google Cloud.

  2. Crea una clave de cuenta de servicio en formato JSON y descárgala.

  3. Guarda la clave en una ubicación permanente de tu sistema de archivos. Se encuentra en esta ubicación durante la vida útil de tu servidor AlloyDB Omni.

    Anota su ubicación en el sistema de archivos, ya que la necesitarás en el siguiente paso.

  4. Añade permisos de gestión de identidades y accesos (IAM) de Vertex AI al proyecto y a la cuenta de servicio correspondientes.

       gcloud projects add-iam-policy-binding PROJECT_ID \
       --member="serviceAccount:SERVICE_ACCOUNT_ID" \
       --role="roles/aiplatform.user"

    Haz los cambios siguientes:

    • PROJECT_ID: el ID de tu proyecto de Google Cloud .

    • SERVICE_ACCOUNT_ID: el ID de la cuenta de servicio que has creado en el paso anterior. Esto incluye el sufijo @PROJECT_ID.iam.gserviceaccount.com completo, por ejemplo, my-service@my-project.iam.gserviceaccount.com.

Instalar AlloyDB Omni con AlloyDB AI

Para instalar AlloyDB Omni e integrar AlloyDB AI, sigue estos pasos:

  1. Asegúrate de haber completado todos los pasos que se indican en Configurar AlloyDB Omni para consultar modelos basados en la nube.

  2. Crea una copia local del archivo de configuración predeterminado de AlloyDB Omni:

       docker run -i --rm google/alloydbomni:15.5.4 cat /usr/share/postgresql/postgresql.conf.sample > my-postgres.conf

  3. Edita la copia del archivo de configuración para añadir directivas de integración de AlloyDB AI:

       echo "omni_enable_ml_agent_process = 'on'" >> my-postgres.conf
   echo "omni_google_cloud_private_key_file_path = '/etc/postgresql/private-key.json'" >> my-postgres.conf

  4. Ajusta los permisos del sistema de archivos de los archivos de clave y de configuración:

       sudo chmod +r my-postgres.conf
   sudo chmod +r KEY_PATH

    Sustituye KEY_PATH por la ruta al archivo de clave privada de la cuenta de servicio en el sistema de archivos de tu host. Esta es la clave que has creado y descargado anteriormente en este procedimiento.

  5. Instala los archivos de clave y de configuración montados en el contenedor:

       docker run --name CONTAINER_NAME -e POSTGRES_PASSWORD=NEW_PASSWORD -p HOST_PORT:5432 -v "$PWD/my-postgres.conf":/etc/postgresql/postgresql.conf -v "FULL_KEY_PATH":/etc/postgresql/private-key.json -d google/alloydbomni:15.5.4 -c 'config_file=/etc/postgresql/postgresql.conf'

    Haz los cambios siguientes:

    • CONTAINER_NAME: el nombre que se asignará a este nuevo contenedor del método de instalación simplificado de AlloyDB Omni en el registro de contenedores de tu máquina host. Por ejemplo, my-omni-1.

    • NEW_PASSWORD: la contraseña asignada al usuario postgres del nuevo contenedor después de su creación.

    • HOST_PORT: el puerto TCP de la máquina host al que debe publicar el contenedor su propio puerto 5432. Para usar el puerto predeterminado de PostgreSQL en la máquina host, especifica 5432.

    • FULL_KEY_PATH: la ruta completa del sistema de archivos al archivo de clave privada de la cuenta de servicio en tu sistema host.

Verificar la instalación de AlloyDB Omni con AlloyDB AI

Para verificar que la instalación se ha realizado correctamente y que usa la predicción del modelo, introduce lo siguiente:

   CREATE EXTENSION google_ml_integration CASCADE;

   SELECT array_dims(embedding( 'textembedding-gecko@001', 'AlloyDB AI')::real[]);
   array_dims

El resultado es similar al siguiente:

    [1:768]
    (1 row)

En la consulta anterior, la llamada embedding() genera incrustaciones para el texto de entrada AlloyDB AI. array_dims devuelve las dimensiones de la matriz devuelta por embedding(). Como el modelo textembedding-gecko prerregistrado devuelve una salida con 768 dimensiones, la salida es [768].

Siguientes pasos