Installer AlloyDB Omni avec AlloyDB AI

Sélectionnez une version de la documentation :

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

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

AlloyDB Omni avec AlloyDB AI vous permet d'interroger des modèles de ML à distance pour utiliser des prédictions en ligne et des embeddings textuels générés à partir de modèles de ML. AlloyDB Omni avec AlloyDB AI peut également traiter des embeddings vectoriels à partir d'autres contenus, comme 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'endroit où vous souhaitez installer AlloyDB Omni avec AlloyDB AI, sélectionnez l'une des options suivantes :

Configurer votre instance AlloyDB Omni pour interroger des modèles à distance

Pour configurer AlloyDB Omni afin d'interroger des modèles à distance, 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. Elle réside à 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 des autorisations Vertex AI Identity and Access Management (IAM) 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 complet @PROJECT_ID.iam.gserviceaccount.com, 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 effectué toutes les étapes décrites dans 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 d'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. 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 les fichiers de clé et de configuration monté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 PostgreSQL par défaut 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 dans 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 de modèle, saisissez les éléments suivants :

   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