Memorizzare i vector embedding

Questa pagina mostra come utilizzare AlloyDB 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.5.0.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.

CREATE EXTENSION IF NOT EXISTS vector;

Archivia gli embedding generati

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

Per archiviare gli embedding vettoriali:

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

    ALTER TABLE TABLE ADD COLUMN EMBEDDING_COLUMN vector(DIMENSIONS);

    Sostituisci quanto segue:

    • TABLE: 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 uno dei modelli in inglese text-embedding, ad esempio 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:

    • TABLE: il nome della tabella.

    • EMBEDDING_COLUMN: il nome della colonna di embedding.

    • 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