Cortex Search サービスを Snowflake Intelligence 経由でツールとして使用するまでの手順を試してみた

USE ROLE ACCOUNTADMIN;
ALTER ACCOUNT SET CORTEX_ENABLED_CROSS_REGION = 'AWS_US';

USE ROLE SYSADMIN;

-- cortex searchサービスなどの作成先となるデータベースを作成
CREATE DATABASE IF NOT EXISTS cortex_db;

-- cortex_searchサービス作成先となるスキーマを作成
CREATE SCHEMA IF NOT EXISTS cortex_db.cortex_search_services;

-- ドキュメントの配置先となる内部ステージ、テーブルの作成先となるスキーマを作成
CREATE SCHEMA IF NOT EXISTS cortex_db.data;

-- cortex agentsの作成先となるスキーマを作成
CREATE SCHEMA IF NOT EXISTS cortex_db.cortex_agents;

-- オプション：仮想ウェアハウスを作成
CREATE OR REPLACE WAREHOUSE cortex_wh WITH
     WAREHOUSE_SIZE='X-SMALL'
     AUTO_SUSPEND = 120
     AUTO_RESUME = TRUE
     INITIALLY_SUSPENDED=TRUE;

-- 作業用のロールを作成
USE ROLE USERADMIN;
CREATE ROLE IF NOT EXISTS cortex_search_role;

-- 権限付与
USE ROLE SECURITYADMIN;

-- 親データベーススキーマの使用権限
GRANT USAGE ON DATABASE cortex_db TO ROLE cortex_search_role;
GRANT USAGE ON SCHEMA cortex_db.cortex_search_services TO ROLE cortex_search_role;
GRANT USAGE ON SCHEMA cortex_db.data TO ROLE cortex_search_role;

-- ウェアハウス権限
GRANT USAGE, MONITOR, OPERATE ON WAREHOUSE cortex_wh TO ROLE cortex_search_role;

-- ステージ作成権限
GRANT CREATE STAGE ON SCHEMA cortex_db.data TO ROLE cortex_search_role;

-- テーブル作成権限
GRANT CREATE TABLE ON SCHEMA cortex_db.data TO ROLE cortex_search_role;

-- Cotex Search Serviceの作成権限
GRANT CREATE CORTEX SEARCH SERVICE ON SCHEMA cortex_db.cortex_search_services TO ROLE cortex_search_role;

-- ロール継承
GRANT ROLE cortex_search_role TO ROLE SYSADMIN;

-- ロールを切り替え
USE ROLE cortex_search_role;

-- コンテキストを指定
USE SCHEMA cortex_db.data;

-- ドキュメントの配置先となるステージを作成
CREATE OR REPLACE STAGE cortex_db.data.documents
    DIRECTORY = (ENABLE = TRUE)
    ENCRYPTION = (TYPE = 'SNOWFLAKE_SSE');

-- テーブルを作成
-- AI_PARSE_DOCUMENT関数を使用し、 PDFからテキストとレイアウト情報を抽出し、RAW_TEXTテーブルに格納
CREATE OR REPLACE TEMPORARY TABLE cortex_db.data.raw_text AS
WITH FILE_TABLE as (
  (SELECT 
        RELATIVE_PATH,
        SIZE,
        'https://app.snowflake.com/XXXXX/XXXXX/#/data/databases/CORTEX_DB/schemas/DATA/stage/DOCUMENTS' AS STAGE_URL,
        TO_FILE('@cortex_db.data.documents', RELATIVE_PATH) AS docs 
    FROM 
        DIRECTORY(@cortex_db.data.documents))
)
SELECT 
    RELATIVE_PATH,
    SIZE,
    STAGE_URL,
    TO_VARCHAR (
        SNOWFLAKE.CORTEX.AI_PARSE_DOCUMENT (
            docs,
            {'mode': 'LAYOUT'} ):content
        ) AS EXTRACTED_LAYOUT 
FROM 
    FILE_TABLE;

-- チャンクテーブル
CREATE OR REPLACE TABLE chunks_table ( 
    RELATIVE_PATH VARCHAR, 
    SIZE NUMBER, 
    STAGE_URL VARCHAR, 
    CHUNK VARCHAR, 
    CHUNK_INDEX INT 
);

-- チャンク化したレコードを追加
INSERT INTO chunks_table (relative_path, size, stage_url,
                            chunk, chunk_index)

    SELECT relative_path, 
            size,
            stage_url, 
            c.value::TEXT as chunk,
            c.INDEX::INTEGER as chunk_index            
    FROM 
        raw_text,
        LATERAL FLATTEN( input => SNOWFLAKE.CORTEX.SPLIT_TEXT_RECURSIVE_CHARACTER (
              EXTRACTED_LAYOUT,
              'markdown',
              2000, -- ドキュメントを最大サイズ2000文字のチャンクに分割
              300, -- オーバーラップ
              ['\n\n', '\n', ' ', '']
           )) c;

-- テーブルを確認
SELECT * FROM chunks_table;

-- cortex search serviceを作成
USE SCHEMA cortex_db.cortex_search_services;

CREATE OR REPLACE CORTEX SEARCH SERVICE cortex_db.cortex_search_services.test_search_service
ON chunk
ATTRIBUTES relative_path
WAREHOUSE = cortex_wh
EMBEDDING_MODEL = 'voyage-multilingual-2'
TARGET_LAG = '1 day'
AS (
    SELECT chunk,
        chunk_index,
        relative_path,
        stage_url
    FROM cortex_db.data.chunks_table
);

