Python 通过雪花的气流会话\u参数

正如您在代码示例中所看到的，我试图通过Airflow DAG任务将会话_参数传递给我的雪花连接，但没有拾取该参数，有解决方案吗

task=snowflake操作符(
task_id='task'，
sql=“创建或替换表MY\u DB.MY\u SCHEMA.MY\u表（test VARCHAR）”，
会话参数={
“查询标签”：“我的标签”
},
雪花连接id=“雪花连接”，
warehouse=“MY_WH”，
dag=dag，
)

您可能正在使用一个过时版本的雪花操作符。 中添加了对会话参数的支持

对于

气流=2.0.0

，您需要：

pip install apache-airflow-backport-providers-snowflake>='2020.11.23'

pip install apache-airflow-providers-snowflake

然后，您可以将代码用作：

from airflow.providers.snowflake.operators.snowflake import SnowflakeOperator
task = SnowflakeOperator(
    task_id='Task',
    sql="CREATE OR REPLACE TABLE MY_DB.MY_SCHEMA.MY_TABLE (test VARCHAR)",
    session_parameters={
        "QUERY_TAG": "my_tag"
    },
    snowflake_conn_id="snowflake_connection",
    warehouse="MY_WH",
    dag=dag,
)