feat(context): render retry repair mandate as final user turn (#293)

buildRetryFeedbackEntry was L1/SYSTEM, so PromptRenderer folded it into the
leading system block — far from the assistant/tool transcript and weaker than
the original stage task. Flip it to USER role and give the renderer an explicit
trailing repair-mandate slot (a sourceType set, extensible for recovery later):
repair mandates are lifted out of the inline flow and emitted once as the final
message, after the tool evidence and the steering anchor. retryFeedback is
already in REQUIRED_SOURCE_TYPES so it stays unprunable. Golden renderer test
proves the final message is the repair USER mandate and leading system no longer
carries it.

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
This commit is contained in:
2026-07-20 18:20:59 +04:00
parent f860d1b8af
commit bc050f8e8a
4 changed files with 47 additions and 4 deletions
@@ -53,6 +53,30 @@ class PromptRendererOrderingTest {
Unit
}
@Test
fun `retry repair mandate renders as the final user turn after the tool transcript`() = kotlinx.coroutines.runBlocking {
// #293: even though the repair mandate is stamped mid-transcript by input order, the renderer
// must lift it to the very end (after assistant/tool evidence) and never fold it into system.
val builder = DefaultContextPackBuilder(DefaultContextCompressor())
val entries = listOf(
entry("sys", ContextLayer.L0, EntryRole.SYSTEM, "systemPrompt"),
entry("task", ContextLayer.L1, EntryRole.USER, "agentPrompt"),
entry("repair", ContextLayer.L1, EntryRole.USER, "retryFeedback"),
entry("call1", ContextLayer.L2, EntryRole.ASSISTANT, "assistantToolCall", sourceId = "tool1"),
entry("result1", ContextLayer.L2, EntryRole.TOOL, "toolResult", sourceId = "tool1"),
)
val pack = builder.build(ContextPackId("p"), sessionId, stageId, entries, TokenBudget(limit = 4000))
val messages = PromptRenderer.render(pack)
assertEquals("user" to "repair", messages.last().role to messages.last().content)
assertEquals(1, messages.count { it.content == "repair" }, "mandate must appear exactly once")
org.junit.jupiter.api.Assertions.assertFalse(
messages.first().content.contains("repair"),
"leading system must not carry the repair mandate",
)
Unit
}
@Test
fun `without ordinals the live user turn still renders last (router chat fallback)`() {
// Packs built outside DefaultContextPackBuilder (router chat) leave ordinal at 0;
@@ -50,7 +50,7 @@ class ContextFeedbackTest {
}
@Test
fun `latest retry event for the stage becomes a labeled L1 system entry`() {
fun `latest retry event for the stage becomes a labeled L1 user repair mandate`() {
val events = listOf(
stored(payload = RetryAttemptedEvent(SessionId("s"), StageId("impl"), 1, 3, "old reason")),
stored(payload = RetryAttemptedEvent(SessionId("s"), StageId("other"), 1, 3, "other stage")),
@@ -59,7 +59,8 @@ class ContextFeedbackTest {
val entry = buildRetryFeedbackEntry(events, StageId("impl"))!!
assertEquals("retryFeedback", entry.sourceType)
assertEquals(ContextLayer.L1, entry.layer)
assertEquals(EntryRole.SYSTEM, entry.role)
// #293: USER role so it renders as the trailing repair mandate, not folded into leading system.
assertEquals(EntryRole.USER, entry.role)
assertTrue(entry.content.contains("Attempt 2 of 3"), "content: ${entry.content}")
assertTrue(entry.content.contains("tests failed: NPE in FooTest"), "content: ${entry.content}")
}