安装 AlloyDB Omni with AlloyDB AI

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

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

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

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

验证安装了 AlloyDB AI 的 AlloyDB Omni

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

   CREATE EXTENSION google_ml_integration CASCADE;

   SELECT array_dims(embedding( 'textembedding-gecko@001', 'AlloyDB AI')::real[]);
   array_dims

输出类似于以下内容：

    [1:768]
    (1 row)

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

后续步骤