모델 엔드포인트 관리로 벡터 임베딩 생성

이 페이지에서는 AI 모델 엔드포인트 등록 및 모델 엔드포인트 관리로 예측 호출을 실험할 수 있는 미리보기를 설명합니다. 프로덕션 환경에서 AI 모델을 사용하는 방법은 AlloyDB AI를 사용하여 생성형 AI 애플리케이션 빌드벡터 임베딩 사용을 참고하세요.

모델 엔드포인트가 모델 엔드포인트 관리에 추가되고 등록된 후에는 모델 ID를 사용하여 엔드포인트를 참조하여 임베딩을 생성할 수 있습니다.

시작하기 전에

모델 엔드포인트 관리에 모델 엔드포인트를 등록했는지 확인합니다. 자세한 내용은 모델 엔드포인트 관리로 모델 엔드포인트 등록을 참고하세요.

임베딩 생성

google_ml.embedding() SQL 함수를 사용하여 텍스트 임베딩 모델 유형으로 등록된 모델 엔드포인트를 호출하여 임베딩을 생성합니다.

모델을 호출하고 임베딩을 생성하려면 다음 SQL 쿼리를 사용하세요.

SELECT
  google_ml.embedding(
    model_id => 'MODEL_ID',
    content => 'CONTENT');

다음을 바꿉니다.

  • MODEL_ID: 모델 엔드포인트를 등록할 때 정의한 모델 ID입니다.
  • CONTENT: 벡터 임베딩으로 변환할 텍스트입니다.

예시

등록된 모델 엔드포인트를 사용하여 임베딩을 생성하는 몇 가지 예는 이 섹션에 나와 있습니다.

내장 지원이 있는 텍스트 임베딩 모델

등록된 textembedding-gecko@002 모델 엔드포인트의 임베딩을 생성하려면 다음 문을 실행합니다.

    SELECT
      google_ml.embedding(
        model_id => 'textembedding-gecko@002',
        content => 'AlloyDB is a managed, cloud-hosted SQL database service');

OpenAI에서 등록된 text-embedding-ada-002 모델 엔드포인트의 임베딩을 생성하려면 다음 문을 실행합니다.

    SELECT
      google_ml.embedding(
        model_id => 'text-embedding-ada-002',
        content => 'e-mail spam');

기타 텍스트 임베딩 모델

OpenAI에서 등록된 text-embedding-3-small 또는 text-embedding-3-large 모델 엔드포인트의 임베딩을 생성하려면 다음 문을 실행합니다.

  SELECT
    google_ml.embedding(
      model_id => 'text-embedding-3-small',
      content => 'Vector embeddings in AI');