儲存向量嵌入

選取說明文件版本：

本頁說明如何搭配使用 vector 擴充功能 (內含 pgvector 函式和運算子)，將 AlloyDB 做為向量資料庫。這些函式和運算子可讓您將嵌入儲存為向量值。

必要的資料庫擴充功能

使用 vector 擴充功能 (版本 0.5.0.google-1 以上)，其中包含 pgvector 函式和運算子，將產生的嵌入內容儲存為 vector 值。這是 Google 擴充的 pgvector 版本，其中包含專為 AlloyDB 進行的最佳化。

CREATE EXTENSION IF NOT EXISTS vector;

儲存生成的嵌入

請確認您已在 AlloyDB 資料庫中建立資料表。

如要儲存向量嵌入內容，請按照下列步驟操作：

  1. 在資料表中建立 vector[] 欄，用來儲存嵌入內容：

    ALTER TABLE TABLE ADD COLUMN EMBEDDING_COLUMN vector(DIMENSIONS);

    更改下列內容：

    • TABLE：資料表名稱。

    • EMBEDDING_COLUMN：新嵌入資料欄的名稱。

    • DIMENSIONS：模型支援的維度數量。

      舉例來說，如果您使用其中一個text-embedding英文模型 (例如 text-embedding-005)，請使用 Vertex AI 指定 768。

  2. 將向量複製到向量資料欄。下列範例假設您的嵌入內容位於 CSV 檔案中：

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

    更改下列內容：

    • PATH_TO_VECTOR_CSV：儲存 CSV 檔案的完整路徑。

儲存嵌入後，您可以使用 vector 擴充功能或 alloydb_scann 擴充功能建立索引，加快查詢速度。

後續步驟