儲存向量嵌入

必要的資料庫擴充功能

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

CREATE EXTENSION IF NOT EXISTS vector;

儲存生成的嵌入

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

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

1. 在資料表中建立 vector 資料欄，用於儲存嵌入：

   ALTER TABLE TABLE_NAME ADD COLUMN EMBEDDING_COLUMN vector(DIMENSIONS);
   

   更改下列內容：

   • TABLE_NAME：資料表名稱。

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

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

     舉例來說，如果您在 Vertex AI 使用 text-embedding-005 等英文模型，請指定 768。

2. 將向量複製到向量資料欄。以下範例假設嵌入項目位於 CSV 檔案中：

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

   更改下列內容：

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

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

後續步驟

• 建立索引和查詢向量。
• 瞭解嵌入工作流程範例。