Instala AlloyDB Omni con AlloyDB AI

Selecciona una versión de la documentación:

En esta página, se muestra cómo instalar AlloyDB Omni y cómo integrar AlloyDB AI.

AlloyDB AI es un paquete de funciones incluidas en AlloyDB Omni que te permiten crear aplicaciones de IA generativa empresarial. Para obtener más información sobre la funcionalidad de IA/ML de AlloyDB, consulta Compila aplicaciones de IA generativa.

AlloyDB Omni con AlloyDB AI te permite consultar modelos de AA remotos con la extensión google_ml_integration para trabajar con predicciones en línea y embeddings de texto generados a partir de modelos de AA. AlloyDB Omni con AlloyDB AI también puede procesar embeddings de vectores de otro contenido, como una imagen, por ejemplo, si usas la interfaz de google_ml.predict_row y realizas la traducción por tu cuenta en la búsqueda.

Configura Google Cloud para consultar modelos remotos

Si deseas consultar modelos de Vertex AI, debes configurar una cuenta de servicio de AlloyDB Omni con Vertex AI antes de instalar AlloyDB Omni.

Para configurar una cuenta de servicio para consultar modelos remotos, sigue estos pasos:

  1. Crea una cuenta de servicio conGoogle Cloud. En un paso posterior, le otorgarás a esta cuenta de servicio permisos para acceder a Vertex AI.

  2. Crea una clave de cuenta de servicio, guárdala en formato JSON en el archivo private-key.json y descárgala.

  3. Copia la clave de la cuenta de servicio que creaste en KEY_PATH. La ruta de acceso a la clave debe ser una ruta de acceso en tu host a la que pueda acceder el usuario que ejecutará AlloyDB Omni y que sea de su propiedad.

  4. Habilita la API de Vertex AI en tu Google Cloud proyecto.

    gcloud services enable aiplatform.googleapis.com

  5. Agrega permisos de Identity and Access Management (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"

    Reemplaza lo siguiente:

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

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

  1. Instala el paquete RPM de AlloyDB Omni.

Configura AlloyDB AI para AlloyDB Omni

Después de instalar el paquete RPM de AlloyDB Omni, completa los siguientes pasos para integrar AlloyDB AI:

  1. Completa todos los pasos que se indican en Configuración Google Cloud para consultar modelos remotos.

  2. Mueve la clave de la cuenta de servicio al directorio de datos de AlloyDB Omni.

    sudo mv KEY_PATH DATA_DIR/private-key.json

    Reemplaza las siguientes variables:

    • KEY_PATH: Es la ruta de acceso al archivo de claves de tu cuenta de servicio.
    • DATA_DIR: Ruta de acceso al directorio del host en el que se almacenan tus datos.

  3. Establece el propietario de la clave de la cuenta de servicio en postgres y sus permisos en 600.

    sudo chown postgres:postgres DATA_DIR/private-key.json
sudo chmod 600 DATA_DIR/private-key.json

  4. Actualiza la configuración de AlloyDB Omni agregando las siguientes opciones al archivo postgresql.conf:

    sudo tee -a DATA_DIR/postgresql.conf << EOF
omni_enable_ml_agent_process = 'on'
omni_google_cloud_private_key_file_path = 'DATA_DIR/private-key.json'
EOF

  5. Reinicia el servicio de AlloyDB Omni.

    sudo systemctl restart alloydbomni18

Conectar a través de psql

Para conectarte a la base de datos de AlloyDB Omni, ejecuta el siguiente comando:

sudo -u postgres /usr/lib/postgresql/18/bin/psql -U postgres

Verifica AlloyDB Omni con la instalación de AlloyDB AI

Para verificar que la instalación se realizó correctamente y que se usa la predicción del modelo, ingresa lo siguiente:

CREATE EXTENSION IF NOT EXISTS google_ml_integration CASCADE;

SELECT array_dims(embedding('text-embedding-005', 'AlloyDB AI')::real[]);

El resultado es similar al siguiente:

array_dims
------------
[1:768]
(1 row)

En la consulta anterior, la llamada a embedding() genera incorporaciones para el texto de entrada AlloyDB AI. array_dims devuelve las dimensiones del array que devuelve embedding(). Dado que el modelo text-embedding-005 devuelve un resultado con 768 dimensiones, el resultado es [768].

¿Qué sigue?