"""
LangGraph State Definition
"""

from typing import TypedDict, Annotated, List

class MathAgentState(TypedDict):
    """Represents the state of the math orchestration pipeline."""
    
    # Input
    user_query: str
    uid: str
    session_id: str
    
    # Planner output
    question_type: str
    class_level: int
    
    # Retrievals (Running in parallel)
    rag_context: str
    history: str
    weak_topics: List[str]
    
    # Solver Output
    raw_solution: str
    
    # Verifier Feedback
    is_correct: bool
    verification_feedback: str
    retries: int
    
    # Formatter Output
    final_answer: str