Installa AlloyDB Omni con AlloyDB AI

Seleziona una versione della documentazione:

Questa pagina mostra come installare AlloyDB Omni e integrare 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à AI/ML di AlloyDB, consulta Creare applicazioni di AI generativa.

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

Configurare l'esecuzione di query sui modelli remoti Google Cloud

Se vuoi eseguire query sui modelli Vertex AI, devi configurare un account di servizio AlloyDB Omni con Vertex AI prima di installare AlloyDB Omni.

Per configurare un account di servizio per eseguire query sui modelli remoti, segui questi passaggi:

  1. Crea un account di servizio con Google Cloud. In un passaggio successivo, concedi a questo account di servizio le autorizzazioni per accedere a Vertex AI.

  2. Crea una chiave dell'account di servizio e salvala in formato JSON nel file private-key.json e scaricala.

  3. Copia la chiave del account di servizio che hai creato in KEY_PATH. Il percorso della chiave deve essere un percorso sull'host accessibile e di proprietà dell'utente che eseguirà AlloyDB Omni.

  4. Abilita l'API Vertex AI nel tuo Google Cloud progetto.

    gcloud services enable aiplatform.googleapis.com

  5. 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: ID del tuo Google Cloud progetto.

    • SERVICE_ACCOUNT_ID: ID del account di servizio che hai creato nel passaggio 1. Include il suffisso completo @PROJECT_ID.iam.gserviceaccount.com. Ad esempio, my-service@my-project.iam.gserviceaccount.com.

  1. Installa il pacchetto RPM di AlloyDB Omni.

Configurare AlloyDB AI per AlloyDB Omni

Dopo aver installato il pacchetto RPM di AlloyDB Omni, completa i seguenti passaggi per integrare AlloyDB AI:

  1. Completa tutti i passaggi elencati in Configurare Google Cloud l'esecuzione di query sui modelli remoti.

  2. Sposta la chiave del account di servizio nella directory dei dati di AlloyDB Omni.

    sudo mv KEY_PATH DATA_DIR/private-key.json

    Sostituisci le seguenti variabili:

    • KEY_PATH: percorso del file della chiave del account di servizio.
    • DATA_DIR: percorso della directory host in cui sono archiviati i dati.

  3. Imposta il proprietario della chiave del account di servizio su postgres e le relative autorizzazioni su 600.

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

  4. Aggiorna la configurazione di AlloyDB Omni aggiungendo le seguenti opzioni al file 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. Riavvia il servizio AlloyDB Omni.

    sudo systemctl restart alloydbomni18

Connettersi utilizzando psql

Per connetterti al database AlloyDB Omni, esegui il seguente comando:

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

Verificare l'installazione di AlloyDB Omni con AlloyDB AI

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

CREATE EXTENSION IF NOT EXISTS google_ml_integration CASCADE;

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

L'output è simile al seguente:

array_dims
------------
[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 text-embedding-005 restituisce un output con 768 dimensioni, l'output è [768].

Passaggi successivi