生成文本嵌入

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

准备工作

确保您已使用模型端点管理功能注册模型端点。如需了解详情，请参阅使用模型端点管理注册模型端点。

生成嵌入

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

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

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

替换以下内容：

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

示例

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

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

如需为已注册的 text-embedding-large-exp-03-07 模型端点生成嵌入，请运行以下语句：

    SELECT
      google_ml.embedding(
        model_id => 'text-embedding-large-exp-03-07',
        content => 'AlloyDB is a managed, cloud-hosted SQL database service');

如需为已注册的 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');

后续步骤