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

本页面介绍了如何使用模型端点管理注册 AI 模型端点并生成向量嵌入。 如需在生产环境中使用 AI 模型，请参阅生成和管理向量嵌入。

在模型端点管理中添加并注册模型端点后，您可以使用模型 ID 引用它们以生成嵌入。

准备工作

请确保完成以下操作：

• 使用模型端点管理注册模型端点。如需了解详情，请参阅使用模型端点管理注册和调用远程 AI 模型。
• 创建或更新 Cloud SQL 实例，使该实例可以与 Vertex AI 集成。如需了解详情，请参阅启用数据库与 Vertex AI 的集成。

生成嵌入

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

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

SELECT
  mysql.ml_embedding(
    'MODEL_ID',
    'CONTENT');

替换以下内容：

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

示例

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

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

默认情况下支持 Vertex AI 嵌入模型。您无需注册这些设备。如需为已注册的 gemini-embedding-001 模型端点生成嵌入，请运行以下语句：

    SELECT
      mysql.ml_embedding(
        'gemini-embedding-001',
        'Cloud SQL is a managed, cloud-hosted SQL database service');