◾️はじめに

Snowflake Cortex AI に関してメモしておく

【１】Snowflake Cortex AI

* Snowflake上でセキュアにLLM/MLを活用できるAI機能群

cf. Cortex (コーテックス) = 皮質, 外皮, 大脳皮質

https://www.snowflake.com/ja/product/features/cortex/

【２】料金

* トークン数に対するクレジット課金と
　メッセージ数に対するクレジット課金の従量課金制で提供
 => 使用した分だけ課金されるので、考えて使わないと大変なことになる

* モデルによっても料金が異なる

https://docs.snowflake.com/ja/user-guide/snowflake-cortex/aisql#label-cortex-llm-cost-considerations

【３】Cortex が実行できるかどうか確認

-- 以下を実行してみる
SHOW USER FUNCTIONS IN SCHEMA SNOWFLAKE.CORTEX;

１）AI関数を呼び出してみる

-- Change here!!
USE WAREHOUSE YOUR_WH;

SELECT
 SNOWFLAKE.CORTEX.AI_COMPLETE(
 'llama3.1-8b',
 'Return only the word OK.'
) AS result;

【４】Tips

１）モデルの表示

SHOW MODELS;

SHOW MODELS IN SNOWFLAKE.CORTEX;

https://docs.snowflake.com/ja/sql-reference/sql/show-models

【５】サンプル

例１：Cortexを使ったテーブル項目のコメント生成

-- Change here!!
USE DATABASE YOUR_DB;

SELECT
  t.TABLE_CATALOG AS database_name,
  t.TABLE_SCHEMA AS schema_name,
  t.TABLE_NAME AS table_name,
  t.COMMENT AS existing_table_comment,
  -- PAY ATTENTION!!
  SNOWFLAKE.CORTEX.AI_COMPLETE(
    'llama3.1-8b',
    'You are a data warehouse documentation assistant. ' ||
    'Write a concise English column description (<= 80 chars). ' ||
    'Do not guess business meaning beyond the name; state neutral meaning. ' ||
    'Return only the description text without any marks. ' ||
    'Table: ' || t.TABLE_SCHEMA || '.' || t.TABLE_NAME
  )::string AS table_comment
FROM
  INFORMATION_SCHEMA.TABLES AS t
WHERE
  t.TABLE_SCHEMA NOT IN ('INFORMATION_SHEMA')
LIMIT 100
;