from itertools import count
from langgraph.graph import MessagesState




class TestCusState(MessagesState):
    """
     第二步：定义状态结构
    """
    route_next: str                                  # 下一个节点  
    
    session_id: str                                  # 会话id  
    trace_id: str                                    # 日志链路跟踪id
    user_input: str                                  # 用户输入问题    
    context: str                                     # 上下文数据
    task_prompt_info: str                            # 任务提示