A thinking block that says how long and how much

Each block reports its own round now. `reasoning_ms` was the reply's first
burst, written once, so on the fifteen-block reply GPT-OSS actually produces
only the first could claim a duration and the other fourteen said "Thought" and
nothing at all. `Generation.thinking_ms` accumulates per round and `close_step`
stamps it cumulatively, so steps.py diffs it exactly as it already diffs the
three lengths beside it.

The interval between a round's first and last reasoning delta, deliberately, not
a sum of gaps between deltas -- that would count the network's latency as the
model's thinking.

While it runs: "Thinking" with an ellipsis that types itself, and the seconds
and tokens climbing beside it. The ellipsis is a `content` keyframe, so there is
no timer to start, stop or clean up when the block is swapped away -- it stops
existing when the element does. The numbers come from a `think` frame, and
`round_thinking_ms` is written by the producer rather than computed by the
follower from a start time: a model that has stopped thinking and moved on to a
tool should show a settled number, not a clock that keeps running.

Tokens read exactly up to 200 and as `0.4k` above it, from one helper shared by
the live label and the stored one, so the two cannot drift into two conventions.
The live duration is terser than the finished one -- `6s` against `6 seconds` --
because it sits beside an animating word and changes every second, where "less
than a second" flickering into "1 second" reads as a glitch.

Checked against the real endpoint: fourteen marks carrying 919ms through
14223ms, per-block labels from "less than a second · 111" to "4 seconds · 0.5k",
and the live frames resetting each round rather than accumulating.

