feat(rag): pass chat history with prior thinking to the LLM
Build and Push Containers / build-and-push-app (push) Successful in 1m39s
Build and Push Containers / build-and-push-db (push) Successful in 11s

Phase 74 (TODO.md L4): a follow-up question now reaches the model WITH
the conversation so far — every prior user/brain turn and the prior
thinking blocks on brain turns (preserve-thinking) — while
POST /api/chat stays stateless (A10): the client provides the history
in the request body and the server stores nothing new.

Server (task 01):
- ChatRequest.history: optional list[HistoryTurn] (who: user|brain,
  text, optional thinking) — absent/empty keeps the request
  byte-identical to pre-phase-74 (the two-message [system, user]
  request; the kill-switch semantics are pinned in the integration
  suite).
- app.rag.prompts.history_to_messages: pure mapper — walks the turns
  newest-first against the settings budgets (history_max_turns=40 /
  history_max_chars=24000, BOR_HISTORY_MAX_TURNS /
  BOR_HISTORY_MAX_CHARS); a capped turn is dropped WHOLE (never cut
  mid-answer); the kept window is returned oldest-first; brain turns
  carry their thinking as reasoning_content (A4) only when
  non-empty.
- Both branches feed it: the deflected path splices it between the
  system prompt and the current user message (the phase-71 recovery
  still rebuilds from messages[1:]), the grounded agent receives
  run_agent(..., history=hist); llm.py's message params widen to
  list[dict[str, Any]] (string-only messages stay byte-identical on
  the wire — the SDK passes message dicts through verbatim).
- The per-turn log line (PLAN §9) gains history_msgs=N after
  kb_chars=N.
- Pins: tests/unit/test_history.py (mapper: mapping, reasoning
  gating, both budgets, drop-whole, ordering, empty default),
  tests/unit/test_config.py (the two settings + env overrides),
  tests/unit/test_agent.py (the history splice + the default),
  tests/integration/test_chat_api.py (deflected AND grounded forward
  the history incl. reasoning_content, no-history byte-identity, 422
  pins, the log field).

Client (task 02):
- runTurn — the single funnel for fresh send / phase-49 retry /
  phase-53 stale-regen — sends history = the conversation record
  minus the current question, with thinking only on brain records
  that streamed one (undefined drops the key from the JSON, the
  record's convention); the question is never duplicated into the
  history.

Wire proof (task 03):
- The mock's echo my history marker (HISTORY_TRIGGER) answers with
  the deterministic history echo — history: N prior messages; last
  answer tail: <last 24 chars>; thinking: yes|no — checked BEFORE
  the DEFLECT_MODE branch (like TABLE_TRIGGER), so it fires on both
  turn branches whatever the gate says; the module docstring records
  the user/assistant-only history invariant that keeps every
  existing (tool-result-classified) marker flow unaffected.
- tests/e2e/test_llm_history.py (isolated): a grounded follow-up and
  a deflected follow-up both receive history: 2 prior messages +
  thinking: yes + the byte-exact tail of turn 1's answer (derived
  from the persisted bor.chat.v1 record — the same array the client
  maps into the body); a cold start receives history: 0 prior
  messages / last answer tail: none / thinking: no.
- Regressions green in isolation: chat_rag, chat_history (phase 50),
  agent_document_tools, harness_aligned_tools, stop_generation,
  retry_answer, response_to_docs.
