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

选择文档版本：

本页面介绍了如何使用已注册的模型端点生成嵌入。

准备工作

确保您已使用模型端点管理注册模型端点。如需了解详情，请参阅在 AlloyDB Omni 中注册和调用远程 AI 模型。

生成嵌入

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

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

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

替换以下内容：

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

示例

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

具有内置支持的文本嵌入模型

如需为已注册的 text-embedding-005 模型端点生成预测，请运行以下语句：

    SELECT
      google_ml.embedding(
        model_id => 'text-embedding-005',
        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');