from typing import TypedDict, Literal

class ReflectState(TypedDict):
    """Состояние агента с рефлексией"""
    question: str           # Исходный вопрос пользователя
    draft: str              # Текущая версия ответа
    critique: str           # Замечания критика
    verdict: Literal["ok", "needs_revision"]  # Решение критика
    round: int              # Текущий раунд доработки
    max_rounds: int         # Максимальное количество раундов