Co-Authored-By: Claude Opus 5 (1M context) <noreply@anthropic.com>
This commit is contained in:
Jaroslav Beneš
2026-08-04 21:03:25 +02:00
parent 74dd19588b
commit 30ddcba787
9 changed files with 385 additions and 10 deletions
+20
View File
@@ -38,6 +38,7 @@ from lembas.services import interaction, settings_store, sse
from lembas.services import metrics as metrics_service
from lembas.services import prompts as prompts_service
from lembas.services import steps as steps_service
from lembas.services import tokens as tokens_service
from lembas.services import tools as tools_service
from lembas.services.agent import policy as agent_policy
from lembas.services.agent import terminal as terminal_service
@@ -855,6 +856,24 @@ def _step_html(message_id: str, step) -> str:
)
def _think_label(generation, thinking_tail: str) -> str:
"""How long this round has been thinking, and roughly how much.
This round's, not the reply's, so the live block means the same thing as the
closed blocks above it and does not change meaning the moment it settles.
The reply's total is already under the bubble, in the metrics chips.
The producer owns the number. Computing it here from a start time would
keep the clock running after the model had stopped thinking and moved on to
a tool, which is a timer rather than a measurement.
"""
return steps_service.thinking_label(
ms=generation.round_thinking_ms,
tokens=tokens_service.estimate(thinking_tail),
live=True,
)
def _ask_html(chat_id: str, pending) -> str:
"""The card asking the reader something, or nothing at all.
@@ -946,6 +965,7 @@ async def _follow(chat_id: str, message_id: str) -> AsyncIterator[str]:
# guarded, or a frame could wipe the answer.
thinking_tail, text_tail = steps_service.tail(generation)
yield sse.event("reasoning", escape_text(thinking_tail))
yield sse.event("think", escape_text(_think_label(generation, thinking_tail)))
yield sse.event("render", render_markdown(text_tail) if text_tail else "")
if generation.canvas.get("tabs"):
# Guarded on truthiness, which puts this in the
+58 -1
View File
@@ -103,6 +103,17 @@ class Generation:
content: list[str] = field(default_factory=list)
reasoning: list[str] = field(default_factory=list)
reasoning_ms: int = 0
# Milliseconds spent thinking, summed over rounds. Distinct from
# `reasoning_ms`, which is the reply's *first* burst and is written once --
# right for "Thought for 8 seconds" on a single-round answer, and unable to
# say anything about round seven of forty. Stamped on each mark by
# `close_step` and diffed by services/steps.py into a per-block figure.
thinking_ms: int = 0
# How long the round *currently* running has been thinking. Read by the live
# block's label, and written by the producer rather than computed from a
# start time by the follower: a model that has stopped thinking and moved on
# to a tool should show a settled number, not a clock that keeps running.
round_thinking_ms: int = 0
# One entry per tool call made while producing this reply, in order. Shown
# live as the model works and kept on the message afterwards.
@@ -234,6 +245,8 @@ class Generation:
"thinking_to": len(self.thinking),
"text_to": len(self.text),
"tools_to": len(self.tool_events),
# Cumulative, like the three above it, and diffed the same way.
"thinking_ms": self.thinking_ms,
}
)
@@ -554,6 +567,13 @@ async def _run(generation: Generation) -> None:
# Text the model produced in *this* round, needed separately from
# generation.content when echoing the assistant turn back.
round_text: list[str] = []
# When this round's thinking started and when it was last seen, so
# the interval can be added to `generation.thinking_ms` at the
# round's end. Per round, because the thinking block is per round:
# `reasoning_ms` is the whole reply's first burst, written once, and
# cannot say how long round seven thought for. `None` until the
# round thinks at all -- plenty of rounds do not.
round_thinking: tuple[float, float] | None = None
async for chunk in stream_chat(endpoint, payload):
counts = chunk_usage(chunk)
@@ -572,6 +592,7 @@ async def _run(generation: Generation) -> None:
if thought:
if reasoning_started is None:
reasoning_started = time.monotonic()
round_thinking = _thought_at(generation, round_thinking)
generation.reasoning.append(thought)
generation.touch()
@@ -586,6 +607,7 @@ async def _run(generation: Generation) -> None:
if kind == REASONING:
if reasoning_started is None:
reasoning_started = time.monotonic()
round_thinking = _thought_at(generation, round_thinking)
generation.reasoning.append(piece)
else:
if reasoning_started is not None and not generation.reasoning_ms:
@@ -624,6 +646,8 @@ async def _run(generation: Generation) -> None:
# output, so the mark belongs at the round's end.
# See metrics._since_counted.
_mark_counted(generation)
# Before `close_step` below, which stamps the total this adds to.
round_thinking = _close_thinking(generation, round_thinking)
calls = accumulator.calls
if generation.stopped or not calls:
@@ -1218,6 +1242,35 @@ def _too_big(generation: Generation) -> bool:
return generation.prompt_estimate > generation.context_limit * CONTEXT_HEADROOM
def _thought_at(generation: Generation, span: tuple[float, float] | None) -> tuple[float, float]:
"""Widen this round's thinking interval to now.
First call in a round opens it; every later one moves its end. The interval
rather than a running sum, because reasoning arrives in a burst of small
deltas and adding a gap per delta would count the network's latency as the
model's thinking.
"""
now = time.monotonic()
span = (now, now) if span is None else (span[0], now)
generation.round_thinking_ms = int((span[1] - span[0]) * 1000)
return span
def _close_thinking(
generation: Generation, span: tuple[float, float] | None
) -> tuple[float, float] | None:
"""Add this round's thinking to the reply's total. Returns None to reopen.
Called where the round ends, so `close_step` can stamp a cumulative figure
that `services/steps.py` diffs into a per-block duration -- the same shape
as the three lengths it already stamps.
"""
if span is not None:
generation.thinking_ms += int((span[1] - span[0]) * 1000)
generation.round_thinking_ms = 0
return None
def _mark_counted(generation: Generation) -> None:
"""Record that everything written so far is covered by a reported count.
@@ -1848,7 +1901,11 @@ def _persist(generation: Generation, title: str, elapsed: float) -> None:
message.content = generation.text
message.reasoning = generation.thinking
message.reasoning_ms = generation.reasoning_ms
# `thinking_ms` in preference: it is the same measurement done
# properly, summed over every round rather than stopping at the
# first burst, and it is what the trailing block's duration is
# derived from. Falls back for a reply that produced no marks.
message.reasoning_ms = generation.thinking_ms or generation.reasoning_ms
message.tool_calls_json = generation.tool_events
# Written together with the three stores it indexes, by the one
# writer, so a row can never carry marks that describe a different
+112 -3
View File
@@ -24,12 +24,18 @@ from __future__ import annotations
from dataclasses import dataclass, field
from typing import Any
from lembas.services import tokens as tokens_service
from lembas.services.markdown import open_fence, render_markdown
from lembas.services.reasoning import format_duration
KIND_THINKING = "thinking"
KIND_TEXT = "text"
KIND_TOOLS = "tools"
# Below this a token count is printed exactly; above it, as `1.4k`. The point is
# where the digits stop meaning anything to a reader.
TOKENS_EXACT_BELOW = 200
@dataclass(frozen=True)
class Step:
@@ -49,6 +55,56 @@ class Step:
text: str = ""
html: str = ""
events: tuple[dict, ...] = field(default_factory=tuple)
# How long this block thought for and roughly how much it produced. Only on
# a thinking step, and only where there is something to say -- a block whose
# duration was never recorded (every row written before the marks carried
# one) shows the word alone rather than "Thought for 0 seconds".
label: str = ""
def format_tokens(count: int) -> str:
"""A token count as a person reads it: `86`, or `0.2k` once it is worth it.
The threshold is where the exact number stops carrying information: nobody
acts on the difference between 214 and 219 tokens of thinking, and four
digits beside a spinner is noise. One helper, so the live label and the
stored one cannot drift into two conventions.
"""
if count <= 0:
return ""
if count <= TOKENS_EXACT_BELOW:
return str(count)
return f"{count / 1000:.1f}k"
def thinking_label(*, ms: int, tokens: int, live: bool) -> str:
"""What a thinking block says about itself.
The two states are one function because they are one sentence with a
different tense, and because the running one becomes the finished one in
place -- a reader watching the numbers climb should see them settle, not be
replaced by something formatted differently.
The word itself is not here: the live block animates its own ellipsis in CSS
and the template owns that. This is only what follows it.
"""
parts = []
if ms > 0:
parts.append(format_duration(ms) if not live else _short_duration(ms))
if counted := format_tokens(tokens):
parts.append(counted)
return " · ".join(parts)
def _short_duration(ms: int) -> str:
"""`6s`, `1m 04s`. Terser than `format_duration` because it sits beside an
animating word and changes every second; "less than a second" flickering
into "1 second" reads as a glitch rather than as a measurement."""
seconds = max(0, ms) // 1000
if seconds < 60:
return f"{seconds}s"
minutes, remainder = divmod(seconds, 60)
return f"{minutes}m {remainder:02d}s"
def for_message(message: Any) -> list[Step]:
@@ -63,6 +119,7 @@ def for_message(message: Any) -> list[Step]:
thinking=(message.reasoning or "") if not message.error else "",
events=list(message.tool_calls_json or []),
marks=list(getattr(message, "steps_json", None) or []),
whole_ms=int(getattr(message, "reasoning_ms", 0) or 0),
# A stored reply has a trailing step and it is not being written. Those
# are two facts and they used to be one flag: `include_open` both
# emitted the tail and marked it live, so every finished bubble ending
@@ -118,6 +175,7 @@ def _build(
since: int = 0,
include_open: bool = True,
live: bool = True,
whole_ms: int = 0,
) -> list[Step]:
"""The shared walk.
@@ -141,7 +199,26 @@ def _build(
# nothing: for that one the two orders are the same list, because there
# are no tool blocks to sit between the prose.
if thinking:
steps.append(Step(index=0, kind=KIND_THINKING, text=thinking))
# No marks means no per-round timing was ever recorded, so the
# duration falls back to whatever the row knows about the reply as a
# whole -- which for a single-round answer is exactly right, and is
# what every row written before the marks existed carries.
steps.append(
Step(
index=0,
kind=KIND_THINKING,
text=thinking,
# Nothing baked in while the reply runs: the live block's
# numbers come from the `think` frame, which knows the
# clock. A label rendered here would be one that never
# moved again.
label=""
if live
else thinking_label(
ms=whole_ms, tokens=tokens_service.estimate(thinking), live=False
),
)
)
if events:
steps.append(Step(index=0, kind=KIND_TOOLS, events=tuple(events)))
if text:
@@ -153,6 +230,7 @@ def _build(
thought_from = 0
text_from = 0
tools_from = 0
thought_ms_from = 0
carry = ""
for index, mark in enumerate(marks):
@@ -175,7 +253,18 @@ def _build(
if index >= since:
# Thinking, then prose, then tools -- the order a model emits them.
if thought:
steps.append(Step(index=index, kind=KIND_THINKING, text=thought))
steps.append(
Step(
index=index,
kind=KIND_THINKING,
text=thought,
label=thinking_label(
ms=_at(mark, "thinking_ms") - thought_ms_from,
tokens=tokens_service.estimate(thought),
live=False,
),
)
)
if said:
steps.append(Step(index=index, kind=KIND_TEXT, html=render_markdown(source)))
if ran:
@@ -183,6 +272,9 @@ def _build(
carry = f"{marker}{info}" if marker else ""
thought_from, text_from, tools_from = thought_to, text_to, tools_to
# Cumulative on the mark, so a block's own duration is the difference --
# the same rule the three lengths above follow.
thought_ms_from = max(_at(mark, "thinking_ms"), thought_ms_from)
if not include_open:
return steps
@@ -191,7 +283,24 @@ def _build(
# live path and the stored one -- one rule instead of two that could drift.
index = len(marks)
if trailing_thought := thinking[thought_from:]:
steps.append(Step(index=index, kind=KIND_THINKING, text=trailing_thought))
# The last round closes no mark -- it is the round that stopped calling
# tools -- so its duration is whatever the reply spent thinking beyond
# the last one that did. Zero while the reply is running, where the live
# block carries its own label instead.
steps.append(
Step(
index=index,
kind=KIND_THINKING,
text=trailing_thought,
label=""
if live
else thinking_label(
ms=max(0, whole_ms - thought_ms_from),
tokens=tokens_service.estimate(trailing_thought),
live=False,
),
)
)
if trailing_text := text[text_from:]:
source = f"{carry}\n{trailing_text}" if carry else trailing_text
steps.append(
+30
View File
@@ -181,6 +181,36 @@
.reasoning__summary::-webkit-details-marker { display: none; }
.reasoning__summary:hover { color: var(--ink); background: var(--surface-hover); }
/* "Thinking" with an ellipsis that types itself: `.` `..` `...`, on a step
timer so it lands on whole dots rather than sliding. Animating `content` is
the only way to do this without a JavaScript timer to start, stop and clean
up when the block is swapped away -- and this one simply stops existing when
the element does.
`min-width` on the pseudo-element so the label does not jog left and right as
the dots come and go; `ch` because it is exactly three dot-widths. */
.reasoning__working::after {
content: "...";
display: inline-block;
min-width: 1.6ch;
text-align: left;
animation: thinking-dots 1.5s steps(1) infinite;
}
@keyframes thinking-dots {
0% { content: "."; }
33% { content: ".."; }
66% { content: "..."; }
}
/* The seconds and the token count. Tabular figures so a climbing number does
not shift the text beside it on every tick. Empty until the first frame
arrives, and `:empty` keeps its separator from showing before that. */
.reasoning__stats {
color: var(--ink-faint);
font-variant-numeric: tabular-nums;
}
.reasoning__stats:not(:empty)::before { content: " · "; }
/* --- Compacted turns -------------------------------------------------------
Summarised messages, kept and readable but out of the way. Collapsed by
default: the point of compacting was that they stopped mattering.
+11 -1
View File
@@ -112,10 +112,20 @@
tail only, so an empty one means the tail is genuinely empty, whereas
the version that carried the whole reply would have wiped it. `steps`
is the one that must never blank now. #}
{# The word animates its own ellipsis in CSS -- `.` `..` `...` -- so there
is no timer to start or clean up, and it stops when the element does.
The numbers beside it come from the `think` frame.
That span is a SIBLING of the body, inside a `<details>` that is never
itself swapped. Two swap targets in one static container is fine; a
swap target inside another is what blanked every agent chat. #}
<details class="reasoning reasoning--live" id="reasoning-{{ message.id }}">
<summary class="reasoning__summary">
{{ icon("sparkle", "icon--sm reasoning__icon") }}
<span class="reasoning__label">Thinking…</span>
<span class="reasoning__label">
<span class="reasoning__working">Thinking</span>
<span class="reasoning__stats" sse-swap="think" hx-swap="innerHTML"></span>
</span>
{{ icon("chevron-down", "icon--sm reasoning__chevron") }}
</summary>
<div class="reasoning__body" sse-swap="reasoning" hx-swap="innerHTML"></div>
+7 -5
View File
@@ -19,12 +19,14 @@
<details class="reasoning" id="think-{{ message.id }}-{{ step.index }}">
<summary class="reasoning__summary">
{{ icon("sparkle", "icon--sm reasoning__icon") }}
{# Each block reports its own round: `step.label` is built from that
block's slice of the thinking and the interval between its mark and the
one before it. It used to read the reply's total, which meant only the
first block could honestly claim it and the other eleven said "Thought"
and nothing else. #}
<span class="reasoning__label">
{% if step.index == 0 and message.reasoning_ms %}
{# The duration is for the whole reply, so only the first block may
claim it. Repeating it on each would be four blocks each saying
they took ninety seconds. #}
Thought for {{ message.reasoning_ms | duration }}
{% if step.label %}
Thought for {{ step.label }}
{% else %}
Thought
{% endif %}