| """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 | |