Installa AlloyDB Omni con AlloyDB AI

Seleziona una versione della documentazione:

Questa pagina mostra come installare AlloyDB Omni e integrarlo AlloyDB AI.

AlloyDB AI è una suite di funzionalità incluse in AlloyDB Omni che ti consentono di creare applicazioni aziendali di AI generativa. Per ulteriori informazioni sulla funzionalità di ML di AlloyDB, vedi Crea applicazioni di AI generativa.

AlloyDB Omni con AlloyDB AI ti consente di eseguire query sui modelli di ML remoti per lavorare con le previsioni online e gli incorporamenti di testo generati dai modelli di ML. AlloyDB Omni con AlloyDB AI può anche elaborare vector embedding da altri contenuti, ad esempio un'immagine, se utilizzi l'interfaccia google_ml.predict_row ed esegui la traduzione direttamente nella query.

In base a dove vuoi installare AlloyDB Omni con AlloyDB AI, seleziona una delle seguenti opzioni:

Configura l'istanza AlloyDB Omni per eseguire query sui modelli remoti

Per configurare AlloyDB Omni per eseguire query sui modelli remoti:

  1. Crea un service account con Google Cloud.

  2. Crea una chiave dell'account di servizio in formato JSON e scaricala.

  3. Archivia la chiave in una posizione permanente del file system. Si trova in questa posizione per tutta la durata del server AlloyDB Omni.

    Prendi nota della sua posizione nel file system, perché ti servirà per il passaggio successivo.

  4. Aggiungi le autorizzazioni IAM (Identity and Access Management) di Vertex AI al progetto e al account di servizio appropriati.

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

    Sostituisci quanto segue:

    • PROJECT_ID: l'ID del tuo Google Cloud progetto.

    • SERVICE_ACCOUNT_ID: l'ID del service account creato nel passaggio precedente. Ciò include il suffisso @PROJECT_ID.iam.gserviceaccount.com completo, ad esempio my-service@my-project.iam.gserviceaccount.com.

Installa AlloyDB Omni con AlloyDB AI

Per installare AlloyDB Omni e integrare AlloyDB AI:

  1. Assicurati di aver completato tutti i passaggi elencati in Configurare AlloyDB Omni per eseguire query sui modelli basati su cloud.

  2. Crea una copia locale del file di configurazione predefinito di AlloyDB Omni:

       docker run -i --rm google/alloydbomni:15.5.4 cat /usr/share/postgresql/postgresql.conf.sample > my-postgres.conf
  3. Modifica la copia del file di configurazione per aggiungere direttive per l'integrazione di 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. Modifica le autorizzazioni del file system dei file di configurazione e delle chiavi:

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

    Sostituisci KEY_PATH con il percorso del file della chiave privata dell'account di servizio nel file system host. Questa è la chiave che hai creato e scaricato in precedenza in questa procedura.

  5. Installa i file di configurazione e della chiave montati sul container:

       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'

    Sostituisci quanto segue:

    • CONTAINER_NAME: il nome da assegnare a questo nuovo contenitore del metodo di installazione semplificato di AlloyDB Omni nel registro dei contenitori della macchina host, ad esempio my-omni-1.

    • NEW_PASSWORD: la password assegnata all'utente postgres del nuovo contenitore dopo la sua creazione.

    • HOST_PORT: la porta TCP sulla macchina host a cui il container deve pubblicare la propria porta 5432. Per utilizzare anche la porta predefinita di PostgreSQL sulla macchina host, specifica 5432.

    • FULL_KEY_PATH: il percorso completo del file system del file della chiave privata del account di servizio sul sistema host.

Verifica l'installazione di AlloyDB Omni con AlloyDB AI

Per verificare che l'installazione sia andata a buon fine e utilizzi la previsione del modello, inserisci quanto segue:

   CREATE EXTENSION google_ml_integration CASCADE;

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

L'output è simile al seguente:

    [1:768]
    (1 row)

Nella query precedente, la chiamata embedding() genera incorporamenti per il testo di input AlloyDB AI. array_dims restituisce le dimensioni dell'array restituito da embedding(). Poiché il modello textembedding-gecko preregistrato restituisce un output con 768 dimensioni, l'output è [768].

Passaggi successivi