이 페이지는 Cloud Translation API를 통해 번역되었습니다.

벡터 검색 수행

이 튜토리얼에서는 Google Cloud 콘솔을 사용하여 PostgreSQL용 AlloyDB에서 벡터 검색을 설정하고 실행하는 방법을 설명합니다. 벡터 검색 기능을 보여주기 위한 예시가 포함되어 있으며, 이는 데모용으로만 제공됩니다.

Vertex AI 임베딩으로 벡터 검색을 실행하는 방법을 알아보려면 AlloyDB AI로 벡터 임베딩 시작하기를 참고하세요.

목표

AlloyDB 클러스터 및 기본 인스턴스를 만듭니다.
데이터베이스에 연결하고 필요한 확장 프로그램을 설치합니다.
product 및 product inventory 테이블을 만듭니다.
product 및 product inventory 테이블에 데이터를 삽입하고 기본 벡터 검색을 실행합니다.
products 테이블에 ScaNN 색인을 만듭니다.
간단한 벡터 검색을 실행합니다.
필터와 조인으로 복잡한 벡터 검색을 실행합니다.

비용

이 문서에서는 비용이 청구될 수 있는 다음과 같은 Google Cloud 구성요소를 사용합니다.

프로젝트 사용량을 기준으로 예상 비용을 산출하려면 가격 계산기를 사용하세요. Google Cloud를 처음 사용하는 사용자는 무료 체험판을 사용할 수 있습니다.

이 문서에 설명된 태스크를 완료했으면 만든 리소스를 삭제하여 청구가 계속되는 것을 방지할 수 있습니다. 자세한 내용은 삭제를 참조하세요.

시작하기 전에

결제 및 필수 API 사용 설정

Google Cloud 콘솔에서 클러스터 페이지로 이동합니다.

프로젝트 선택기로 이동
Google Cloud 프로젝트에 결제가 사용 설정되어 있는지 확인합니다.
PostgreSQL용 AlloyDB를 만들고 연결하는 데 필요한 Cloud API를 사용 설정합니다.

API 사용 설정
1. 프로젝트 확인 단계에서 다음을 클릭하여 변경할 프로젝트의 이름을 확인합니다.
2. API 사용 설정 단계에서 사용 설정을 클릭하여 다음을 사용 설정합니다.
  - AlloyDB API
  - Compute Engine API
  - Service Networking API
  - Vertex AI API

AlloyDB 클러스터 및 기본 인스턴스 만들기

Google Cloud 콘솔에서 클러스터 페이지로 이동합니다.

클러스터로 이동
클러스터 만들기를 클릭합니다.
클러스터 ID에 my-cluster를 입력합니다.
비밀번호를 입력합니다. 이 비밀번호는 이 튜토리얼에서 사용되므로 기록해 둡니다.
리전을 선택합니다(예: us-central1 (Iowa)).
기본 네트워크를 선택합니다.

비공개 액세스 연결이 있는 경우 다음 단계로 진행합니다. 그렇지 않은 경우 연결 설정을 클릭하고 다음 단계를 따르세요.
1. IP 범위 할당에서 자동으로 할당된 IP 범위 사용을 클릭합니다.
2. 계속을 클릭한 다음 연결 만들기를 클릭합니다.
영역 가용성에서 단일 영역을 선택합니다.
2 vCPU,16 GB 머신 유형을 선택합니다.
연결에서 공개 IP 사용 설정을 선택합니다.
클러스터 만들기를 클릭합니다. AlloyDB에서 클러스터를 만들고 기본 클러스터 개요 페이지에 표시하는 데 몇 분 정도 걸릴 수 있습니다.
클러스터의 인스턴스에서 연결 창을 펼칩니다. 이 튜토리얼에서 사용되므로 연결 URI를 기록해 둡니다.

연결 URI는 projects/<var>PROJECT_ID</var>/locations/<var>REGION_ID</var>/clusters/my-cluster/instances/my-cluster-primary 형식입니다.

AlloyDB 서비스 에이전트에 Vertex AI 사용자 권한 부여

AlloyDB에서 Vertex AI 텍스트 임베딩 모델을 사용하도록 설정하려면 클러스터와 인스턴스가 있는 프로젝트의 AlloyDB 서비스 에이전트에 Vertex AI 사용자 권한을 추가해야 합니다.

권한을 추가하는 방법에 관한 자세한 내용은 AlloyDB 서비스 에이전트에 Vertex AI 사용자 권한 부여를 참고하세요.

