From 3ffb9a3b5e39d3c8f2d5140895ffc298595a72af Mon Sep 17 00:00:00 2001 From: Peter Steinberger Date: Mon, 12 Jan 2026 17:57:24 +0000 Subject: [PATCH] fix: keep session sanitizer stable --- src/agents/pi-embedded-helpers.ts | 3 ++- src/agents/session-tool-result-guard.test.ts | 1 - 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/agents/pi-embedded-helpers.ts b/src/agents/pi-embedded-helpers.ts index 4a2368696..793c68257 100644 --- a/src/agents/pi-embedded-helpers.ts +++ b/src/agents/pi-embedded-helpers.ts @@ -95,9 +95,10 @@ export async function sanitizeSessionMessagesImages( ): Promise { // We sanitize historical session messages because Anthropic can reject a request // if the transcript contains oversized base64 images (see MAX_IMAGE_DIMENSION_PX). - const base = options?.sanitizeToolCallIds + const sanitizedIds = options?.sanitizeToolCallIds ? sanitizeToolCallIdsForCloudCodeAssist(messages) : messages; + const base = sanitizedIds; const out: AgentMessage[] = []; for (const msg of base) { if (!msg || typeof msg !== "object") { diff --git a/src/agents/session-tool-result-guard.test.ts b/src/agents/session-tool-result-guard.test.ts index 07cdd6618..ef7eefca5 100644 --- a/src/agents/session-tool-result-guard.test.ts +++ b/src/agents/session-tool-result-guard.test.ts @@ -1,5 +1,4 @@ import type { AgentMessage } from "@mariozechner/pi-agent-core"; - import { SessionManager } from "@mariozechner/pi-coding-agent"; import { describe, expect, it } from "vitest";