Snowflake Cortex AIのCortex AnalystのREST APIにSnowflake外のサービスからリクエストしてみる

-- ROLEの作成
CREATE ROLE SNOWPARK_ROLE;

-- ウェアハウスへのUSAGEアクセス権の付与
GRANT USAGE ON warehouse ウェアハウス名 TO ROLE SNOWPARK_ROLE;

-- データベースへのUSAGEアクセス権の付与
GRANT USAGE ON DATABASE データベース名 TO ROLE SNOWPARK_ROLE;

-- スキーマへのUSAGEアクセス権の付与
GRANT USAGE ON SCHEMA データベース名.スキーマ名 TO ROLE SNOWPARK_ROLE;

-- テーブルへのSELECT権限の付与
GRANT SELECT ON ALL TABLES IN SCHEMA データベース名.スキーマ名 TO ROLE SNOWPARK_ROLE;

-- セマンティックモデルの読み取り権限の付与
GRANT READ ON データベース名.スキーマ名.ステージ名  TO ROLE SNOWPARK_ROLE;

-- ユーザーの作成
CREATE USER SNOWPARK_OPERATOR
    PASSWORD = 'パスワード'
    DEFAULT_ROLE = 'SNOWPARK_ROLE'

-- ユーザーへのロールの付与
GRANT ROLE SNOWPARK_ROLE TO USER SNOWPARK_OPERATOR;

import requests
import snowflake.connector

# 接続の作成
session = snowflake.connector.connect(
    user="SNOWPARK_OPERATOR",
    password="パスワード",
    account="アカウント識別子",
    warehouse="ウェアハウス名",
    role="SNOWPARK_ROLE",
    )

# Cortex Analyst向けの設定変数
DATABASE = "データベース名"
SCHEMA = "スキーマ名"
STAGE = "ステージ名"
FILE = "セマンティックモデルファイル名"
API_ENDPOINT = "/api/v2/cortex/analyst/message"

# Cortex Analystへの問い合わせ
def send_message(message, session):
    request_body = {
        "messages": [message],
        "semantic_model_file": f"@{DATABASE}.{SCHEMA}.{STAGE}/{FILE}",
    }
    resp = requests.post(
        url=f"https://{session.host}/api/v2/cortex/analyst/message",
        json=request_body,
        headers={
            "Authorization": f'Snowflake Token="{session.rest.token}"',
            "Content-Type": "application/json",
        },
        stream=True,
    )
    if resp.status_code < 400:
        return resp  # type: ignore
    else:
        raise Exception(f"Failed request with status {resp.status_code}: {resp.text}")

# 質問例
query = "総取引額トップ３の顧客を教えてください。"
message = {"role": "user",
           "content": [
                {
                    "type": "text",
                    "text": query
                }]}

# 質問の実行
resp = send_message(message, session)
resp.json()

    resp = _snowflake.send_snow_api_request(
        "POST",
        f"/api/v2/cortex/analyst/message",
        {},
        {},
        request_body,
        {},
        30000,
    )