"""LangGraph state definitions for agent workflows."""

from typing import TypedDict, List, Annotated, Optional
from langgraph.graph.message import add_messages
from langchain_core.messages import BaseMessage


class AgentState(TypedDict):
    """State for agent graph."""
    messages: Annotated[List[BaseMessage], add_messages]
    user_id: str
    room_id: str
    retrieved_docs: List[dict]
    needs_search: bool