Snowflake中Streamlit切换角色失败：USE ROLE语句报错如何解决？

问题 #

在Snowflake的Streamlit代码中，使用Snowpark的session.sql执行以下命令：

session.sql("USE ROLE SECURITYADMIN").collect()

收到错误信息：

SQL execution failed with error: (1304): 01b9ac90-0001-abd6-0000-17ed2f9d380e: 090236 (42601): 01b9ac90-0001-abd6-0000-17ed2f9d380e: Stored procedure execution error: Unsupported statement type 'USE'.

相关疑问：

• Snowflake文档未提及USE语句不支持，之前在Snowpark中使用该语句正常，想知道Streamlit运行的会话有何差异导致不支持？
• 不清楚错误中的“存储过程”具体指什么，原本以为是直接用会话令牌执行SQL，而非调用存储过程。
• 期望成功将会话角色切换为SECURITYADMIN，尝试过USE SECONDARY ROLES ALL也无效，希望在不提升Streamlit运行角色权限的前提下执行授权操作。

解决方案 #

会话差异原因 #

Streamlit在Snowflake环境中运行时，所有SQL语句都会通过内置存储过程执行，而非直接的会话级SQL调用——这就是错误提示里“存储过程执行错误”的由来。Snowflake的存储过程不允许执行USE这类会话上下文切换语句，而你之前在本地Snowpark中是直接通过会话执行SQL，不受这个限制。

具体解决办法 #

• 用Snowpark API替代SQL切换角色
  放弃session.sql("USE ROLE ...")的写法，改用Snowpark提供的原生API方法切换角色：

  session.use_role("SECURITYADMIN")
  

  这个方法直接操作Snowpark会话对象，绕过了存储过程执行SQL的限制，能正常完成角色切换。

• 用Snowpark API启用二级角色
  如果你的Streamlit运行角色已被授予SECURITYADMIN作为二级角色，直接调用API启用二级角色：

  session.use_secondary_roles("ALL")
  

  注意必须用Snowpark的API，而非SQL的USE SECONDARY ROLES ALL，后者同样会因为存储过程的限制失效。

• 临时权限备选（不推荐）
  如果上述方法无法实现，也可以直接给Streamlit运行角色授予所需的授权权限（比如创建用户、角色的权限），但这不符合你“不提升运行角色权限”的需求，仅作为最后备选。

内容的提问来源于stack exchange，提问作者Ethan Hawkins