test_cus_state.py 602 B

123456789101112131415161718192021
  1. from itertools import count
  2. from langgraph.graph import MessagesState
  3. class TestCusState(MessagesState):
  4. """
  5. 第二步:定义状态结构
  6. """
  7. route_next: str # 下一个节点
  8. session_id: str # 会话id
  9. trace_id: str # 日志链路跟踪id
  10. user_input: str # 用户输入问题
  11. context: str # 上下文数据
  12. task_prompt_info: str # 任务提示