from typing import Annotated
from typing_extensions import TypedDict
from langgraph.graph.message import add_messages

class State(TypedDict):
    """
    State for the graph.
    """
    messages: Annotated[list, add_messages]  # List of messages in the graph