From 1a2e5e35cc38808cdc4fdfe955a0060ddfdae593 Mon Sep 17 00:00:00 2001
From: Eva <eva@100yen.org>
Date: Sun, 12 Apr 2026 14:56:26 +0700
Subject: [PATCH] Fix maintenance test message typing

---
 .../context-engine-maintenance.test.ts            | 15 +++++++++++++--
 1 file changed, 13 insertions(+), 2 deletions(-)

diff --git a/src/agents/pi-embedded-runner/context-engine-maintenance.test.ts b/src/agents/pi-embedded-runner/context-engine-maintenance.test.ts
index 48189eec0af..2a32a310d18 100644
--- a/src/agents/pi-embedded-runner/context-engine-maintenance.test.ts
+++ b/src/agents/pi-embedded-runner/context-engine-maintenance.test.ts
@@ -204,7 +204,14 @@ describe("runContextEngineMaintenance", () => {
     const maintain = vi.fn(async (params?: unknown) => {
       await (params as { runtimeContext?: ContextEngineRuntimeContext } | undefined)?.runtimeContext?.rewriteTranscriptEntries?.({
         replacements: [
-          { entryId: "entry-1", message: { role: "assistant", content: "done", timestamp: 2 } },
+          {
+            entryId: "entry-1",
+            message: {
+              role: "assistant",
+              content: [{ type: "text", text: "done" }],
+              timestamp: 2,
+            },
+          },
         ],
       });
       return {
@@ -242,7 +249,11 @@ describe("runContextEngineMaintenance", () => {
         replacements: [
           {
             entryId: "entry-1",
-            message: { role: "assistant", content: "done", timestamp: 2 },
+            message: {
+              role: "assistant",
+              content: [{ type: "text", text: "done" }],
+              timestamp: 2,
+            },
           },
         ],
       },