LangGraph无法访问Streamlit的st.experimental_user.email属性的问题排查与解决方案咨询

看起来你在LangGraph和Streamlit集成时碰到了一个挺典型的上下文隔离问题——LangGraph的工具函数没法访问Streamlit会话里的st.experimental_user.email，我来帮你梳理下问题根源和可行的解决思路。

问题核心原因 #

你猜的没错，问题出在上下文隔离上：Streamlit的st.experimental_user是绑定在当前用户的Streamlit会话上下文里的，只有在Streamlit主流程的请求/响应周期内才能正常访问。而LangGraph的工具函数运行时，可能是在独立的执行上下文（比如异步线程、或者脱离了Streamlit会话生命周期的环境）里，这时候直接去访问st.experimental_user.email自然会拿不到值，抛出KeyError。你测试时其他函数能打印，是因为那些函数本身就在Streamlit的主会话流程里，没有脱离上下文。

可行的解决方案 #

1. 提前把用户信息注入LangGraph的状态（最稳妥） #

不要让LangGraph工具直接去碰Streamlit的对象，而是在初始化LangGraph之前，先把用户信息提取出来，作为参数或状态传递给LangGraph：

• 第一步，在Streamlit主流程里先获取并校验用户信息：

  # 先处理用户登录逻辑
  if not st.experimental_user.is_authenticated:
      st.warning("请先登录以使用高级功能")
      if st.button("登录"):
          st.experimental_user.login()
      st.stop()
  # 提取用户邮箱
  user_email = st.experimental_user.email
  

• 第二步，把用户邮箱存入LangGraph的自定义状态中，让工具从这个状态里取数据：

  from langgraph.graph import StateGraph, MessagesState
  
  # 定义包含用户信息的自定义状态
  class ResumeBotState(MessagesState):
      user_email: str
  
  # 初始化LangGraph时传入用户邮箱
  initial_state = ResumeBotState(user_email=user_email)
  
  # 改写你的get_user_info工具，从LangGraph状态里拿邮箱
  def get_user_info(state: ResumeBotState):
      user_email = state.user_email
      # 接下来执行你的数据库查询逻辑即可
  

这种方式彻底让LangGraph和Streamlit的上下文解耦，工具只依赖LangGraph自己的状态，不会再出现上下文访问不到的问题。

2. 用闭包捕获Streamlit的session_state #

如果不想修改LangGraph的状态结构，可以在定义LangGraph工具时，用闭包的方式提前捕获Streamlit会话里的用户信息：

# 在Streamlit主流程中先把用户邮箱存入session_state
if st.experimental_user.is_authenticated:
    st.session_state["user_email"] = st.experimental_user.email

# 定义工具时用闭包捕获session_state里的值
def get_user_info():
    user_email = st.session_state.get("user_email")
    if not user_email:
        raise ValueError("用户未登录，请先完成登录")
    # 后续的数据库操作逻辑

不过要注意，这个方法只适用于LangGraph工具在同一个Streamlit会话上下文里同步执行的场景，如果你的LangGraph用了异步或者多线程，可能还是会有上下文丢失的问题，所以优先推荐第一种方法。

3. 手动获取Streamlit会话上下文（不推荐，依赖内部API） #

如果前面两种方法都不适用，可以尝试手动获取Streamlit的会话上下文，但这个方法依赖Streamlit的内部实现，版本更新可能会失效：

from streamlit.runtime.scriptrunner import get_script_run_ctx

def get_user_info():
    ctx = get_script_run_ctx()
    if not ctx:
        raise ValueError("无法获取当前Streamlit会话上下文")
    # 从上下文的session_state里取用户邮箱
    user_email = ctx.session_state.get("user_email")
    if not user_email:
        raise ValueError("用户未登录")
    # 后续逻辑

额外的优化建议 #

• 尽量替换st.experimental_user为st.session_state["user"]，因为带experimental前缀的API是测试版，后续可能会有变动，Streamlit官方现在更推荐用st.session_state来管理用户会话信息（OAuth登录后，用户信息会自动存入st.session_state["user"]）。
• 任何时候访问用户邮箱前，都先判断用户是否已认证，比如用st.experimental_user.is_authenticated来做前置校验，避免直接访问email属性导致的KeyError。

内容来源于stack exchange