Memorizzare i vector embedding

L'estensione vector in AlloyDB per PostgreSQL fornisce funzioni e operatori pgvector che ti consentono di utilizzare il tuo database come archivio vettoriale. Puoi aggiungere colonne vector alle tue tabelle e archiviare gli incorporamenti generati come valori vettoriali per i workload di ricerca semantica e AI generativa.

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 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 tuo database AlloyDB.

Per archiviare i vector embedding, segui questi passaggi:

  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 di inglese text-embedding, ad esempio text-embedding-005 con Agent Platform, specifica 768.

  2. Copia i vettori nella colonna dei vettori. L'esempio seguente 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