from typing_extensions import NotRequired, TypedDict

from langgraph.graph import END, START, StateGraph


class WorkflowState(TypedDict):
    text: str
    word_count: NotRequired[int]