app/agent/state.py

from typing import TypedDict, List
from app.models.question import QuestionOut, Answer
from app.models.checklist import ChecklistItem


class AgentState(TypedDict):
    session_id: str
    current_round: int
    max_rounds: int
    current_questions: List[QuestionOut]
    all_answers: List[Answer]
    round_summaries: List[str]
    checklist_items: List[ChecklistItem]
    markdown_content: str
    is_complete: bool