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