Snowflake CortexのCOMPLETEでSQL文のみを取得できるか試してみた

import pandas as pd
from snowflake.cortex import complete, CompleteOptions

messages = []
messages.append({"role": "system", "content": "SQL文のみを回答してください。```sqlのようなコードブロックは付けないでください。"})
messages.append({"role": "user", "content": "Snowflakeの構文のSQLでなにか複雑なものを考えてください。"})

answer = complete(model='mistral-large2', prompt=messages)

SELECT  
    employee_id,  
    first_name,  
    last_name,  
    department,  
    salary,  
    CASE  
        WHEN salary > 100000 THEN 'High'  
        WHEN salary BETWEEN 50000 AND 100000 THEN 'Medium'  
        ELSE 'Low'  
    END AS salary_level,  
    RANK() OVER (PARTITION BY department ORDER BY salary DESC) AS rank_in_department,  
    LAG(salary, 1) OVER (PARTITION BY department ORDER BY salary DESC) AS previous_salary,  
    LEAD(salary, 1) OVER (PARTITION BY department ORDER BY salary DESC) AS next_salary  
FROM  
    employees  
WHERE  
    hire_date > '2020-01-01'  
ORDER BY  
    department, salary DESC;

import pandas as pd
from snowflake.cortex import complete, CompleteOptions

response_format = {
    "type": "json",
    "schema": {
        "type": "object",
        "properties": {
            "sql": {
                "type": "string",
                "description": "SQL文"
            },
            "text": {
                "type": "string",
                "description": "SQL文の解説"
            },
        }
    },
    "required": ["sql", "text"],
}
prompt = [{
    "role": "user",
    "content": "snowflakeの構文のSQLでなにか複雑なものを考えてください。SQL文および解説文をください。"
}]

options = CompleteOptions(
        max_tokens=4096,
        guardrails=False,
        response_format=response_format
    )

result = complete(
    model="mistral-large2",
    prompt=prompt,
    options=options,
)

output = "".join(result)
print(output)

{  
  "sql": "WITH RECURSIVE cte AS (\n  SELECT 1 AS n\n  UNION ALL\n  SELECT n + 1\n  FROM cte\n  WHERE n < 10\n)\nSELECT n, n * 2 AS double_n\nFROM cte;",  
  "text": "このSQL文は、再帰的な共通テーブル式（CTE）を使用して、1から10までの数字を生成し、それぞれの数字の2倍を計算します。\n\n1. `WITH RECURSIVE cte AS (...)`: 再帰的なCTEを定義します。\n2. `SELECT 1 AS n`: 初期値として1を設定します。\n3. `UNION ALL`: 初期値と再帰的な部分を結合します。\n4. `SELECT n + 1 FROM cte WHERE n < 10`: 再帰的に次の数字を生成し、nが10未満の場合に繰り返します。\n5. `SELECT n, n * 2 AS double_n FROM cte`: 生成された数字とその2倍を選択します。\n\nこのクエリは、再帰的なCTEを使用して数列を生成し、その数列に対して計算を行う方法を示しています。"  
}