ea4all/src/ea4all_mcp/state.py

"""State management for the MCP graph."""
from typing_extensions import TypedDict

from langgraph.prebuilt.chat_agent_executor import AgentState

class InputState(TypedDict):
    """Represents the input state for the graph.

    This class is used to pass the user question to the graph.
    It contains a single field, `question`
    """

    pass

# The index state defines the simple IO for the single-node index graph
class OutputState(TypedDict):
    """Represents the output schema for the graph.
    """

    pass

class OverallState(AgentState, InputState, OutputState):
    """Represents the overall state of the graph.

    This class combines the input and output states, allowing for
    both input and output to be managed within
    the same state.
    """
    pass