Cortex Search Service 쿼리하기¶

Snowflake Python APIs 라이브러리 설치¶

먼저 PyPI에서 최신 버전의 Snowflake Python APIs 패키지를 설치 관리자에 설치합니다. PyPI에서 이 패키지를 설치하는 방법에 대한 지침은 Snowflake Python APIs 라이브러리 설치 섹션을 참조하십시오.

pip install snowflake -U

Snowflake에 연결하기¶

Snowpark Session 또는 Python Connector Connection 을 사용하여 Snowflake에 연결하고 Root 오브젝트를 생성합니다. Snowflake에 연결하는 방법에 대한 자세한 지침은 Snowflake Python APIs 을 사용하여 Snowflake에 연결 섹션을 참조하십시오. 다음 예제에서는 Snowpark Session 오브젝트와 Python 사전을 사용하여 구성합니다.

import os
from snowflake.core import Root
from snowflake.snowpark import Session

CONNECTION_PARAMETERS = {
    "account": os.environ["snowflake_account_demo"],
    "user": os.environ["snowflake_user_demo"],
    "password": os.environ["snowflake_password_demo"],
    "role": "test_role",
    "database": "test_database",
    "warehouse": "test_warehouse",
    "schema": "test_schema",
}

session = Session.builder.configs(CONNECTION_PARAMETERS).create()
root = Root(session)

서비스 쿼리¶

다음 구문을 사용하여 서비스를 쿼리합니다.

# fetch service
my_service = (root
  .databases["<service_database>"]
  .schemas["<service_schema>"]
  .cortex_search_services["<service_name>"]
)

# query service
resp = my_service.search(
  query="<query>",
  columns=["<col1>", "<col2>"],
  filter={"@eq": {"<column>": "<value>"} },
  limit=5
)
print(resp.to_json())

매개 변수¶

필터 구문¶

Cortex Search는 CREATE CORTEX SEARCH SERVICE 명령에 지정된 ATTRIBUTES 열에 대한 필터링을 지원합니다.

Cortex Search는 2개의 일치 연산자를 지원합니다.

• 문자열 동등성: @eq

• 배열에 포함된 항목: @contains

이러한 일치 연산자는 다양한 논리 연산자로 구성될 수 있습니다.

• @and

• @or

• @not

이러한 연산자를 1개의 필터 오브젝트로 결합할 수 있습니다. 다음은 그 예입니다.

• 문자열과 같은 열 string_col 은 value 값과 같은 행을 필터링합니다.

  { "@eq": { "string_col": "value" } }
  

  Copy

• ARRAY 열 array_col 에 값 value 가 포함된 행을 필터링합니다.

  { "@contains": { "array_col": "arr_value" } }
  

  Copy

• 논리 연산자를 사용하여 필터 구성:

  // Rows where the "array_col" column contains "arr_value" and the "string_col" column equals "value":
  {
      "@and": [
        { "@contains": { "array_col": "arr_value" } },
        { "@eq": { "string_col": "value" } }
      ]
  }
  
  // Rows where the "string_col" column does not equal "value"
  {
    "@not": { "@eq": { "string_col": "value" } }
  }
  
  // Rows where the "array_col" column contains at least one of "val1", "val2", or "val3"
  {
    "@or": [
        { "@contains": { "array_col": "val1" } },
        { "@contains": { "array_col": "val1" } },
        { "@contains": { "array_col": "val1" } }
    ]
  }
  

  Copy

REST API 인증 구성¶

Snowflake REST APIs는 현재 키 페어 인증 및 OAuth를 통한 인증을 지원합니다. 자세한 내용은 Snowflake를 사용하여 Snowflake REST APIs 인증하기 섹션을 참조하십시오.

서비스 쿼리의 예¶

curl 및 키페어 인증을 사용하여 서비스를 쿼리하려면:

curl --location https://<ACCOUNT_URL>/api/v2/databases/<DB_NAME>/schemas/<SCHEMA_NAME>/cortex-search-services/<SERVICE_NAME>\:query \
--header 'X-Snowflake-Authorization-Token-Type: KEYPAIR_JWT' \
--header 'Content-Type: application/json' \
--header 'Accept: application/json' \
--header "Authorization: Bearer $CORTEX_SEARCH_JWT" \
--data '{
  "query": "<search_query>",
  "columns": ["col1", "col2"],
  "filter": <filter>
  "limit": <limit>
}'

예¶

다음 예제에서는 preview query 쿼리 문자열로 서비스를 미리 보고 결과를 VARIANT 오브젝트로 구문 분석합니다.

SELECT PARSE_JSON(
  SNOWFLAKE.CORTEX.SEARCH_PREVIEW(
      'my_search_service',
      '{
         "query": "preview query",
         "columns":[
            "col1",
            "col2"
         ],
         "filter": {"@eq": {"col1": "filter value"} },
         "limit":10
      }'
  )
)['results'] as results;