웹브라우저를 사용하여 데이터베이스에 연결

Google Cloud 콘솔에서 클러스터 페이지로 이동합니다.

클러스터로 이동
리소스 이름 열에서 클러스터 이름 my-cluster을 클릭합니다.
탐색창에서 AlloyDB 스튜디오를 클릭합니다.
AlloyDB Studio에 로그인 페이지에서 다음 단계를 따르세요.
1. postgres 데이터베이스를 선택합니다.
2. postgres 사용자를 선택합니다.
3. 클러스터 및 클러스터의 기본 인스턴스 만들기에서 만든 비밀번호를 입력합니다.
4. 인증을 클릭합니다. 탐색기 창에 postgres 데이터베이스의 객체 목록이 표시됩니다.
+ 새 SQL 편집기 탭 또는 + 새 탭을 클릭하여 새 탭을 엽니다.

필수 확장 프로그램 설치

다음 쿼리를 실행하여 vector, alloydb_scann, google_ml_integration 확장 프로그램을 설치합니다.

  CREATE EXTENSION IF NOT EXISTS vector;
  CREATE EXTENSION IF NOT EXISTS alloydb_scann;
  CREATE EXTENSION IF NOT EXISTS google_ml_integration CASCADE;

제품 및 제품 인벤토리 데이터 삽입 및 기본 벡터 검색 실행

다음 문을 실행하여 다음을 실행하는 product 테이블을 만듭니다.
- 기본 제품 정보를 저장합니다.
- 각 제품의 제품 설명에 대한 임베딩 벡터를 계산하고 저장하는 embedding 벡터 열을 포함합니다.
참고: 테이블에 행이 10만 개를 초과하는 경우 embedding() 함수를 사용하여 테이블의 기존 데이터에 저장된 임베딩을 생성하지 않는 것이 좋습니다. 인라인 삽입 생성에 가장 적합합니다.
```
  CREATE TABLE product (
    id INT PRIMARY KEY,
    name VARCHAR(255) NOT NULL,
    description TEXT,
    category VARCHAR(255),
    color VARCHAR(255),
    embedding vector(768) GENERATED ALWAYS AS (embedding('text-embedding-005', description)) STORED
  );
```
다음 쿼리를 실행하여 사용 가능한 인벤토리 및 해당 가격에 관한 정보를 저장하는 product_inventory 테이블을 만듭니다. 이 튜토리얼에서는 product_inventory 및 product 테이블을 사용하여 복잡한 벡터 검색 쿼리를 실행합니다.
```
CREATE TABLE product_inventory (
  id INT PRIMARY KEY,
  product_id INT REFERENCES product(id),
  inventory INT,
  price DECIMAL(10,2)
);
```

다음 쿼리를 실행하여 product 테이블에 제품 데이터를 삽입합니다.

INSERT INTO product (id, name, description,category, color) VALUES
(1, 'Stuffed Elephant', 'Soft plush elephant with floppy ears.', 'Plush Toys', 'Gray'),
(2, 'Remote Control Airplane', 'Easy-to-fly remote control airplane.', 'Vehicles', 'Red'),
(3, 'Wooden Train Set', 'Classic wooden train set with tracks and trains.', 'Vehicles', 'Multicolor'),
(4, 'Kids Tool Set', 'Toy tool set with realistic tools.', 'Pretend Play', 'Multicolor'),
(5, 'Play Food Set', 'Set of realistic play food items.', 'Pretend Play', 'Multicolor'),
(6, 'Magnetic Tiles', 'Set of colorful magnetic tiles for building.', 'Construction Toys', 'Multicolor'),
(7, 'Kids Microscope', 'Microscope for kids with different magnification levels.', 'Educational Toys', 'White'),
(8, 'Telescope for Kids', 'Telescope designed for kids to explore the night sky.', 'Educational Toys', 'Blue'),
(9, 'Coding Robot', 'Robot that teaches kids basic coding concepts.', 'Educational Toys', 'White'),
(10, 'Kids Camera', 'Durable camera for kids to take pictures and videos.', 'Electronics', 'Pink'),
(11, 'Walkie Talkies', 'Set of walkie talkies for kids to communicate.', 'Electronics', 'Blue'),
(12, 'Karaoke Machine', 'Karaoke machine with built-in microphone and speaker.', 'Electronics', 'Black'),
(13, 'Kids Drum Set', 'Drum set designed for kids with adjustable height.', 'Musical Instruments', 'Blue'),
(14, 'Kids Guitar', 'Acoustic guitar for kids with nylon strings.', 'Musical Instruments', 'Brown'),
(15, 'Kids Keyboard', 'Electronic keyboard with different instrument sounds.', 'Musical Instruments', 'Black'),
(16, 'Art Easel', 'Double-sided art easel with chalkboard and whiteboard.', 'Arts & Crafts', 'White'),
(17, 'Finger Paints', 'Set of non-toxic finger paints for kids.', 'Arts & Crafts', 'Multicolor'),
(18, 'Modeling Clay', 'Set of colorful modeling clay.', 'Arts & Crafts', 'Multicolor'),
(19, 'Watercolor Paint Set', 'Watercolor paint set with brushes and palette.', 'Arts & Crafts', 'Multicolor'),
(20, 'Beading Kit', 'Kit for making bracelets and necklaces with beads.', 'Arts & Crafts', 'Multicolor'),
(21, '3D Puzzle', '3D puzzle of a famous landmark.', 'Puzzles', 'Multicolor'),
(22, 'Race Car Track Set', 'Race car track set with cars and accessories.', 'Vehicles', 'Multicolor'),
(23, 'RC Monster Truck', 'Remote control monster truck with oversized tires.', 'Vehicles', 'Green'),
(24, 'Train Track Expansion Set', 'Expansion set for wooden train tracks.', 'Vehicles', 'Multicolor');

