存储向量嵌入

本页面介绍了如何通过包含 pgvector 函数和运算符的 vector 扩展程序将 AlloyDB 用作向量数据库。借助这些函数和运算符，您可以将嵌入存储为向量值。

必需的数据库扩展程序

使用 vector 扩展程序（版本 0.5.0.google-1 或更高版本，其中包含 pgvector 函数和运算符）可将生成的嵌入存储为 vector 值。这是 Google 通过特定于 AlloyDB 的优化进行扩展的 pgvector 版本。

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 扩展程序创建索引，从而提高查询性能。

后续步骤