Memorizzare i vector embedding

Seleziona una versione della documentazione:

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

Estensione del database richiesta

Utilizza l'estensione vector, versione 0.5.0.google-1 o successive, che include funzioni e operatori pgvector, per archiviare gli incorporamenti generati come valori vector. Questa è una versione di pgvector che Google ha esteso con ottimizzazioni specifiche per AlloyDB.

CREATE EXTENSION IF NOT EXISTS vector;

Incorporamenti generati dallo Store

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

Per memorizzare gli incorporamenti 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 incorporamento.

    • DIMENSIONS: il numero di dimensioni supportate dal modello.

      Ad esempio, se utilizzi uno dei text-embeddingmodelli in inglese, ad esempio text-embedding-005 con Vertex AI, specifica 768.

  2. Copia i vettori nella colonna dei vettori. Il seguente esempio presuppone che i tuoi incorporamenti 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 memorizzato il file CSV.

Dopo aver memorizzato gli incorporamenti, puoi utilizzare l'estensione vector o l'estensione alloydb_scann per creare indici per prestazioni delle query più rapide.

Passaggi successivi