import operator
from typing import Annotated, Sequence, TypedDict, Union, Optional
from langchain_core.messages import BaseMessage


class AgentState(TypedDict):
    # messages 存储完整的对话历史，operator.add 表示新消息是追加而不是覆盖
    messages: Annotated[Sequence[BaseMessage], operator.add]
    # next 存储 Supervisor 决定的下一步是谁
    next: str
    require_suggestion: bool  # 是否需要建议按钮