src/core/graph_state.py

from typing import List, TypedDict
from langchain_core.documents import Document

class GraphState(TypedDict):
    question: str
    context: List[Document]
    answer: str