File size: 418 Bytes
bef5e76 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 | """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
|