From 2f82ade66f3451270939b95dc611108b5af9abd5 Mon Sep 17 00:00:00 2001
From: Peter Steinberger <steipete@gmail.com>
Date: Fri, 13 Mar 2026 21:35:31 +0000
Subject: [PATCH] test: tighten assistant scaffolding coverage

---
 .../text/assistant-visible-text.test.ts       | 32 +++++++++++++++++++
 1 file changed, 32 insertions(+)

diff --git a/src/shared/text/assistant-visible-text.test.ts b/src/shared/text/assistant-visible-text.test.ts
index 234d37b96da..1962a86e371 100644
--- a/src/shared/text/assistant-visible-text.test.ts
+++ b/src/shared/text/assistant-visible-text.test.ts
@@ -42,8 +42,40 @@ describe("stripAssistantInternalScaffolding", () => {
     expect(stripAssistantInternalScaffolding(input)).toBe(input);
   });
 
+  it("keeps relevant-memories tags inside inline code", () => {
+    const input = "Use `<relevant-memories>example</relevant-memories>` literally.";
+    expect(stripAssistantInternalScaffolding(input)).toBe(input);
+  });
+
   it("hides unfinished relevant-memories blocks", () => {
     const input = ["Hello", "<relevant-memories>", "internal-only"].join("\n");
     expect(stripAssistantInternalScaffolding(input)).toBe("Hello\n");
   });
+
+  it("trims leading whitespace after stripping scaffolding", () => {
+    const input = [
+      "<thinking>",
+      "secret",
+      "</thinking>",
+      "   ",
+      "<relevant-memories>",
+      "internal note",
+      "</relevant-memories>",
+      "  Visible",
+    ].join("\n");
+    expect(stripAssistantInternalScaffolding(input)).toBe("Visible");
+  });
+
+  it("preserves unfinished reasoning text while still stripping memory blocks", () => {
+    const input = [
+      "Before",
+      "<thinking>",
+      "secret",
+      "<relevant-memories>",
+      "internal note",
+      "</relevant-memories>",
+      "After",
+    ].join("\n");
+    expect(stripAssistantInternalScaffolding(input)).toBe("Before\n\nsecret\n\nAfter");
+  });
 });