Teknium
2ad284b473
- atlas: _border_flood/_unvisited_components/_place/_clear_region/_remove_masked replace 6 hand-rolled flood + canvas-composite loops; _sever_expected_gutters uses channel ops instead of per-pixel writes; dead _color_distance, _has_slot_padding, _slot_bounds, atlas_to_webp_bytes, FRAME_COUNTS (0 refs). - render/store: dead _open_sheet/_png_bytes/_union_alpha_bbox/_thumbs_dir removed. - imagegen: _available() unifies 3 provider-availability checks; forced/preferred provider resolution collapsed into one ordered loop (same precedence). - state: derive_pet_state as a ranked table (same priority order). - Comment/docstring compaction keeping the pipeline invariants.
77 lines
2.8 KiB
Python
77 lines
2.8 KiB
Python
"""Map agent activity → a :class:`PetState`.
|
|
|
|
This is the one place the "what is the agent doing right now?" → "which
|
|
animation row?" decision lives. Each surface feeds it the signals it already
|
|
tracks:
|
|
|
|
- CLI — ``KawaiiSpinner`` waiting/thinking state + tool outcomes.
|
|
- TUI — gateway ``tool.start/complete`` + ``message.delta/complete`` events.
|
|
- Desktop — the ``$busy``/``$awaitingResponse``/tool-event nanostores
|
|
(re-implemented in TS, but mirroring this priority order).
|
|
|
|
Keeping the priority order here (and documenting it) lets the TypeScript
|
|
mirror stay faithful without a second design.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from collections.abc import Iterable
|
|
from typing import Any
|
|
|
|
from agent.pet.constants import PetState
|
|
|
|
|
|
def todos_all_done(todos: Iterable[Any] | None) -> bool:
|
|
"""True iff there's ≥1 todo and every one is completed/cancelled.
|
|
|
|
The "celebrate" beat (``JUMP``) fires when a plan finishes; this mirrors
|
|
the TUI's ``isTodoDone`` so the trigger is defined once across surfaces.
|
|
Accepts dicts (``{"status": ...}``) or objects with a ``status`` attr.
|
|
"""
|
|
items = list(todos or [])
|
|
if not items:
|
|
return False
|
|
|
|
def _status(t: Any) -> Any:
|
|
return t.get("status") if isinstance(t, dict) else getattr(t, "status", None)
|
|
|
|
return all(_status(t) in ("completed", "cancelled") for t in items)
|
|
|
|
|
|
def derive_pet_state(
|
|
*,
|
|
busy: bool = False,
|
|
awaiting_input: bool = False,
|
|
error: bool = False,
|
|
celebrate: bool = False,
|
|
just_completed: bool = False,
|
|
tool_running: bool = False,
|
|
reasoning: bool = False,
|
|
) -> PetState:
|
|
"""Resolve the animation state from coarse activity signals.
|
|
|
|
Priority (highest first) — only one row can show at a time, so the most
|
|
salient signal wins:
|
|
|
|
1. ``error`` → ``FAILED`` (a tool/turn just failed)
|
|
2. ``celebrate`` → ``JUMP`` (explicit success beat, e.g. todos done)
|
|
3. ``just_completed`` → ``WAVE`` (turn finished cleanly / greeting)
|
|
4. ``awaiting_input`` → ``WAITING`` (blocked on the user — a clarify/approval
|
|
prompt is open; this outranks the in-flight signals below because the turn
|
|
is paused on *you*, even though a tool is technically mid-call)
|
|
5. ``tool_running`` → ``RUN`` (a tool is executing)
|
|
6. ``reasoning`` → ``REVIEW`` (model is thinking / reading)
|
|
7. ``busy`` → ``RUN`` (turn in flight, unspecified work)
|
|
8. otherwise → ``IDLE``
|
|
"""
|
|
ranked = (
|
|
(error, PetState.FAILED),
|
|
(celebrate, PetState.JUMP),
|
|
(just_completed, PetState.WAVE),
|
|
(awaiting_input, PetState.WAITING),
|
|
(tool_running, PetState.RUN),
|
|
(reasoning, PetState.REVIEW),
|
|
(busy, PetState.RUN),
|
|
)
|
|
return next((state for flag, state in ranked if flag), PetState.IDLE)
|