diff --git a/agent.py b/agent.py index 53390a4..1349f25 100644 --- a/agent.py +++ b/agent.py @@ -528,11 +528,19 @@ def _priority_rank(priority: str) -> int: return {"urgent": 0, "high": 1, "normal": 2, "low": 3}.get(priority, 9) +def _queue_sort_key(queue_entry: QueueItem) -> int: + return _priority_rank(queue_entry.priority) + + def _summary(queue: list[QueueItem]) -> str: counts: dict[str, int] = {} for queue_entry in queue: counts[queue_entry.priority] = counts.get(queue_entry.priority, 0) + 1 - parts = [f"{counts.get(name, 0)} {name}" for name in ["urgent", "high", "normal", "low"] if counts.get(name, 0)] + parts: list[str] = [] + for priority_name in ("urgent", "high", "normal", "low"): + priority_count = counts.get(priority_name, 0) + if priority_count: + parts.append(f"{priority_count} {priority_name}") return f"Prioritized {len(queue)} people-inbox message(s): " + ", ".join(parts) + "."