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

本页介绍了如何注册 AI 模型端点以及如何使用模型端点管理生成矢量嵌入。如需在生产环境中使用 AI 模型,请参阅使用 Cloud SQL 构建生成式 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 => 'Cloud SQL is a managed, cloud-hosted SQL database service');