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

选择文档版本：

本页面介绍了预览版，您可以使用该预览版试验注册 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');