Installer AlloyDB Omni avec AlloyDB AI

Cette page vous explique comment installer AlloyDB Omni et y intégrer AlloyDB AI.

AlloyDB AI est une suite de fonctionnalités incluses avec AlloyDB Omni qui vous permet de créer des applications d'IA générative d'entreprise. Pour en savoir plus sur les fonctionnalités de ML d'AlloyDB, consultez la page Créer des applications d'IA générative.

AlloyDB Omni avec AlloyDB AI vous permet d'interroger des modèles de ML distants pour travailler avec des prédictions en ligne et des embeddings de texte générés à partir de modèles de ML. AlloyDB Omni avec AlloyDB AI peut également traiter des représentations vectorielles continues à partir d'autres contenus, tels qu'une image, par exemple, si vous utilisez l'interface google_ml.predict_row et effectuez vous-même la traduction dans la requête.

En fonction de l'emplacement où vous souhaitez installer AlloyDB Omni avec l'IA AlloyDB, sélectionnez l'une des options suivantes:

Configurer votre instance AlloyDB Omni pour interroger des modèles distants

Pour configurer AlloyDB Omni afin d'interroger des modèles distants, procédez comme suit:

  1. Créez un compte de service avec Google Cloud.

  2. Créez une clé de compte de service au format JSON, puis téléchargez-la.

  3. Stockez la clé dans un emplacement permanent de votre système de fichiers. Il se trouve à cet emplacement pendant toute la durée de vie de votre serveur AlloyDB Omni.

    Notez son emplacement dans votre système de fichiers. Vous en aurez besoin pour l'étape suivante.

  4. Ajoutez les autorisations IAM (Identity and Access Management) Vertex AI au projet et au compte de service appropriés.

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

    Remplacez les éléments suivants :

    • PROJECT_ID: ID de votre Google Cloud projet.

    • SERVICE_ACCOUNT_ID: ID du compte de service que vous avez créé à l'étape précédente. Cela inclut le suffixe @PROJECT_ID.iam.gserviceaccount.com complet (par exemple, my-service@my-project.iam.gserviceaccount.com).

Installer AlloyDB Omni avec AlloyDB AI

Pour installer AlloyDB Omni et intégrer AlloyDB AI, procédez comme suit:

  1. Assurez-vous d'avoir suivi toutes les étapes de la section Configurer AlloyDB Omni pour interroger des modèles cloud.

  2. Créez une copie locale du fichier de configuration AlloyDB Omni par défaut:

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

  3. Modifiez la copie du fichier de configuration pour ajouter des directives d'intégration de l'IA AlloyDB:

       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. Ajustez les autorisations du système de fichiers des fichiers de clé et de configuration:

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

    Remplacez KEY_PATH par le chemin d'accès au fichier de clé privée du compte de service dans votre système de fichiers hôte. Il s'agit de la clé que vous avez créée et téléchargée précédemment dans cette procédure.

  5. Installez la clé et les fichiers de configuration installés sur le conteneur:

       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'

    Remplacez les éléments suivants :

    • CONTAINER_NAME: nom à attribuer à ce nouveau conteneur de méthode d'installation AlloyDB Omni simplifiée dans le registre de conteneurs de votre machine hôte (par exemple, my-omni-1).

    • NEW_PASSWORD: mot de passe attribué à l'utilisateur postgres du nouveau conteneur après sa création.

    • HOST_PORT: port TCP sur la machine hôte sur lequel le conteneur doit publier son propre port 5432. Pour utiliser également le port par défaut PostgreSQL sur la machine hôte, spécifiez 5432.

    • FULL_KEY_PATH: chemin d'accès complet au fichier de clé privée du compte de service sur votre système hôte.

Vérifier l'installation d'AlloyDB Omni avec AlloyDB AI

Pour vérifier que votre installation a réussi et qu'elle utilise la prédiction du modèle, saisissez ce qui suit:

   CREATE EXTENSION google_ml_integration CASCADE;

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

La sortie ressemble à ceci :

    [1:768]
    (1 row)

Dans la requête précédente, l'appel embedding() génère des embeddings pour le texte d'entrée AlloyDB AI. array_dims renvoie les dimensions du tableau renvoyé par embedding(). Étant donné que le modèle textembedding-gecko préenregistré renvoie une sortie avec 768 dimensions, la sortie est [768].

Étape suivante