AlloyDB AI와 함께 AlloyDB Omni 설치

문서 버전을 선택합니다.

이 페이지에서는 AlloyDB Omni를 설치하고 AlloyDB AI를 통합하는 방법을 보여줍니다.

AlloyDB AI는 엔터프라이즈 생성형 AI 애플리케이션을 빌드할 수 있는 AlloyDB Omni에 포함된 기능 모음입니다. AlloyDB의 AI/ML 기능에 관한 자세한 내용은 생성형 AI 애플리케이션 빌드를 참고하세요.

AlloyDB AI가 포함된 AlloyDB Omni를 사용하면 google_ml_integration 확장 프로그램을 사용하여 원격 ML 모델을 쿼리하여 ML 모델에서 생성된 온라인 예측 및 텍스트 임베딩을 사용할 수 있습니다. AlloyDB AI가 포함된 AlloyDB Omni는 이미지와 같은 다른 콘텐츠의 벡터 임베딩도 처리할 수 있습니다(예: google_ml.predict_row 인터페이스를 사용하고 쿼리에서 직접 번역하는 경우).

원격 모델을 쿼리하도록 Google Cloud 설정

Vertex AI 모델을 쿼리하려면 AlloyDB Omni를 설치하기 전에 Vertex AI로 AlloyDB Omni 서비스 계정을 구성해야 합니다.

원격 모델을 쿼리할 서비스 계정을 설정하려면 다음 단계를 따르세요.

  1. Google Cloud사용을 통해 서비스 계정을 만듭니다. 이 서비스 계정에는 이후 단계에서 Vertex AI에 액세스할 수 있는 권한이 부여됩니다.

  2. 서비스 계정 키를 만들고 JSON 형식으로 private-key.json 파일에 저장한 후 다운로드합니다.

  3. 만든 서비스 계정 키를 KEY_PATH에 복사합니다. 키 경로는 AlloyDB Omni를 실행할 사용자가 액세스하고 소유할 수 있는 호스트의 경로여야 합니다.

  4. Google Cloud 프로젝트에서 Vertex AI API를 사용 설정합니다.

    gcloud services enable aiplatform.googleapis.com

  5. 적절한 프로젝트 및 서비스 계정에 Vertex AI Identity and Access Management(IAM) 권한을 추가합니다.

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

    다음을 바꿉니다.

    • PROJECT_ID: Google Cloud 프로젝트의 ID입니다.

    • SERVICE_ACCOUNT_ID: 1단계에서 만든 서비스 계정의 ID입니다. 여기에는 전체 @PROJECT_ID.iam.gserviceaccount.com 접미사가 포함됩니다. 예를 들면 my-service@my-project.iam.gserviceaccount.com입니다.

  1. AlloyDB Omni RPM 패키지를 설치합니다.

AlloyDB Omni용 AlloyDB AI 구성

AlloyDB Omni RPM 패키지를 설치한 후 다음 단계를 완료하여 AlloyDB AI를 통합합니다.

  1. 원격 모델을 쿼리하도록 설정 Google Cloud 에 나열된 모든 단계를 완료합니다.

  2. 서비스 계정 키를 AlloyDB Omni의 데이터 디렉터리로 이동합니다.

    sudo mv KEY_PATH DATA_DIR/private-key.json

    다음 변수를 바꿉니다.

    • KEY_PATH: 서비스 계정 키 파일의 경로입니다.
    • DATA_DIR: 데이터가 저장된 호스트 디렉터리 경로

  3. 서비스 계정 키의 소유자를 postgres로, 권한을 600로 설정합니다.

    sudo chown postgres:postgres DATA_DIR/private-key.json
sudo chmod 600 DATA_DIR/private-key.json

  4. postgresql.conf 파일에 다음 옵션을 추가하여 AlloyDB Omni 구성을 업데이트합니다.

    sudo tee -a DATA_DIR/postgresql.conf << EOF
omni_enable_ml_agent_process = 'on'
omni_google_cloud_private_key_file_path = 'DATA_DIR/private-key.json'
EOF

  5. AlloyDB Omni 서비스를 다시 시작합니다.

    sudo systemctl restart alloydbomni18

psql를 사용하여 연결

AlloyDB Omni 데이터베이스에 연결하려면 다음 명령어를 실행합니다.

sudo -u postgres /usr/lib/postgresql/18/bin/psql -U postgres

AlloyDB AI 설치로 AlloyDB Omni 확인

설치가 완료되었으며 모델 예측을 사용하는지 확인하려면 다음을 입력합니다.

CREATE EXTENSION IF NOT EXISTS google_ml_integration CASCADE;

SELECT array_dims(embedding('text-embedding-005', 'AlloyDB AI')::real[]);

결과는 다음과 유사합니다.

array_dims
------------
[1:768]
(1 row)

이전 쿼리에서 embedding() 호출은 입력 텍스트 AlloyDB AI의 임베딩을 생성합니다. array_dims는 embedding()에서 반환된 배열의 크기를 반환합니다. text-embedding-005 모델은 768개의 측정기준이 있는 출력을 반환하므로 출력은 [768]입니다.

다음 단계