与 Vertex AI 集成

本页详细介绍了如何设置 AlloyDB for PostgreSQL 与 Vertex AI 之间的集成，以便您将对云端存储的大语言模型 (LLM) 的查询应用于您的数据。

以下说明仅适用于使用 AlloyDB，而非 AlloyDB Omni。如需改用 Vertex AI 集成本地安装的 AlloyDB Omni，请参阅使用 AlloyDB AI 安装 AlloyDB Omni。

如需详细了解如何将 ML 模型与 AlloyDB 搭配使用，请参阅使用 AlloyDB AI 构建生成式 AI 应用。

如需详细了解 Vertex AI，请参阅 Vertex AI 简介。

准备工作

  1. Sign in to your Google Cloud account. If you're new to Google Cloud, create an account to evaluate how our products perform in real-world scenarios. New customers also get $300 in free credits to run, test, and deploy workloads.

  2. In the Google Cloud console, on the project selector page, select or create a Google Cloud project.

    Go to project selector

  3. Make sure that billing is enabled for your Google Cloud project.

  4. Enable the Vertex AI API.

    Enable the API

  5. Install the Google Cloud CLI.

  6. To initialize the gcloud CLI, run the following command: 

    gcloud init

  7. In the Google Cloud console, on the project selector page, select or create a Google Cloud project.

    Go to project selector

  8. Make sure that billing is enabled for your Google Cloud project.

  9. Enable the Vertex AI API.

    Enable the API

  10. Install the Google Cloud CLI.

  11. To initialize the gcloud CLI, run the following command: 

    gcloud init

向 AlloyDB 服务代理授予 Vertex AI 用户权限

如需启用数据库与 Vertex AI 的集成，请按以下步骤授予 AlloyDB 服务代理访问 Vertex AI 的 Identity and Access Management (IAM) 权限：

为 AlloyDB 数据库集群所在项目的 AlloyDB 服务代理添加 Vertex AI 权限：

控制台

  1. 前往 Google Cloud 控制台的欢迎页面，然后复制包含 AlloyDB 集群或实例的项目的项目编号。您将在后续步骤中使用此项目编号。

    前往“欢迎”页面

  2. 在 Google Cloud 控制台中，前往 IAM 页面。

    进入 IAM

  3. 选择包含 Vertex AI 端点的项目。

  4. 启用包括 Google 提供的角色授权。

  5. 点击 授予访问权限。

  6. 在新的主账号字段中，输入以下内容：

        service-PROJECT_NUMBER@gcp-sa-alloydb.iam.gserviceaccount.com

    将 PROJECT_NUMBER 替换为项目编号。

  7. 在角色字段中，输入 Vertex AI User。

  8. 点击保存。

gcloud

  1. 如需使用 gcloud CLI，您可以安装并初始化 Google Cloud CLI，也可以使用 Cloud Shell。

    gcloud projects add-iam-policy-binding PROJECT_ID 
    
--member="serviceAccount:service-PROJECT_NUMBER@gcp-sa-alloydb.iam.gserviceaccount.com" 
    
--role="roles/aiplatform.user"
    替换以下内容：
  • PROJECT_ID：具有 Vertex AI 端点的项目的 ID。
  • PROJECT_NUMBER：包含 AlloyDB 集群或实例的项目的项目编号。

验证已安装的扩展程序

验证 google_ml_integration 是否已安装在包含您要根据其运行预测的数据的数据库中：

  1. 按照将 psql 客户端连接到实例中所述，将 psql 客户端连接到集群的主实例。

  2. 在 psql 命令提示符下，连接到数据库： 

    \c DB_NAME

    将 DB_NAME 替换为您要在其中安装扩展程序的数据库的名称。

  3. 验证是否已安装 google_ml_integration 扩展程序 1.4.2 或更高版本： 

    SELECT extversion FROM pg_extension WHERE extname = 'google_ml_integration';

  4. 可选。手动安装最新版本的扩展程序： 

    CREATE EXTENSION google_ml_integration VERSION '1.4.2';

  5. 可选。如果您使用的是低于 1.4.2 的扩展程序，请将其更新到最新版本： 

    ALTER EXTENSION google_ml_integration UPDATE TO '1.4.2'

后续步骤