diff --git a/state.py b/state.py index f5053e1..f2895d9 100644 --- a/state.py +++ b/state.py @@ -1,9 +1,9 @@ -from typing import TypedDict, List, Optional +from typing import TypedDict, Optional -class BriefState(TypedDict): - """Состояние исследовательского брифа""" - topic: str # Тема исследования - outline: Optional[List[str]] # План из 4-5 пунктов - step_index: int # Текущий индекс исследуемого пункта - notes: List[str] # Собранные заметки по пунктам - final_brief: Optional[str] # Финальный связный бриф \ No newline at end of file +class ReflectState(TypedDict): + """Состояние агента с retry через try/except""" + question: str # Исходный вопрос пользователя + draft: str # Текущая версия ответа + error: Optional[str] # Ошибка, если была + round: int # Текущий раунд попытки + max_rounds: int # Максимальное количество попыток \ No newline at end of file