From e344b7df9ceac337482f60a48d3f8f4f1cada210 Mon Sep 17 00:00:00 2001
From: Peter Steinberger <steipete@gmail.com>
Date: Thu, 22 Jan 2026 08:31:07 +0000
Subject: [PATCH] fix: preserve antigravity thinking block types

---
 src/agents/pi-embedded-runner/google.ts | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/src/agents/pi-embedded-runner/google.ts b/src/agents/pi-embedded-runner/google.ts
index 41b93225f..0ec334192 100644
--- a/src/agents/pi-embedded-runner/google.ts
+++ b/src/agents/pi-embedded-runner/google.ts
@@ -91,7 +91,8 @@ function sanitizeAntigravityThinkingBlocks(messages: AgentMessage[]): AgentMessa
       out.push(msg);
       continue;
     }
-    const nextContent = [];
+    type AssistantContentBlock = Extract<AgentMessage, { role: "assistant" }>["content"][number];
+    const nextContent: AssistantContentBlock[] = [];
     let contentChanged = false;
     for (const block of assistant.content) {
       if (
@@ -115,7 +116,11 @@ function sanitizeAntigravityThinkingBlocks(messages: AgentMessage[]): AgentMessa
         continue;
       }
       if (rec.thinkingSignature !== candidate) {
-        nextContent.push({ ...rec, thinkingSignature: candidate });
+        const nextBlock = {
+          ...(block as unknown as Record<string, unknown>),
+          thinkingSignature: candidate,
+        } as AssistantContentBlock;
+        nextContent.push(nextBlock);
         contentChanged = true;
       } else {
         nextContent.push(block);