使用模型端点管理功能生成向量嵌入

本页介绍了一种预览版功能，可让您通过模型端点管理功能试用注册 AI 模型端点和调用预测功能。如需了解如何在生产环境中使用 AI 模型，请参阅使用 AlloyDB AI 构建生成式 AI 应用和使用向量嵌入。

在“模型端点管理”中添加并注册模型端点后，您可以使用模型 ID 引用这些端点来生成嵌入。

准备工作

确保您已在“模型端点管理”中注册模型端点。如需了解详情，请参阅使用模型端点管理服务注册模型端点

生成嵌入

使用 google_ml.embedding() SQL 函数调用具有文本嵌入模型类型的已注册模型端点，以生成嵌入。

如需调用模型并生成嵌入，请使用以下 SQL 查询：

SELECT
  google_ml.embedding(
    model_id => 'MODEL_ID',
    content => 'CONTENT');

替换以下内容：

MODEL_ID：您在注册模型端点时定义的模型 ID。
CONTENT：要转换为向量嵌入的文本。

示例

本部分列出了使用已注册的模型端点生成嵌入的部分示例。

内置支持的文本嵌入模型

如需为已注册的 textembedding-gecko@002 模型端点生成嵌入，请运行以下语句：

    SELECT
      google_ml.embedding(
        model_id => 'textembedding-gecko@002',
        content => 'AlloyDB is a managed, cloud-hosted SQL database service');

如需为 OpenAI 注册的 text-embedding-ada-002 模型端点生成嵌入，请运行以下语句：

    SELECT
      google_ml.embedding(
        model_id => 'text-embedding-ada-002',
        content => 'e-mail spam');

其他文本嵌入模型

如需为 OpenAI 注册的 text-embedding-3-small 或 text-embedding-3-large 模型端点生成嵌入，请运行以下语句：

  SELECT
    google_ml.embedding(
      model_id => 'text-embedding-3-small',
      content => 'Vector embeddings in AI');