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 les fonctionnalités 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 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 les embeddings vectoriels d'autres contenus, comme une image, par exemple si vous utilisez l'interface google_ml.predict_row et que vous effectuez vous-même la traduction dans la requête.

Selon 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 distants

Pour configurer AlloyDB Omni afin qu'il interroge 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 et téléchargez-la.

  3. Stockez la clé dans un emplacement permanent de votre système de fichiers. Il 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, car vous en aurez besoin pour l'étape suivante.

  4. Ajoutez les 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 projet Google Cloud .

    • 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 décrites dans Configurer AlloyDB Omni pour interroger des modèles basés sur le 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 pour les 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 le système de fichiers de votre 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 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 simplifiée AlloyDB Omni 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 sur votre système hôte.

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

Pour vérifier que l'installation a réussi et qu'elle utilise la prédiction du modèle, saisissez la commande suivante :

   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].

Étapes suivantes