how to set snowflake query_tag for ibis snowflake backend.

[ruiyang2015]

Search the docs and also tried using chatgpt, none give a good answer on how to set the connection and session level query_tag for ibis.snowflake backend, please help

[lostmygithubaccount]

hi @ruiyang2015, is that something that can be set in the Snowflake Python connector? Ibis passes through additional kwargs to the underlying connection library if so

[cpcloud]

You can set this using

con.raw_sql("ALTER SESSION SET QUERY_TAG = 'my_tag'")

or by passing the appropriate session_parameters argument when connecting:

ibis.snowflake.connect(
    ...,
    session_parameters={"QUERY_TAG": "my_tag"}
)