-- 例：SQLでクエリ
SELECT PARSE_JSON(
  SNOWFLAKE.CORTEX.SEARCH_PREVIEW(
      'cortex_db.cortex_search_services.test_search_service',
      '{
         "query": "Snowpipeの価格体系変更について教えてください",
         "limit":10
      }'
  )
)['results'] as results;

-- Cortex Agents ロール
USE ROLE USERADMIN;
CREATE ROLE IF NOT EXISTS cortex_agent_role;

-- 権限付与
USE ROLE SECURITYADMIN;

-- Cortex Searchサービスの使用権限
GRANT USAGE ON DATABASE cortex_db TO ROLE cortex_agent_role;
GRANT USAGE ON SCHEMA cortex_db.cortex_search_services TO ROLE cortex_agent_role;
GRANT USAGE ON CORTEX SEARCH SERVICE cortex_db.cortex_search_services.test_search_service
    TO ROLE cortex_agent_role;

-- スキーマへのエージェント作成権限
GRANT USAGE ON SCHEMA cortex_db.cortex_agents TO ROLE cortex_agent_role;
GRANT CREATE AGENT ON SCHEMA cortex_db.cortex_agents TO ROLE cortex_agent_role;

-- ロール継承
GRANT ROLE cortex_agent_role TO ROLE SYSADMIN;

あなたは「社内ナレッジに基づく一次回答」を行うサポートエージェントです。
回答を作成する際は、次の方針に従ってください。

- 一次情報（ドキュメント）を最優先する
  - 提示された社内/公式ドキュメント、仕様書、手順書の内容を最優先に参照し、それに基づいて回答する。
  - ドキュメントと推測が混ざる場合は、ドキュメントに書かれている内容と補足（推測/一般論）を明確に分けて記載する。

- 不明点は断定せず、確認先を案内する
  - ドキュメントに明記されていない点、環境依存・権限依存・運用ルール依存の点は断定しない。
  - 回答に必要な情報が不足している場合は、必要な追加情報を質問するか、次のように案内する：
    - 「詳細については○○部門（例：情報システム部門／運用担当／セキュリティ担当）にお問い合わせください」
  - 可能なら「何を確認すべきか（確認観点）」も併記する。

- 読み手にとって親切で分かりやすい表現にする
  - 専門用語は必要最小限にし、初出時は短く説明する。
  - 結論→理由→補足の順で、要点が先に伝わる構成にする。
  - 曖昧な表現を避け、条件（前提・対象・例外）を明確にする。
  - 手順がある場合は番号付きで具体的に書く

- 操作や実行手順は、必要に応じて番号付きリストで記載する。
  - 可能なら「前提条件」「期待される結果」「よくあるエラー/注意点」も簡潔に添える。

USE ROLE ACCOUNTADMIN;
CREATE SNOWFLAKE INTELLIGENCE SNOWFLAKE_INTELLIGENCE_OBJECT_DEFAULT;

-- ロールを作成
USE ROLE USERADMIN;
CREATE ROLE IF NOT EXISTS snowflake_intelligence_admin;

-- 権限付与
USE ROLE SECURITYADMIN;
GRANT ROLE snowflake_intelligence_admin TO ROLE SYSADMIN;

-- Snowflake Intelligenceオブジェクトへの操作権限
GRANT USAGE ON SNOWFLAKE INTELLIGENCE SNOWFLAKE_INTELLIGENCE_OBJECT_DEFAULT TO ROLE snowflake_intelligence_admin;
GRANT MODIFY ON SNOWFLAKE INTELLIGENCE SNOWFLAKE_INTELLIGENCE_OBJECT_DEFAULT TO ROLE snowflake_intelligence_admin;

-- Cortex Agentsの使用権限が必要なので付与
GRANT USAGE ON DATABASE cortex_db TO ROLE snowflake_intelligence_admin;
GRANT USAGE ON SCHEMA cortex_db.cortex_agents TO ROLE snowflake_intelligence_admin;
GRANT USAGE ON AGENT cortex_db.cortex_agents.test_agent TO ROLE snowflake_intelligence_admin;

-- Agentを追加
USE ROLE snowflake_intelligence_admin;
ALTER SNOWFLAKE INTELLIGENCE SNOWFLAKE_INTELLIGENCE_OBJECT_DEFAULT ADD AGENT cortex_db.cortex_agents.my_agent;

CREATE ROLE cortex_agent_user_role;
GRANT ROLE cortex_agent_user_role TO USER <ユーザー>;

--権限を与える
USE ROLE SECURITYADMIN;

-- cortex agent の使用権限
GRANT USAGE ON DATABASE cortex_db TO ROLE cortex_agent_user_role;
GRANT USAGE ON SCHEMA cortex_db.cortex_agents TO ROLE cortex_agent_user_role;
GRANT USAGE ON AGENT cortex_db.cortex_agents.test_agent TO ROLE cortex_agent_user_role;

-- cortex search サービスの使用権限
GRANT USAGE ON SCHEMA cortex_db.cortex_search_services TO ROLE cortex_agent_user_role;
GRANT USAGE ON CORTEX SEARCH SERVICE cortex_db.cortex_search_services.test_search_service
    TO ROLE cortex_agent_user_role;

-- 仮想ウェアハウスの使用権限
GRANT USAGE ON WAREHOUSE <ウェアハウス> TO ROLE cortex_agent_user_role;