Integrar com a plataforma de agentes

Esta página detalha como configurar a integração entre o AlloyDB para PostgreSQL e a Gemini Enterprise Agent Platform, permitindo que você emita consultas que transmitem seus dados para modelos da Agent Platform.

Estas instruções são específicas para o uso do AlloyDB, e não do AlloyDB Omni. Para integrar uma instalação local do AlloyDB Omni usando a Agent Platform, consulte Instalar o AlloyDB Omni com o AlloyDB AI.

Para mais informações sobre como usar modelos de ML com o AlloyDB, consulte Criar aplicativos de IA generativa usando o AlloyDB AI.

Para mais informações sobre a Agent Platform, consulte Visão geral da Agent Platform.

Antes de começar

  1. Faça login na sua Google Cloud conta do. Se você não conhece o Google Cloud, crie uma conta para avaliar o desempenho dos nossos produtos em cenários reais. Clientes novos também recebem US $300 em créditos para executar, testar e implantar cargas de trabalho.

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

    Roles required to select or create a project

    • Select a project: Selecting a project doesn't require a specific IAM role—you can select any project that you've been granted a role on.
    • Create a project: To create a project, you need the Project Creator role (roles/resourcemanager.projectCreator), which contains the resourcemanager.projects.create permission. Learn how to grant roles.

    Go to project selector

  3. Verify that billing is enabled for your Google Cloud project.

  4. Enable the Agent Platform API.

    Roles required to enable APIs

    To enable APIs, you need the Service Usage Admin IAM role (roles/serviceusage.serviceUsageAdmin), which contains the serviceusage.services.enable permission. Learn how to grant roles.

    Enable the API

  5. Instale a Google Cloud CLI.

  6. Ao usar um provedor de identidade (IdP) externo, primeiro faça login na CLI gcloud com sua identidade federada.

  7. Para inicializar a CLI gcloud, execute o seguinte comando: 

    gcloud init

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

    Roles required to select or create a project

    • Select a project: Selecting a project doesn't require a specific IAM role—you can select any project that you've been granted a role on.
    • Create a project: To create a project, you need the Project Creator role (roles/resourcemanager.projectCreator), which contains the resourcemanager.projects.create permission. Learn how to grant roles.

    Go to project selector

  9. Verify that billing is enabled for your Google Cloud project.

  10. Enable the Agent Platform API.

    Roles required to enable APIs

    To enable APIs, you need the Service Usage Admin IAM role (roles/serviceusage.serviceUsageAdmin), which contains the serviceusage.services.enable permission. Learn how to grant roles.

    Enable the API

  11. Instale a Google Cloud CLI.

  12. Ao usar um provedor de identidade (IdP) externo, primeiro faça login na CLI gcloud com sua identidade federada.

  13. Para inicializar a CLI gcloud, execute o seguinte comando: 

    gcloud init

Sobre o agente de serviço do AlloyDB

Quando você usa o AlloyDB com outros Google Cloud serviços, ele usa um agente de serviço do Identity and Access Management (IAM) para autenticar esses serviços. Esse agente de serviço é um principal do IAM controlado pelo AlloyDB. Você concede papéis do IAM a esse agente de serviço para permitir que ele acesse outros Google Cloud serviços em seu nome.

Conceder o papel Vertex AI User ao agente de serviço do AlloyDB

Para ativar a integração do banco de dados com a Agent Platform, conceda o papel Vertex AI User ao agente de serviço do AlloyDB. Isso permite que o agente de serviço do AlloyDB acesse a Agent Platform em seu nome.

Siga estas etapas para conceder o papel Vertex AI User ao agente de serviço do AlloyDB:

Adicione permissões da Agent Platform ao agente de serviço do AlloyDB para o projeto em que o cluster do banco de dados do AlloyDB está localizado:

Console

  1. Acesse a página Welcome no Google Cloud console e copie o número do projeto que tem clusters ou instâncias do AlloyDB. Você vai usar esse número do projeto nas próximas etapas.

    Acessar a página de boas-vindas

  2. No Google Cloud console, acesse a página IAM.

    Acessar IAM

  3. Selecione o projeto que precisa chamar os endpoints da Agent Platform.

  4. Ative a opção Incluir atribuições de papel fornecidas pelo Google.

  5. Clique em Conceder acesso.

  6. No campo Novos principais, insira o seguinte:

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

    Substitua PROJECT_NUMBER pelo número do projeto.

  7. No campo Papel, insira Usuário da Vertex AI.

  8. Clique em Salvar.

gcloud

Para usar a CLI gcloud, você pode instalar e inicializar a Google Cloud CLI ou você pode usar o 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"

Substitua:

  • PROJECT_ID: o ID do projeto que tem o endpoint da Agent Platform.
  • PROJECT_NUMBER: o número do projeto que tem clusters ou instâncias do AlloyDB.

Verificar a extensão instalada

Verifique se a google_ml_integration está instalada no banco de dados que contém os dados em que você quer executar previsões:

Console

  1. No Google Cloud console, acesse a página Clusters.

    Acessar Clusters

  2. Para mostrar a página Visão geral do cluster, clique no nome do cluster do AlloyDB na coluna Nome do recurso.

  3. No menu de navegação, clique em AlloyDB Studio.

  4. Na página Fazer login no AlloyDB Studio , autentique-se usando o nome do banco de dados, o nome de usuário e a senha.

  5. Na guia Editor 1, faça o seguinte:

    1. Verifique se a versão 1.4.2 ou mais recente da extensão google_ml_integration está instalada:

      SELECT extversion FROM pg_extension WHERE extname = 'google_ml_integration';

    2. Clique em Executar. Aguarde a versão da extensão aparecer no painel Resultados.

psql

  1. Conecte um cliente psql à instância principal do cluster, conforme descrito em Conectar um cliente psql a uma instância.

  2. No prompt de comando psql, conecte-se ao banco de dados: 

    \c DB_NAME

    Substitua DB_NAME pelo nome do banco de dados em que você quer instalar a extensão.

  3. Verifique se a versão 1.4.2 ou mais recente da extensão google_ml_integration está instalada: 

    SELECT extversion FROM pg_extension WHERE extname = 'google_ml_integration';

A seguir