from typing import Any, Dict, List
from langchain_core.messages import BaseMessage


class AgentState:
    """代理工作流的状态定义"""
    messages: List[BaseMessage]
    current_step: str
    context: Dict[str, Any]

    def __init__(self, messages: List[BaseMessage] = None, current_step: str = "start", context: Dict[str, Any] = None):
        self.messages = messages or []
        self.current_step = current_step
        self.context = context or {}