from typing import TypedDict, Literal, Optional

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