state.py

# state.py

from typing import TypedDict, Optional
from langchain_core.messages import BaseMessage

class AgentState(TypedDict):
    """
    The single source of truth that travels
    through every node in the graph.

    Every field starts empty and gets filled
    as the graph progresses node by node.
    """

    # ── Input ──────────────────────────────
    topic:               str

    # ── Conversation history ───────────────
    # Pure LangGraph manages this directly
    # CrewAI was hiding this from us before
    messages:            list[BaseMessage]

    # ── Research stage ─────────────────────
    research_notes:      Optional[str]
    research_attempts:   int
    research_passed:     bool

    # ── Concepts stage ─────────────────────
    concepts:            Optional[str]

    # ── Study guide stage ──────────────────
    study_guide:         Optional[str]
    guide_attempts:      int
    guide_passed:        bool

    # ── Quiz stage ─────────────────────────
    quiz:                Optional[str]

    # ── Meta ───────────────────────────────
    current_node:        str
    errors:              list[str]