在 AlloyDB Omni 中安装 AlloyDB AI

本页介绍了如何安装 AlloyDB Omni 并集成 AlloyDB AI。

AlloyDB AI 是 AlloyDB Omni 附带的一套功能，可让您构建企业生成式 AI 应用。如需详细了解 AlloyDB 的 AI/ML 功能，请参阅构建生成式 AI 应用。

借助 AlloyDB Omni 和 AlloyDB AI，您可以查询远程机器学习模型，以使用机器学习模型生成的在线预测和文本嵌入。结合使用 AlloyDB Omni 和 AlloyDB AI 还可以处理其他内容（例如图片）中的向量嵌入，例如，如果您使用 google_ml.predict_row 接口并在查询中自行进行转换。

根据您要将 AlloyDB Omni with AlloyDB AI 安装在何处，选择以下选项之一：

验证安装了 AlloyDB AI 的 AlloyDB Omni

如需验证安装是否成功并使用模型预测，请输入以下内容：

CREATE EXTENSION IF NOT EXISTS google_ml_integration CASCADE;

SELECT array_dims(embedding('text-embedding-005', 'AlloyDB AI')::real[]);

输出类似于以下内容：

array_dims
------------
[1:768]
(1 row)

在上一个查询中，embedding() 调用会为输入文本 AlloyDB AI 生成嵌入。array_dims 会返回 embedding() 返回的数组的维度。由于 text-embedding-005 模型会返回一个具有 768 个维度的输出，因此输出为 [768]。

后续步骤