Abfrage eines Cortex Search Service¶

Installieren der Snowflake Python APIs-Bibliothek¶

Installieren Sie zunächst die neueste Version des Pakets Snowflake Python APIs von PyPI. Siehe Installieren der Snowflake Python APIs-Bibliothek für Anweisungen zur Installation dieses Pakets von PyPI.

pip install snowflake -U

Mit Snowflake verbinden¶

Stellen Sie die Verbindung zu Snowflake entweder über einen Snowpark-Session oder einen Python Connector Connection her und erstellen Sie ein Root Objekt. Unter Verbinden mit Snowflake mit dem Snowflake Python APIs finden Sie weitere Anweisungen zur Verbindung mit Snowflake. Das folgende Beispiel verwendet das Snowpark Session-Objekt und ein Python Wörterbuch für die Konfiguration.

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)

Den Dienst abfragen¶

Fragen Sie den Dienst mit der folgenden Syntax ab:

# 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())

Parameter¶

Filtersyntax¶

Cortex Search unterstützt das Filtern nach den ATTRIBUTES-Spalten, die im Befehl CREATE CORTEX SEARCH SERVICE angegeben sind.

Cortex Search unterstützt zwei Abgleichsoperatoren:

• Zeichenfolgengleichheit: @eq

• Array enthält: @contains

Diese Vergleichsoperatoren können mit verschiedenen logischen Operatoren kombiniert werden:

• @and

• @or

• @not

Diese Operatoren können in einem einzigen Filterobjekt kombiniert werden. Es folgen Beispiele:

• Filtern von Zeilen, bei denen die Zeichenfolge-ähnliche Spalte string_col gleich dem Wert value ist.

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

  Copy

• Filtern nach Zeilen, in denen die Spalte ARRAY array_col den Wert value enthält.

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

  Copy

• Zusammenstellen von Filtern mit logischen Operatoren:

  // 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-Authentifizierung konfigurieren¶

Snowflake REST APIs unterstützt derzeit die Authentifizierung über Schlüsselpaar-Authentifizierung und OAuth. Weitere Details dazu finden Sie unter Authentifizierung von Snowflake REST APIs mit Snowflake.

Beispiel für die Abfrage des Dienstes¶

So fragen Sie den Dienst ab: curl und Authentifizierung des Schlüsselpaares:

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>
}'

Beispiel¶

Das folgende Beispiel zeigt eine Vorschau des Services mit der Abfragezeichenfolge preview query und analysiert die Ergebnisse in einem VARIANT-Objekt.

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;