This commit is contained in:
2026-09-05 16:04:40 -04:00
parent a16130c71d
commit 055c0b5d85
29 changed files with 1418 additions and 18 deletions
+77 -1
View File
@@ -30,7 +30,7 @@ import asyncio
import json
import logging
import uuid
from collections.abc import AsyncGenerator, AsyncIterator
from collections.abc import AsyncGenerator, AsyncIterator, Sequence
from copy import deepcopy
from typing import TYPE_CHECKING, Any, cast
@@ -115,7 +115,11 @@ async def _run(
holder: AgentHolder,
settings: Settings,
seed_docs: list[Document] | None = None,
history: Sequence[dict[str, Any]] = (),
) -> list[StreamPiece | ToolCallPiece | RetryPiece]:
"""Consume one ``run_agent`` turn; *history* (phase 74) is the
client's prior turns spliced between system and user (default
``()`` — the pre-phase-74 two-message request)."""
out: list[StreamPiece | ToolCallPiece | RetryPiece] = []
async for piece in run_agent(
cast("LLMClient", llm),
@@ -125,6 +129,7 @@ async def _run(
seed_docs=seed_docs or [],
settings=settings,
holder=holder,
history=history,
):
out.append(piece)
return out
@@ -458,6 +463,77 @@ def test_content_and_tool_call_in_one_stream_keeps_both(
assert llm.requests[1][0][3]["content"] == "0 documents:\n"
# ---------- phase 74: client history between system and user ----------
def test_run_agent_default_history_keeps_two_message_request() -> None:
"""No *history* (the default ``()``) → the model sees exactly the
pre-phase-74 two-message request ``[system, user]`` — byte-identical
behavior (owner-locked A2)."""
llm = ScriptedLLM([StreamPiece("content", "the answer")])
asyncio.run(_run(llm, AgentHolder(), _settings()))
(messages, _tools) = llm.requests[0]
assert messages == [
{"role": "system", "content": "SYSTEM_PROMPT"},
{"role": "user", "content": "QUESTION"},
]
def test_run_agent_places_history_between_system_and_user() -> None:
"""A non-empty *history* (the client's prior turns, already mapped by
``history_to_messages``) is spliced between the system prompt and the
CURRENT user message — oldest-first, with the assistant turn's prior
thinking riding on ``reasoning_content`` (A4). The current question
stays LAST."""
history = [
{"role": "user", "content": "old question"},
{
"role": "assistant",
"content": "old answer",
"reasoning_content": "old thinking",
},
]
llm = ScriptedLLM([StreamPiece("content", "the answer")])
pieces = asyncio.run(
_run(llm, AgentHolder(), _settings(), history=history)
)
assert [p for p in pieces if isinstance(p, StreamPiece)] == [
StreamPiece("content", "the answer")
]
(messages, _tools) = llm.requests[0]
assert messages == [
{"role": "system", "content": "SYSTEM_PROMPT"},
{"role": "user", "content": "old question"},
{
"role": "assistant",
"content": "old answer",
"reasoning_content": "old thinking",
},
{"role": "user", "content": "QUESTION"},
]
def test_run_agent_history_survives_a_tool_round(
monkeypatch: pytest.MonkeyPatch,
) -> None:
"""The tool rounds append assistant/tool messages to the SAME
``messages`` list — the prior history stays in place between the
system prompt and the current question on the SECOND request too."""
monkeypatch.setattr(agent, "list_catalog", lambda db: [])
llm = ScriptedLLM(
[ToolCallPiece(id="call_1", name="ls", arguments={})],
[StreamPiece("content", "the answer")],
)
history = [{"role": "assistant", "content": "old answer"}]
asyncio.run(_run(llm, AgentHolder(), _settings(), history=history))
_first, second = llm.requests
assert second[0][:3] == [
{"role": "system", "content": "SYSTEM_PROMPT"},
{"role": "assistant", "content": "old answer"},
{"role": "user", "content": "QUESTION"},
]
# ---------- ls: full catalog + scoping ----------
+42
View File
@@ -147,6 +147,48 @@ def test_llm_retry_delay_rejects_negative(monkeypatch: pytest.MonkeyPatch) -> No
_settings()
def test_history_budget_defaults(monkeypatch: pytest.MonkeyPatch) -> None:
"""Phase 74 (TODO L4): the client-provided history is trimmed
newest-first against the newest 40 turns within a total of
24 000 chars (text + prior thinking, owner-locked A3)."""
monkeypatch.delenv("BOR_HISTORY_MAX_TURNS", raising=False)
monkeypatch.delenv("BOR_HISTORY_MAX_CHARS", raising=False)
s = _settings()
assert s.history_max_turns == 40
assert s.history_max_chars == 24_000
def test_history_budget_env_overrides(monkeypatch: pytest.MonkeyPatch) -> None:
"""``BOR_HISTORY_MAX_TURNS`` / ``BOR_HISTORY_MAX_CHARS`` override the
defaults; ``0`` on either is the no-history kill switch (the
pre-phase-74 two-message requests)."""
monkeypatch.setenv("BOR_HISTORY_MAX_TURNS", "12")
monkeypatch.setenv("BOR_HISTORY_MAX_CHARS", "5000")
s = _settings()
assert s.history_max_turns == 12
assert s.history_max_chars == 5000
monkeypatch.setenv("BOR_HISTORY_MAX_TURNS", "0")
assert _settings().history_max_turns == 0
monkeypatch.setenv("BOR_HISTORY_MAX_CHARS", "0")
assert _settings().history_max_chars == 0
def test_history_max_turns_rejects_negative(monkeypatch: pytest.MonkeyPatch) -> None:
"""``0`` is the no-history kill switch — a negative value is a typo,
so the validator fails loudly at startup (the ``agent_max_rounds``
pattern)."""
monkeypatch.setenv("BOR_HISTORY_MAX_TURNS", "-1")
with pytest.raises(ValidationError, match="history_max_turns"):
_settings()
def test_history_max_chars_rejects_negative(monkeypatch: pytest.MonkeyPatch) -> None:
"""A negative char budget is a typo — fail loudly at startup."""
monkeypatch.setenv("BOR_HISTORY_MAX_CHARS", "-1")
with pytest.raises(ValidationError, match="history_max_chars"):
_settings()
def test_agent_max_rounds_default_and_env_override(monkeypatch: pytest.MonkeyPatch) -> None:
"""Phase 45: the per-tool budgets are gone — ``BOR_AGENT_MAX_ROUNDS``
(default 10) is the single agent-loop knob; ``0`` is the no-tools
+214
View File
@@ -0,0 +1,214 @@
"""Unit: the client-history → model-messages mapper (phase 74, TODO L4).
``app.rag.prompts.history_to_messages`` is pure (no I/O) — every branch
is pinned here: the user/brain role mapping, the ``reasoning_content``
gating (prior thinking travels ONLY when non-empty — the preserve-
thinking wire convention, A4), the turn-count budget (newest kept,
oldest dropped), the char budget (``text`` + ``thinking`` accounted,
drop-WHOLE semantics — never cut mid-answer, A3), the budgets working
together, and the chronological (oldest → newest) order of the result.
"""
from __future__ import annotations
from typing import Any, Literal
from app.config import Settings
from app.rag.prompts import history_to_messages
from app.schemas import HistoryTurn
def _settings(**kwargs: Any) -> Settings:
kwargs.setdefault("_env_file", None)
return Settings(**kwargs) # pyright: ignore[reportCallIssue] (kwarg exists at runtime)
def _turn(
who: Literal["user", "brain"], text: str, thinking: str | None = None
) -> HistoryTurn:
return HistoryTurn(who=who, text=text, thinking=thinking)
# ---------- mapping ----------
def test_empty_history_yields_no_messages() -> None:
"""Absent client history (the pre-phase-74 request shape) → ``[]`` —
the caller then builds the byte-identical two-message request."""
assert history_to_messages([], _settings()) == []
def test_user_turn_maps_to_user_role() -> None:
got = history_to_messages([_turn("user", "What port does Tailscale use?")], _settings())
assert got == [{"role": "user", "content": "What port does Tailscale use?"}]
def test_brain_turn_without_thinking_maps_to_assistant_role() -> None:
"""No ``thinking`` key → a plain assistant message: NO
``reasoning_content`` key at all (A4 gating, ``None`` case)."""
got = history_to_messages([_turn("brain", "Tailscale runs on 41641/udp.")], _settings())
assert got == [{"role": "assistant", "content": "Tailscale runs on 41641/udp."}]
assert "reasoning_content" not in got[0]
def test_brain_turn_with_thinking_carries_reasoning_content() -> None:
"""A prior thinking block travels as ``reasoning_content`` on the
assistant message (A4 — the preserve-thinking wire convention the
response side already reads)."""
thinking = "Tailscale's wire protocol port is 41641/udp."
got = history_to_messages(
[_turn("brain", "Tailscale runs on 41641/udp.", thinking=thinking)],
_settings(),
)
assert got == [
{
"role": "assistant",
"content": "Tailscale runs on 41641/udp.",
"reasoning_content": thinking,
}
]
def test_brain_turn_with_empty_thinking_omits_reasoning_content() -> None:
"""``thinking=""`` is "empty" for the A4 gate — no
``reasoning_content`` key (an empty scratchpad carries nothing)."""
got = history_to_messages(
[_turn("brain", "Same answer.", thinking="")], _settings()
)
assert got == [{"role": "assistant", "content": "Same answer."}]
assert "reasoning_content" not in got[0]
def test_result_is_chronological_oldest_to_newest() -> None:
"""The input is oldest-first; the output must be too — the newest
turn ends up LAST, directly ahead of the current user message the
caller appends."""
turns = [
_turn("user", "q1"),
_turn("brain", "a1", thinking="t1"),
_turn("user", "q2"),
_turn("brain", "a2"),
_turn("user", "q3"),
]
got = history_to_messages(turns, _settings())
assert [m["role"] for m in got] == ["user", "assistant", "user", "assistant", "user"]
assert [m["content"] for m in got] == ["q1", "a1", "q2", "a2", "q3"]
assert got[1]["reasoning_content"] == "t1"
assert "reasoning_content" not in got[3]
# ---------- turn-count budget ----------
def test_turn_cap_keeps_newest_and_drops_oldest() -> None:
"""The newest ``history_max_turns`` turns are kept; the OLDEST are
the ones dropped (newest-first walk, stop at the count cap)."""
turns = [_turn("user", f"q{i}") for i in range(1, 6)] # q1 … q5, oldest first
got = history_to_messages(turns, _settings(history_max_turns=3, history_max_chars=10_000))
assert [m["content"] for m in got] == ["q3", "q4", "q5"]
def test_default_turn_cap_is_40() -> None:
"""45 turns under the DEFAULT caps (40 turns / 24 000 chars, short
texts so the char budget never binds) keep the newest 40."""
turns = [_turn("user", f"question number {i}") for i in range(1, 46)]
got = history_to_messages(turns, _settings())
assert len(got) == 40
assert got[0]["content"] == "question number 6" # the five oldest are gone
assert got[-1]["content"] == "question number 45"
# ---------- char budget ----------
def test_char_budget_counts_text_plus_thinking() -> None:
"""The per-turn size is ``len(text) + len(thinking or "")`` — prior
thinking blocks count against the same budget as the answer text."""
# Newest-first sizes: 5 + 100 (20+80) + 10; budget 110 keeps the
# newest two (105) and drops the oldest (115 > 110).
turns = [
_turn("user", "a" * 10), # oldest — dropped whole
_turn("brain", "b" * 20, thinking="c" * 80),
_turn("user", "d" * 5), # newest
]
got = history_to_messages(turns, _settings(history_max_turns=40, history_max_chars=110))
assert len(got) == 2
assert got[0]["content"] == "b" * 20
assert got[0]["reasoning_content"] == "c" * 80
assert got[1]["content"] == "d" * 5
def test_char_budget_exact_fit_is_kept() -> None:
"""Cumulative chars EQUAL to the cap fit (≤, not <) — the exact-fit
turn is kept, and the older turn that would push past is dropped."""
turns = [
_turn("user", "a" * 10), # oldest — 100+10=110 > 100, dropped
_turn("brain", "b" * 100), # newest — exactly the 100-char cap, kept
]
got = history_to_messages(turns, _settings(history_max_turns=40, history_max_chars=100))
assert [m["content"] for m in got] == ["b" * 100]
def test_overflowing_turn_is_dropped_whole_never_truncated() -> None:
"""A turn that would overflow the remaining budget is DROPPED WHOLE
(A3) — its text appears nowhere in the result, not even partially,
and the walk stops there (the kept history stays a contiguous
newest window)."""
big = "x" * 120 # alone it would overflow the 100-char budget
turns = [
_turn("user", "old question"),
_turn("brain", "old answer"),
_turn("brain", big), # newest — does not fit at all
]
got = history_to_messages(turns, _settings(history_max_turns=40, history_max_chars=100))
assert got == [] # the newest does not fit → nothing is kept
assert not any("x" in m["content"] for m in got)
def test_overflowing_middle_turn_stops_the_walk() -> None:
"""Newest-first: the newest fits, the NEXT (middle) turn would
overflow → it is dropped whole AND the walk stops — the oldest turn
is not sneaked in across the gap (no discontinuous history)."""
turns = [
_turn("user", "a" * 5), # oldest — never even considered
_turn("user", "b" * 51), # middle — 60+51=111 > 100, dropped whole
_turn("user", "c" * 60), # newest — fits (60 ≤ 100)
]
got = history_to_messages(turns, _settings(history_max_turns=40, history_max_chars=100))
assert [m["content"] for m in got] == ["c" * 60]
def test_zero_char_budget_yields_no_history() -> None:
"""``history_max_chars=0`` is a budget that fits nothing — the
kill-switch shape (no history, pre-phase-74 two-message request)."""
turns = [_turn("user", "q1"), _turn("brain", "a1")]
assert history_to_messages(turns, _settings(history_max_chars=0)) == []
def test_zero_turn_budget_yields_no_history() -> None:
"""``history_max_turns=0`` keeps no turns even though chars are free."""
turns = [_turn("user", "q1"), _turn("brain", "a1")]
assert history_to_messages(turns, _settings(history_max_turns=0)) == []
# ---------- budgets together ----------
def test_turn_cap_wins_when_chars_remain() -> None:
"""Both budgets in play: plenty of chars, a small turn cap — the
count cap stops the walk first (newest 3 of 5 kept)."""
turns = [_turn("user", f"q{i}") for i in range(1, 6)]
got = history_to_messages(turns, _settings(history_max_turns=3, history_max_chars=10_000))
assert len(got) == 3
assert [m["content"] for m in got] == ["q3", "q4", "q5"]
def test_char_cap_wins_when_turns_remain() -> None:
"""Symmetrically: plenty of turn budget, a tight char cap — the char
budget stops the walk (2 of 3 turns kept)."""
turns = [
_turn("user", "a" * 50), # oldest — dropped (50+60=110 > 100)
_turn("user", "b" * 60),
_turn("user", "c" * 40), # newest
]
got = history_to_messages(turns, _settings(history_max_turns=40, history_max_chars=100))
assert [m["content"] for m in got] == ["b" * 60, "c" * 40]