Memorizzare i vector embedding

Seleziona una versione della documentazione:

Questa pagina mostra come utilizzare AlloyDB Omni come database vettoriale con l'estensione vector che include pgvector funzioni e operatori. Queste funzioni e questi operatori consentono di archiviare gli embedding come valori vettoriali.

Estensione del database richiesta

Utilizza l'estensione vector, versione 0.8.2.google-1 o successive, che include pgvector funzioni e operatori, per archiviare gli embedding generati come valori vector. Si tratta di una versione di pgvector che Google ha esteso con ottimizzazioni specifiche per AlloyDB Omni.

CREATE EXTENSION IF NOT EXISTS vector;

Archivia gli embedding generati

Assicurati di aver già creato una tabella nel database AlloyDB Omni.

Per archiviare gli embedding vettoriali:

  1. Crea una colonna vector[] nella tabella per archiviare gli embedding:

    ALTER TABLE TABLE_NAME ADD COLUMN EMBEDDING_COLUMN vector(DIMENSIONS);

    Sostituisci quanto segue:

    • TABLE_NAME: il nome della tabella.

    • EMBEDDING_COLUMN: il nome della nuova colonna di embedding.

    • DIMENSIONS: il numero di dimensioni supportate dal modello.

      Ad esempio, se utilizzi un modello in inglese come text-embedding-005 con Vertex AI, specifica 768.

  2. Copia i vettori nella colonna vettoriale. Il seguente esempio presuppone che gli embedding siano disponibili in un file CSV:

    COPY TABLE (EMBEDDING_COLUMN) FROM 'PATH_TO_VECTOR_CSV' (FORMAT CSV);

    Sostituisci quanto segue:

    • PATH_TO_VECTOR_CSV: il percorso completo in cui hai archiviato il file CSV.

Dopo aver archiviato gli embedding, puoi utilizzare l'estensione vector o l'estensione alloydb_scann per creare indici per prestazioni di query più rapide.

Passaggi successivi