在 AlloyDB Omni 中安装 AlloyDB AI

选择文档版本：

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

AlloyDB AI 是 AlloyDB Omni 中包含的一套功能，可让您构建企业生成式 AI 应用。如需详细了解 AlloyDB 的 AI/机器学习功能，请参阅构建生成式 AI 应用。

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

根据您要在其中安装具有 AlloyDB AI 的 AlloyDB Omni 的位置，选择以下选项之一：

验证具有 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]。

后续步骤