本页面介绍了如何安装 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]
。