How to Build an AI Agent
How to Build an AI Agent
s03

TodoWrite

Планирование и координация

Plan Before You Act

176 LOC5 инструментовTodoManager + nag reminder
An agent without a plan drifts; list the steps first, then execute

s01 > s02 > [ s03 ] s04 > s05 > s06 | s07 > s08 > s09 > s10 > s11 > s12

"Агент без плана блуждает" -- сначала перечислите шаги, затем выполняйте.

Проблема

При многошаговых задачах модель теряет след. Она повторяет работу, пропускает шаги или отклоняется. Длинные разговоры усугубляют это — системный промпт затухает, когда контекст заполняется результатами инструментов. 10-шаговый рефакторинг может завершить шаги 1-3, затем модель начинает импровизировать, забыв шаги 4-10.

Решение

+--------+      +-------+      +---------+
|  User  | ---> |  LLM  | ---> | Tools   |
| prompt |      |       |      | + todo  |
+--------+      +---+---+      +----+----+
                    ^                |
                    |   tool_result  |
                    +----------------+
                          |
              +-----------+-----------+
              | Состояние TodoManager |
              | [ ] задача A          |
              | [>] задача B  <- делает |
              | [x] задача C          |
              +-----------------------+
                          |
              if rounds_since_todo >= 3:
                inject <reminder> into tool_result

Как Это Работает

  1. TodoManager хранит элементы со статусами. Только один элемент может быть in_progress одновременно.
class TodoManager:
    def update(self, items: list) -> str:
        validated, in_progress_count = [], 0
        for item in items:
            status = item.get("status", "pending")
            if status == "in_progress":
                in_progress_count += 1
            validated.append({"id": item["id"], "text": item["text"],
                              "status": status})
        if in_progress_count > 1:
            raise ValueError("Only one task can be in_progress")
        self.items = validated
        return self.render()
  1. Инструмент todo добавляется в карту диспетчеризации как любой другой инструмент.
TOOL_HANDLERS = {
    # ...base tools...
    "todo": lambda **kw: TODO.update(kw["items"]),
}
  1. Напоминание "nag" вставляет подсказку, если модель делает 3+ раунда без вызова todo.
if rounds_since_todo >= 3 and messages:
    last = messages[-1]
    if last["role"] == "user" and isinstance(last.get("content"), list):
        last["content"].insert(0, {
            "type": "text",
            "text": "<reminder>Update your todos.</reminder>",
        })

Ограничение "один in_progress за раз" вынуждает к последовательному фокусу. Напоминание создаёт подотчётность.

Что Изменилось с s02

КомпонентДо (s02)После (s03)
Инструменты45 (+todo)
ПланированиеНетTodoManager со статусами
Вставка nagНет<reminder> после 3 раундов
Цикл агентаПростая диспетчеризация+ счётчик rounds_since_todo

Попробуйте

python agents/s03_todo_write.py
  1. Рефакторить файл hello.py: добавить type hints, docstrings и main guard
  2. Создать Python-пакет с __init__.py, utils.py и tests/test_utils.py
  3. Проверить все Python-файлы и исправить проблемы со стилем