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

class RAGState(TypedDict):
    messages: Annotated[List[BaseMessage], add_messages]
    documents: List[str]
    context: str