Instale o AlloyDB Omni com o AlloyDB AI

Selecione uma versão da documentação:

Esta página mostra como instalar o AlloyDB Omni e integrar o AlloyDB AI no mesmo.

O AlloyDB AI é um conjunto de funcionalidades incluídas no AlloyDB Omni que lhe permitem criar aplicações de IA generativa empresariais. Para mais informações sobre a funcionalidade de ML do AlloyDB, consulte o artigo Crie aplicações de IA generativa.

O AlloyDB Omni com o AlloyDB AI permite-lhe consultar modelos de AA remotos para trabalhar com previsões online e incorporações de texto geradas a partir de modelos de AA. O AlloyDB Omni com o AlloyDB AI também pode processar incorporações de vetores de outro conteúdo, como uma imagem, por exemplo, se usar a interface google_ml.predict_row e fizer a tradução na consulta.

Com base no local onde quer instalar o AlloyDB Omni com o AlloyDB AI, selecione uma das seguintes opções:

Configure a sua instância do AlloyDB Omni para consultar modelos remotos

Para configurar o AlloyDB Omni para consultar modelos remotos, siga estes passos:

  1. Crie uma conta de serviço com Google Cloud.

  2. Crie uma chave de conta de serviço no formato JSON e transfira-a.

  3. Armazene a chave numa localização permanente no seu sistema de ficheiros. Reside nesta localização durante a duração do seu servidor AlloyDB Omni.

    Tome nota da respetiva localização no sistema de ficheiros. Vai precisar dela para o passo seguinte.

  4. Adicione autorizações da gestão de identidade e de acesso (IAM) do Vertex AI ao projeto e à conta de serviço adequados.

       gcloud projects add-iam-policy-binding PROJECT_ID \
       --member="serviceAccount:SERVICE_ACCOUNT_ID" \
       --role="roles/aiplatform.user"

    Substitua o seguinte:

    • PROJECT_ID: o ID do seu projeto Google Cloud .

    • SERVICE_ACCOUNT_ID: o ID da conta de serviço que criou no passo anterior. Isto inclui o sufixo @PROJECT_ID.iam.gserviceaccount.com completo, por exemplo, my-service@my-project.iam.gserviceaccount.com.

Instale o AlloyDB Omni com o AlloyDB AI

Para instalar o AlloyDB Omni e integrar o AlloyDB AI, siga estes passos:

  1. Certifique-se de que concluiu todos os passos indicados no artigo Configure o AlloyDB Omni para consultar modelos baseados na nuvem.

  2. Crie uma cópia local do ficheiro de configuração predefinido do AlloyDB Omni:

       docker run -i --rm google/alloydbomni:15.5.4 cat /usr/share/postgresql/postgresql.conf.sample > my-postgres.conf

  3. Edite a cópia do ficheiro de configuração para adicionar diretivas para a integração da IA do AlloyDB:

       echo "omni_enable_ml_agent_process = 'on'" >> my-postgres.conf
   echo "omni_google_cloud_private_key_file_path = '/etc/postgresql/private-key.json'" >> my-postgres.conf

  4. Ajuste as autorizações do sistema de ficheiros dos ficheiros de configuração e de chaves:

       sudo chmod +r my-postgres.conf
   sudo chmod +r KEY_PATH

    Substitua KEY_PATH pelo caminho para o ficheiro de chave privada da conta de serviço no sistema de ficheiros do anfitrião. Esta é a chave que criou e transferiu anteriormente neste procedimento.

  5. Instale os ficheiros de configuração e de chave montados no contentor:

       docker run --name CONTAINER_NAME -e POSTGRES_PASSWORD=NEW_PASSWORD -p HOST_PORT:5432 -v "$PWD/my-postgres.conf":/etc/postgresql/postgresql.conf -v "FULL_KEY_PATH":/etc/postgresql/private-key.json -d google/alloydbomni:15.5.4 -c 'config_file=/etc/postgresql/postgresql.conf'

    Substitua o seguinte:

    • CONTAINER_NAME: o nome a atribuir a este novo contentor do método de instalação simplificado do AlloyDB Omni no registo de contentores da sua máquina anfitriã, por exemplo, my-omni-1.

    • NEW_PASSWORD: a palavra-passe atribuída ao utilizador postgres do novo contentor após a respetiva criação.

    • HOST_PORT: a porta TCP na máquina anfitriã para a qual o contentor deve publicar a sua própria porta 5432. Para usar a porta predefinida do PostgreSQL também na máquina anfitriã, especifique 5432.

    • FULL_KEY_PATH: o caminho completo do sistema de ficheiros para o ficheiro de chave privada da conta de serviço no seu sistema anfitrião.

Valide o AlloyDB Omni com a instalação do AlloyDB AI

Para verificar se a instalação foi bem-sucedida e se usa a previsão do modelo, introduza o seguinte:

   CREATE EXTENSION google_ml_integration CASCADE;

   SELECT array_dims(embedding( 'textembedding-gecko@001', 'AlloyDB AI')::real[]);
   array_dims

O resultado tem um aspeto semelhante ao seguinte:

    [1:768]
    (1 row)

Na consulta anterior, a chamada embedding() gera incorporações para o texto de entrada AlloyDB AI. array_dims devolve as dimensões da matriz devolvida por embedding(). Uma vez que o modelo textembedding-gecko pré-registado devolve um resultado com 768 dimensões, o resultado é [768].

O que se segue?