如何在LangGraph工具函数中获取当前会话ID？

LangGraph的工具函数在运行时是否有内置方式获取当前会话（session/thread）ID？我需要该ID用于日志记录与外部服务关联，但在函数签名及文档中均未找到相关方法，恳请提供指导。

from langchain_core.tools import tool
from pydantic import BaseModel

class UserAuthStatusInput(BaseModel):
    xxx: str

@tool(args_schema=UserAuthStatusInput)
def xxx_func(xxx: str) -> str:
    # 业务逻辑...
    # TODO: 在此处如何获取当前对话/会话ID？
    return "some result"

LangGraph本身没有内置的直接在工具函数中获取会话ID的方法，但可以通过以下两种方式实现：

• 方法一：将会话ID作为工具参数传入
  修改工具的输入Schema，新增thread_id字段，在调用工具时从会话上下文里提取并传入该值。示例代码如下：

  from langchain_core.tools import tool
  from pydantic import BaseModel
  
  class UserAuthStatusInput(BaseModel):
      xxx: str
      thread_id: str  # 新增会话ID字段
  
  @tool(args_schema=UserAuthStatusInput)
  def xxx_func(xxx: str, thread_id: str) -> str:
      # 可直接使用thread_id进行日志或外部服务关联
      print(f"当前会话ID: {thread_id}")
      # 业务逻辑...
      return "some result"
  

  在LangGraph的运行逻辑中，调用该工具时需从会话状态或外部传入thread_id参数。

• 方法二：通过LangGraph的状态（State）传递
  定义包含会话ID的Graph State，在工具调用时从State中读取会话ID。示例代码如下：

  from langchain_core.tools import tool
  from pydantic import BaseModel, Field
  from langgraph.graph import StateGraph, MessagesState
  
  # 自定义包含会话ID的状态
  class CustomState(MessagesState):
      thread_id: str = Field(description="当前会话ID")
  
  class UserAuthStatusInput(BaseModel):
      xxx: str
  
  @tool(args_schema=UserAuthStatusInput)
  def xxx_func(xxx: str, state: CustomState) -> str:
      # 从状态中获取会话ID
      thread_id = state.thread_id
      print(f"当前会话ID: {thread_id}")
      # 业务逻辑...
      return "some result"
  
  # 构建Graph时绑定状态
  graph_builder = StateGraph(CustomState)
  # 添加节点、边等逻辑...
  

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