선택사항: 다음 쿼리를 실행하여 데이터가 product 테이블에 삽입되었는지 확인합니다.
```
SELECT * FROM product;
```

다음 쿼리를 실행하여 product_inventory 테이블에 인벤토리 데이터를 삽입합니다.

INSERT INTO product_inventory (id, product_id, inventory, price) VALUES
(1, 1, 9, 13.09),
(2, 2, 40, 79.82),
(3, 3, 34, 52.49),
(4, 4, 9, 12.03),
(5, 5, 36, 71.29),
(6, 6, 10, 51.49),
(7, 7, 7, 37.35),
(8, 8, 6, 10.87),
(9, 9, 7, 42.47),
(10, 10, 3, 24.35),
(11, 11, 4, 10.20),
(12, 12, 47, 74.57),
(13, 13, 5, 28.54),
(14, 14, 11, 25.58),
(15, 15, 21, 69.84),
(16, 16, 6, 47.73),
(17, 17, 26, 81.00),
(18, 18, 11, 91.60),
(19, 19, 8, 78.53),
(20, 20, 43, 84.33),
(21, 21, 46, 90.01),
(22, 22, 6, 49.82),
(23, 23, 37, 50.20),
(24, 24, 27, 99.27);

music라는 단어와 유사한 제품을 찾는 다음 벡터 검색 쿼리를 실행합니다. 즉, 제품 설명에 music라는 단어가 명시적으로 언급되지 않더라도 검색어와 관련된 제품이 결과에 표시됩니다.
```
SELECT * FROM product
ORDER BY embedding <=> embedding('text-embedding-005', 'music')::vector
LIMIT 3;
```
쿼리 결과는 다음과 같습니다.

색인을 만들지 않고 기본 벡터 검색을 실행하면 효율적인 검색을 제공하는 정확한 최근접 이웃 검색 (KNN)이 사용됩니다. 대규모로 KNN을 사용하면 성능에 영향을 줄 수 있습니다. 쿼리 성능을 개선하려면 지연 시간이 짧으면서 재현율이 높은 근사 최근접 이웃 (ANN) 검색에 ScaNN 색인을 사용하는 것이 좋습니다.

색인을 만들지 않으면 AlloyDB는 기본적으로 정확한 최근접 이웃 검색 (KNN)을 사용합니다.

대규모로 ScaNN을 사용하는 방법에 관한 자세한 내용은 AlloyDB AI로 벡터 임베딩 시작하기를 참고하세요.

제품 테이블에 ScaNN 색인 만들기

다음 쿼리를 실행하여 product 테이블에 product_index ScaNN 색인을 만듭니다.

  CREATE INDEX product_index ON product
  USING scann (embedding cosine)
  WITH (num_leaves=5);

num_leaves 매개변수는 트리 기반 색인이 색인을 빌드하는 데 사용하는 리프 노드 수를 나타냅니다. 이 매개변수를 조정하는 방법에 관한 자세한 내용은 벡터 쿼리 성능 조정을 참고하세요.