[Azure] Cognitive Search로 자동 완성 검색 구현하기 Suggest API(Python)

 

[Azure] Cognitive Search로 검색 및 필터 설정 (Python)

 

기본적인 검색과 필터링을 하였으니 검색 클라이언트 중

Suggest API를 이용한 자동 완성 기능을 구현해 보겠습니다.

 

기본적인 검색을 구현하기 위해서 인덱스를 생성하였는데

자동 완성 검색을 구현하려면 추가적으로 suggesters를 설정해주셔야 합니다.

 

# 인덱스 생성
from azure.core.credentials import AzureKeyCredential
from azrue.search.documents.indexes import SearchIndexClient

search_service_name = 'your-service-name'
key = 'your-api-key'
endpoint = f'https://{serach_service_name}.search.windows.net/'

# 인덱스 클라이언트를 생성
index_client = SearchIndexClient(endpoint, AzureKeyCredential(key))

# 샘플 인덱스명 / 필드 상세
index_name = 'test'
fields = [
        SimpleField(name="testId", type=SearchFieldDataType.String, key=True),
        SearchableField(name="testRate", type=SearchFieldDataType.Double, 
                        filterable=True),
        SearchableField(name="description", type=SearchFieldDataType.String),
        SearchableField(name="testTag", type=SearchFieldDataType.String, 
                        filterable=True, collection=True)
    ]

 

위 필드에 상세 설정까지는 기본적으로 동일합니다만 CORS Option과 Scoring Profile은 동일하게 두고,

추가적으로 suggeter를 생성하는 코드를 작성합니다.

 

# Suggester 설정

from azure.search.documents.indexes.models import (
    Suggester, 
    CorsOptions,
    ScoringProfile
)

suggester = Suggester(
    name = 'testSuggester' # suggester title
    source_fields = ['description', 'testTag'] # 자동 완성 대상 필드
)

cors_options = CorsOptions(allowed_origins=["*"], max_age_in_seconds=60)
scoring_profiles = []

index = SearchIndex(
    name=index_name,
    fields=fields,
    suggester=suggester,
    scoring_profiles=scoring_profiles,
    cors_options=cors_options)

result = index_client.create_index(index)

 

이제 자동 완성 검색을 위한 준비는 모두 끝났습니다.

이제 자동 완성 검색을 해보도록 하겠습니다.

 

from azure.search.documents import SearchClient

# 검색 클라이언트 생성
search_client = SearchClient(endpoint, index_name, AzureKeyCredential(key))

# 자동완성 검색
auto_response = list(search_client.suggest('가중', 'testSuggester'))

# 일반 검색
response = list(search_client.search('가중'))

# 데이터 예제
{
    "testId": "4",
    "testRate": 4.5,
    "description" : "가중치 +0 테스트"
    "testTag": [
        "성공",
        "가중치",
        "검색"
    ]
},
{
    "testId": "6",
    "testRate": 3.5,
    "description" : "가중치 +1 테스트"
    "testTag": [
        "실패",
        "종료"
    ] 
}

 

위 코드에서 "description"과 "testTag"를 대상 필드로 지정하였으므로

자동 완성 단어는 두 필드를 기준으로 생성하여 결과를 보내주게 됩니다.